Python 學習之 zip 函式

問題的引出

有時候，你可能想同時迭代兩個序列。假設有下面兩個列表：

names = ['anne', 'beth', 'george', 'damon']
ages = [12, 45, 32, 102]
如果要列印名字和對應的年齡，可以這樣做：

>>> names = ['anne', 'beth', 'george', 'damon']
>>> ages = [12, 45, 32, 102]
>>> for i in range(len(names)): 
...     print(names[i], 'is', ages[i], 'years old')
... 
anne is 12 years old
beth is 45 years old
george is 32 years old
damon is 102 years old
>>> 

你可能覺得有點麻煩，尤其是第3行，寫的程式碼有點多。有沒有更好的辦法呢？

並行迭代工具zip

Python 提供了一個很有用的並行迭代工具——內建函式 zip，它將兩個序列“縫合”起來，並返回一個由元組組成的序列。返回值是一個適合迭代的物件，要檢視其內容，可使用list將其轉換為列表。

>>> list(zip(names, ages)) 
[('anne', 12), ('beth', 45), ('george', 32), ('damon', 102)]

“縫合”後，可在迴圈中將元組解包。

>>> for name, age in zip(names, ages): 
...     print(name, 'is', age, 'years old') 
... 
anne is 12 years old
beth is 45 years old
george is 32 years old
damon is 102 years old
>>> 

函式zip可用於“縫合”任意數量的序列。需要指出的是，當序列的長度不同時，函式zip將在最短的序列用完後停止“縫合”。

>>> list(zip(range(5), range(100000000))) 
[(0, 0), (1, 1), (2, 2), (3, 3), (4, 4)]

以下是其他的例子：

>>> a = [1,2,3]
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b)     # 返回一個物件
>>> zipped
<zip object at 0x103abc288>
>>> list(zipped)          # 用 list() 轉換為列表
[(1, 4), (2, 5), (3, 6)]
>>> list(zip(a,c))        # 元素個數與最短的列表一致
[(1, 4), (2, 5), (3, 6)]
>>> list(zip(a,b,c))      # 縫合3個序列
[(1, 4, 4), (2, 5, 5), (3, 6, 6)]
>>> 

也可以把列表和元組縫合在一起。

>>> a = (1,2,3)
>>> b = [4,5,6]
>>> list(zip(a,b))
[(1, 4), (2, 5), (3, 6)]

zip(*)的用法

操作符*與zip函式配合可以實現與zip相反的功能，即將“縫合”的序列拆成多個元組。

>>> a = [1,2]
>>> b = [3,4]
>>> c = ['a','b']
>>> zipped = zip(a,b,c)    # 縫合
>>> a1, a2, a3 = zip(*zipped)   # 拆包
>>> a1
(1, 2)
>>> a2
(3, 4)
>>> a3
('a', 'b')
>>> b1,b2,b3 = zip(*zip(a,b,c))
>>> b1
(1, 2)
>>> b2
(3, 4)
>>> b3
('a', 'b')

參考資料

[1] http://www.runoob.com/python3/python3-func-zip.html

[2]《Python基礎教程》人民郵電出版社，第3版