4.11 同時(shí)迭代多個(gè)序列

問(wèn)題

你想同時(shí)迭代多個(gè)序列，每次分別從一個(gè)序列中取一個(gè)元素。

解決方案

為了同時(shí)迭代多個(gè)序列，使用 zip() 函數(shù)。比如：

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

zip(a, b) 會(huì)生成一個(gè)可返回元組 (x, y) 的迭代器，其中x來(lái)自a，y來(lái)自b。一旦其中某個(gè)序列到底結(jié)尾，迭代宣告結(jié)束。因此迭代長(zhǎng)度跟參數(shù)中最短序列長(zhǎng)度一致。

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

如果這個(gè)不是你想要的效果，那么還可以使用 itertools.zip_longest() 函數(shù)來(lái)代替。比如：

>>> 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')
>>>

討論

當(dāng)你想成對(duì)處理數(shù)據(jù)的時(shí)候 zip() 函數(shù)是很有用的。比如，假設(shè)你頭列表和一個(gè)值列表，就像下面這樣：

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

使用zip()可以讓你將它們打包并生成一個(gè)字典：

s = dict(zip(headers,values))

或者你也可以像下面這樣產(chǎn)生輸出：

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

雖然不常見(jiàn)，但是 zip() 可以接受多于兩個(gè)的序列的參數(shù)。這時(shí)候所生成的結(jié)果元組中元素個(gè)數(shù)跟輸入序列個(gè)數(shù)一樣。比如;

>>> 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')
>>>

最后強(qiáng)調(diào)一點(diǎn)就是，zip() 會(huì)創(chuàng)建一個(gè)迭代器來(lái)作為結(jié)果返回。如果你需要將結(jié)對(duì)的值存儲(chǔ)在列表中，要使用 list() 函數(shù)。比如：

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