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版