Effective Python 第7条：enumerate取代range

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）。