美文网首页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