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
要点