问题:迭代一个序列,但是又想记录下序列中的当前处理到的元素索引
使用内建的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)