Python内建函数:zip

时间:2022-06-21 18:40:15

zip()是Python的一个内建函数,它接受一系列可迭代的对象作为参数,将对象中对应的元素打包成一个个tuple(元组),然后返回由这些tuples组成的list。若传入参数的长度不等,则返回list的长度和参数中长度最短的对象相同。

>>> a = [1, 2, 3]
>>> b = [4, 5, 6]
>>> c = [4, 5, 6, 7, 8, 9]
>>> zipped = zip(a, b)
[(1,4),(2,5),(3,6)]
>>> zip(a, c)
[(1,4),(2,5),(3,6)]
>>> zip(*zipped)
[(1,2,3),(4,5,6)]

zip()利用*号操作符,可以将list unzip(解压),看下面的例子
* 二维矩阵变换(矩阵的行列互换)
比如我们有一个由列表描述的二维矩阵
a = [[1,2,3], [4,5,6], [7,8,9]]
通过python列表推导的方法,我们也能轻易完成这个任务
print [[row[col] for row in a] for col in range(len(a[0]))]
[[1,4,7],[2,5,8],[3,6,9]]
另外一种让人困惑的方法就是利用zip函数:

>>> a = [[1,2,3], [4,5,6], [7,8,9]]
>>> zip(*a)
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
>>> map(list,zip(*a))
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]

总结zip()的三种使用使用场景

  • tuple的新序列
>>> x = [1,2,3], y = ['a','b','c']
>>> zip(x, y)
[(1,'a'),(2,'b'),(3,'c')]
  • 新的序列的长度以参数中最短的序列为准
>>> x = [1,2], y = ['a', 'b', 'c']
>>> zip(x, y)
[(1,'a'),(2,'b')]
  • (*)操作符与zip函数配合可以实现与zip相反的功能,即将合并的序列拆成多个tuple。
>>>>x=[1,2,3],y=['a','b','c']
>>>>zip(*zip(x,y))
[(1,2,3),('a','b','c')]

本文参考链接