Effective Python 第7条:enumerate取代range

range可以用来迭代一系列整数,如下:

for i in range(10):
    print(i)

当我们迭代数据结构,例如字符串列表,可以直接在这个序列上迭代,如下:

flavor_list = ['vanilla', 'chocolate', 'pecan']
for flavor in flavor_list:
    print(f'{flavor} is delicious')
# vanilla is delicious
# chocolate is delicious
# pecan is delicious

但是当我们迭代list的过程中,要知道元素在list里的位置时,就需要使用range,如下:

flavor_list = ['vanilla', 'chocolate', 'pecan']
for i in range(len(flavor_list)):
    flavor = flavor_list[i]
    print(f'{i+1}: {flavor}')
# 1: vanilla
# 2: chocolate
# 3: pecan

但是上面步骤太多了,我们可以使用enumerate简化。enumerate能够把任何一种迭代器(iterator)封装成惰性生成器(lazy generator,参见第30条)。这样每次循环就从iterator获取下一个值,同时还会给出本轮循环的序号,即生成器每次产生一对输出值。下面通过内置的next函数手动推进enumerate所返回的这个iterator。如下:

it = enumerate(flavor_list)
print(next((it)))
print(next((it)))
# (0, 'vanilla')
# (1, 'chocolate')

enumerate输出的每一对数据,都可以拆分(unpacking)到for语句两个变量里,如下:

for i, flavor in enumerate(flavor_list):
    print(f'{i+1}: {flavor}')
# 1: vanilla
# 2: chocolate
# 3: pecan

还可以通过enumerate的第二个参数指定起始序列号,如下,从1开始:

for i, flavor in enumerate(flavor_list, 1):
    print(f'{i}: {flavor}')
# 1: vanilla
# 2: chocolate
# 3: pecan

要点

  1. enumerate函数可以简洁的代码迭代iterator,而且可以指出当前这轮循环的序号。
  2. 使用enumerate取代range指定下标的取值。
  3. 可以通过enumerate的第二个参数指定起始序号(默认为0)。
上一篇:SWR的使用教程


下一篇:chocolate Counting(牛客)