ZH奶酪：Python中zip函数的使用方法

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

定义：zip([iterable, …])
zip()是Python的一个内建函数，它接受一系列可迭代的对象作为参数，将对象中对应的元素打包成一个个tuple（元组），然后返回由这些
tuples组成的list（列表）。若传入参数的长度不等，则返回list的长度和参数中长度最短的对象相同。利用*号操作符，可以将list
unzip（解压），看下面的例子就明白了：

>>> a = [,,]
>>> b = [,,]
>>> c = [,,,,]
>>> zipped = zip(a,b)
[(, ), (, ), (, )]
>>> zip(a,c)
[(, ), (, ), (, )]
>>> zip(*zipped)
[(, , ), (, , )]

（1）二维矩阵转置

比如我们有一个由列表描述的二维矩阵
a = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
通过python列表推导的方法，我们也能轻易完成这个任务

（从下边的代码也可以看出这种one-line for-loop的index顺序从左向右依次是维数）

print [ [row[col] for row in a] for col in range(len(a[]))]
[[, , ], [, , ], [, , ]]

另外一种方法就是利用zip函数：

>>> a = [[, , ], [, , ], [, , ]]
>>> zip(*a)
[(, , ), (, , ), (, , )]
>>> map(list,zip(*a))
[[, , ], [, , ], [, , ]]

这种方法速度更快但也更难以理解

zip(*a)：将list看成tuple解压，恰好得到我们“行列互换”的效果；

map(list,zip(*a))：再通过对每个元素应用list()函数，将tuple转换为list；

zip函数接受任意多个序列作为参数，将所有序列按相同的索引组合成一个元素是各个序列合并成的tuple的新序列，新的序列的长度以参数中最短的序列为准。另外(*)操作符与zip函数配合可以实现与zip相反的功能，即将合并的序列拆成多个tuple。

①tuple的新序列

>>>x=[,,],y=['a','b','c']
>>>zip(x,y)
[(,'a'),(,'b'),(,'c')]

②新的序列的长度以参数中最短的序列为准.

>>>x=[,],y=['a','b','c']
>>>zip(x,y)
[(,'a'),(,'b')]

③(*)操作符与zip函数配合可以实现与zip相反的功能,即将合并的序列拆成多个tuple。

>>>x=[,,],y=['a','b','c']
>>>zip(*zip(x,y))
[(,,),('a','b','c')]

（2）以指定概率获取元素

>>> import random
>>> def random_pick(seq,probabilities):
x = random.uniform(, )
cumulative_probability = 0.0
for item, item_probability in zip(seq, probabilities):
cumulative_probability += item_probability
if x < cumulative_probability: break
return item>>> for i in range():
random_pick("abc",[0.1,0.3,0.6])'c'
'b'
'c'
'c'
'a'
'b'
'c'
'c'
'c'
'a'
'b'
'b'
'c'
'a'
'c'