你想同時疊代多個序列,每次分别從一個序列中取一個元素。
zip函數傳回元組疊代器
為了同時疊代多個序列,使用
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)
多于兩個參數
>>> 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(a, b)
<zip object at 0x1007001b8>
>>> list(zip(a, b))
[(1, 10), (2, 11), (3, 12)]