enumerate() --以索引-值对的形式迭代序列

2023年2月27日 243次阅读 来源: cook__

问题：迭代一个序列，但是又想记录下序列中的当前处理到的元素索引

使用内建的enumerate()函数来解决
my_list = ['a', 'b', 'c']
for index, value in enumerate(my_list):
    print(index, value)
    
0 a
1 b
2 c

# 可以传入一个start参数作为起始索引
for idx, val in enumerate(my_list, 1):
    print(idx, val)
    
1 a
2 b
3 c

使用示例：将文件中单词和它们所出现的行之间建立映射关系:

import collections

word_summary = collections.defaultdict(list)

with open('/etc/passwd') as f:
    lines = f.readlines()

for idx, line in enumerate(lines):
    words = [word.strip().lower() for word in line.split()]
    for word in words:
        word_summary[word].append(idx)

print(word_summary)  # key-list 一键多值的dict, value是list包裹起来的
总结：

1、enumerate()的返回值是一个enumerate对象实例，它是一个迭代器，可返回连续的元组；
2、正确的在元组序列上使用enumerate()函数：

data = [(1, 2), (3, 4), (5, 6)]
# 正确写法
for idx, (a, b) in enumerate(data):
    print(idx, (a, b))
    
0 (1, 2)
1 (3, 4)
2 (5, 6)

# 错误写法
for idx, a, b in enumerate(data):
    print(idx, a, b)  # ValueError: not enough values to unpack (expected 3, got 2)
    原文作者：cook__
    原文地址: https://www.jianshu.com/p/f4a6627eb5a6
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞