Python学习之zip函数

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版