• 4.11 同时迭代多个序列
    • 问题
    • 解决方案
    • 讨论

    4.11 同时迭代多个序列

    问题

    你想同时迭代多个序列,每次分别从一个序列中取一个元素。

    解决方案

    为了同时迭代多个序列,使用 zip() 函数。比如:

    1. >>> xpts = [1, 5, 4, 2, 10, 7]
    2. >>> ypts = [101, 78, 37, 15, 62, 99]
    3. >>> for x, y in zip(xpts, ypts):
    4. ... print(x,y)
    5. ...
    6. 1 101
    7. 5 78
    8. 4 37
    9. 2 15
    10. 10 62
    11. 7 99
    12. >>>

    zip(a, b) 会生成一个可返回元组 (x, y) 的迭代器,其中x来自a,y来自b。一旦其中某个序列到底结尾,迭代宣告结束。因此迭代长度跟参数中最短序列长度一致。

    1. >>> a = [1, 2, 3]
    2. >>> b = ['w', 'x', 'y', 'z']
    3. >>> for i in zip(a,b):
    4. ... print(i)
    5. ...
    6. (1, 'w')
    7. (2, 'x')
    8. (3, 'y')
    9. >>>

    如果这个不是你想要的效果,那么还可以使用 itertools.zip_longest() 函数来代替。比如:

    1. >>> from itertools import zip_longest
    2. >>> for i in zip_longest(a,b):
    3. ... print(i)
    4. ...
    5. (1, 'w')
    6. (2, 'x')
    7. (3, 'y')
    8. (None, 'z')
    9.  
    10. >>> for i in zip_longest(a, b, fillvalue=0):
    11. ... print(i)
    12. ...
    13. (1, 'w')
    14. (2, 'x')
    15. (3, 'y')
    16. (0, 'z')
    17. >>>

    讨论

    当你想成对处理数据的时候 zip() 函数是很有用的。比如,假设你头列表和一个值列表,就像下面这样:

    1. headers = ['name', 'shares', 'price']
    2. values = ['ACME', 100, 490.1]

    使用zip()可以让你将它们打包并生成一个字典:

    1. s = dict(zip(headers,values))

    或者你也可以像下面这样产生输出:

    1. for name, val in zip(headers, values):
    2. print(name, '=', val)

    虽然不常见,但是 zip() 可以接受多于两个的序列的参数。这时候所生成的结果元组中元素个数跟输入序列个数一样。比如;

    1. >>> a = [1, 2, 3]
    2. >>> b = [10, 11, 12]
    3. >>> c = ['x','y','z']
    4. >>> for i in zip(a, b, c):
    5. ... print(i)
    6. ...
    7. (1, 10, 'x')
    8. (2, 11, 'y')
    9. (3, 12, 'z')
    10. >>>

    最后强调一点就是, zip() 会创建一个迭代器来作为结果返回。如果你需要将结对的值存储在列表中,要使用 list() 函数。比如:

    1. >>> zip(a, b)
    2. <zip object at 0x1007001b8>
    3. >>> list(zip(a, b))
    4. [(1, 10), (2, 11), (3, 12)]
    5. >>>

    原文:

    http://python3-cookbook.readthedocs.io/zh_CN/latest/c04/p11_iterate_over_multiple_sequences_simultaneously.html