编写高质量Python（第7条） 尽量用enumerate取代range

第7条 尽量用enumerate取代range

​ Python 内置的 range 函数适合用于迭代一系列整数。

from random import randint

random_bits = 0
for i in range(32):
	if randint(0, 1):
		print(bin(random_bits))
		
>>>
0b10000100000101100101000011111101

​ 如果要迭代的是某种数据结构，例如字符串列表，那么可以在这个序列上进行迭代，用不着专门通过 range 设定一个取值范围，然后把这个范围里的整数值，依次当作下标来访问列表中的元素。

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

​ 当然有的时候，在迭代 list 的过程中也需要知道当前处理的元素在 list 里的位置。 例如，我把爱吃的冰淇淋口味写在了 flavor_list 列表里面，在打印每种口味时，我还想指出这种口味在自己心中的排名。有了实现这样的功能，我们可以用传统的方法实现。

for i in range(len(flavor_list)):
    flavor = flavor_list[i]
    print(f'{i + 1}: {flavor}')

>>>
1: vanilla
2: chocolate
3: peacan
4: strawberry

​ 这种写法跟前面那几个例子有点乱。因为我们要首先知道列表的长度，然后要根据列表的长度构造取值范围，分别访问列表里的对应元素。

​ Python 中有个内置函数，叫做 enumerate，它可以解决刚才的问题。enumerate 能够把任何一种迭代器封装成惰性生成器(lazy generator，参见第三十条)。这样的话，每次循环的时候，它只需要从 iterator 里面获取下一个值就行了，同时还会给出本轮循环的序号，即生成器每次产生的一对值。下面通过内置的 next 函数手动推进 enumerate 所返回的这个 iterator ，给大家展示 enumerate 的原理。

it = enumerate(flavor_list)
print(next(it))
print(next(it))

>>>
(0, 'vanilla')
(1, 'chocolate')	

​ enumerate 输出的每一对数据，都可以拆分(unpacking)到 for 语句的那两个变量里面(unpacking机制参考 第六条)，这样会让代码更加清晰。

for i, flavor in enumerate(flavor_list):
    print(f'{i + 1}: {flavor}')

>>>
1: vanilla
2: chocolate
3: pecan
4: strawberry

​ 另外，还可以通过 enumerate 的第二个参数指定起始序号，这样就不用在每次打印的时候去调整了。例如，本例可以从1开始算

for i, flavor in enumerate(flavor_list, 1):
    print(f'{i}: {flavor}')