欢迎访问宙启技术站
智能推送

Python函数:如何使用enumerate()函数在迭代列表时同时返回索引和元素?

发布时间:2023-06-24 14:07:11

在Python中,enumerate()函数用于在迭代列表时同时返回元素及其索引。它返回一个迭代器对象,该迭代器对象包含一系列元素,每个元素都是一个包含两个元素的元组, 个元素是当前迭代元素的索引,第二个元素是当前迭代元素。

使用enumerate()函数可以使代码更加简洁易读,并且使得在需要知道当前元素所在列表位置的时候更加方便。

下面是一些例子,以展示如何使用enumerate()函数在迭代一个列表时同时返回元素及其索引。

例子1:打印列表元素及其索引

fruits = ['apple', 'banana', 'orange', 'grape']

for index, fruit in enumerate(fruits):
    print(index, fruit)

输出:

0 apple
1 banana
2 orange
3 grape

例子2:创建一个字典,键为列表元素的索引,值为列表元素的平方

numbers = [1, 2, 3, 4, 5]
squares_dict = {}

for index, num in enumerate(numbers):
    squares_dict[index] = num**2

print(squares_dict)

输出:

{0: 1, 1: 4, 2: 9, 3: 16, 4: 25}

例子3:在列表中搜索元素的索引

fruits = ['apple', 'banana', 'orange', 'grape']

def search_fruit(fruit_name):
    for index, fruit in enumerate(fruits):
        if fruit == fruit_name:
            return index
    return -1

print(search_fruit('orange')) # 输出:2
print(search_fruit('pear'))   # 输出:-1

总之,使用enumerate()函数可以在迭代列表时同时返回元素及其索引,这对于当前元素所在的位置有要求的情况非常有用,而不需要手动记录当前元素所在的位置。