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
要点
- enumerate函数可以简洁的代码迭代iterator,而且可以指出当前这轮循环的序号。
- 使用enumerate取代range指定下标的取值。
- 可以通过enumerate的第二个参数指定起始序号(默认为0)。