同時迭代多個序列

問題

你想同時迭代多個序列，每次分別從一個序列中取一個元素。

解法

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

>>> 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) 會生成一個可返回元組 (x, y) 的迭代器，其中 x 來自 a， y 來自 b。一旦其中某個序列到底結尾，迭代宣告結束。因此迭代長度跟參數中最短序列長度一致。

>>> 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() 函數是很有用的。比如，假設你頭列表和一個值列表，就像下面這樣：

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

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

s = dict(zip(headers,values))

或者你也可以像下面這樣產生輸出：

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

雖然不常見，但是 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() 函數。比如：

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

總結

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（4.11）：同时迭代多个序列的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。