一、什么是enumerate函数
在Python的标准函数库中，我们可以找到一个叫做enumerate()的非常有用的函数。该函数的作用是为列表和其他可遍历的对象，在遍历的时候，自动生成一个计数器，用于计数遍历的元素的序号。
fruits = ["apple", "banana", "cherry"]
for index, fruit in enumerate(fruits):
print(index, fruit)
输出结果为：
0 apple
1 banana
2 cherry
enumerate()函数返回包含两个元素的元组，第一个元素是计数器的值，第二个元素是可迭代对象中对应的元素。
二、enumerate函数的用途
一个常见的使用场景是遍历列表时需要知道元素的下标，同时又不想自己维护一个下标计数器，这时候就可以使用enumerate()函数。
1. 在遍历列表时获取元素下标
在遍历列表时，需要使用元素的下标，可以使用正常的for循环遍历，然后使用range(len(list))的方式获取对应下标的元素。但是，这种方式显然不够简单优雅。可以使用enumerate()函数，代码如下：
fruits = ["apple", "banana", "cherry"]
for index, fruit in enumerate(fruits):
print(index, fruit)
2. 将列表转换为字典
有时候需要将一个列表转换成字典，字典的键是列表的元素，而字典的值是元素在列表中的下标。可以使用enumerate()函数配合字典推导式完成这个任务，代码如下：
fruits = ["apple", "banana", "cherry"]
fruit_dict = {fruit: index for index, fruit in enumerate(fruits)}
print(fruit_dict)
输出结果为：
{'apple': 0, 'banana': 1, 'cherry': 2}
3. 实现多个可变参数的遍历
在某些场景中，需要同时遍历多个可变参数的元素，就可以使用enumerate()函数。
fruits = ["apple", "banana", "cherry"]
prices = [1.5, 2, 3.5]
for index, (fruit, price) in enumerate(zip(fruits, prices)):
print(f"#{index}: {fruit} costs {price} dollars.")
输出结果为：
#0: apple costs 1.5 dollars.
#1: banana costs 2 dollars.
#2: cherry costs 3.5 dollars.
三、enumerate函数的实现原理
enumerate()函数的实现其实非常简单，它的源代码可以以以下方式来实现：
def enumerate(sequence, start=0):
n = start
for elem in sequence:
yield n, elem
n += 1
该函数定义了一个iterable对象（sequence），并使用yield返回包含计数器n和元素elem的二元组。在for循环期间，该函数自动增加计数器n的值，这样，enumerate()函数就自动在可迭代对象上遍历并返回计数器值和与之对应的元素。
原创文章，作者：WOMO，如若转载，请注明出处：https://www.506064.com/n/150022.html
微信扫一扫 
支付宝扫一扫 