Python3 - 同时迭代多个序列

问题

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

解决方案

同时迭代多个序列时，使用 zip() 函数。比如：

m = [1, 5, 4, 2, 10, 7]
n = [101, 78, 37, 15, 62, 99]

for x, y in zip(m, n):
    print(x, y)

1 101
5 78
4 37
2 15
10 62
7 99

zip(m, n) 会生成一个可返回元组 (x, y) 的迭代器，其中x来自m，y来自n。 一旦其中某个序列的元素迭代完，整个迭代宣告结束。 因此迭代长度跟参数中最短序列长度一致。比如：

a = [1, 2, 3]
b = ['w', 'x', 'y', 'z']

for i in zip(a, b):
    print(i)

(1, 'w')
(2, 'x')
(3, 'y')

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

from itertools import zip_longest

for i in zip_longest(a, b):
    print(i)

(1, 'w')
(2, 'x')
(3, 'y')
(None, 'z')


for i in zip_longest(a, b, fillvalue=0):
    print(i)

(1, 'w')
(2, 'x')
(3, 'y')
(0, 'z')

讨论

当成对处理数据时，zip() 函数是很有用的。 比如，将2个序列打包成一个字典：

headers = ['name', 'shares', 'price']
values = ['ACME', 100, 490.1]
s = dict(zip(headers, values))
print(s)
{'name': 'ACME', 'shares': 100, 'price': 490.1}

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

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

name = ACME
shares = 100
price = 490.1

zip() 可以接受多于两个的序列的参数，返回的结果元组中元素个数跟输入序列个数一样。比如：

a = [1, 2, 3]
b = [10, 11, 12]
c = ['x','y','z']
for i in zip(a, b, c):
    print(i)

(1, 10, 'x')
(2, 11, 'y')
(3, 12, 'z')

zip() 会创建一个迭代器来作为结果返回。 如果需要将成对的值存储在列表中，要使用 list() 函数。比如：

print(zip(a, b))
<zip object at 0x10f4b92c8>

print(list(zip(a, b)))
[(1, 10), (2, 11), (3, 12)]