天天看點

同時疊代多個序列

你想同時疊代多個序列,每次分别從一個序列中取一個元素。

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

繼續閱讀