美文网首页Python
enumerate() --以索引-值对的形式迭代序列

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

作者: cook__ | 来源:发表于2018-10-03 15:09 被阅读6次

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

使用内建的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)

相关文章

网友评论

    本文标题:enumerate() --以索引-值对的形式迭代序列

    本文链接:https://www.haomeiwen.com/subject/ocwzoftx.html