Python 的zip函数

Python 的zip函数，简言之，实现矩阵的转置。比如，给出4x2矩阵

1  5
2  6
3  7
4  8

通过zip函数，可以得到它的2x4转置矩阵

5  6  7  8

具体做法是：把前面的4x2矩阵表示为两个向量的列表，向量可以用列表(list)或者元组(tuple)表示，但是zip返回的一定是tuple表示的向量：

>>> x = (1, 2, 3, 4)
>>> y = (5, 6, 7, 8)
>>> zip(x,y)
[(1, 5), (2, 6), (3, 7), (4, 8)]  #得到2x4转置矩阵
>>> z = (9, 10, 11, 12)  
>>> zip(x,y,z)  #再加一列
[(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)]  #得到3x4转置矩阵

可以看出，zip的参数是任意多个向量，zip函数定义为：

def zip(seq1, seq2, *more_seqs):

返回多个向量组成的列表（list of tuple）

[tuple1, tuple2,...]

如果要把得到的转置矩阵再次转置，只要把向量的列表用*号解开传给zip就可以了，因为用*号解开后，仍然是zip需要的多个向量：

>>> T = zip(x,y,z)
>>> T
[(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)] #转置后的矩阵
>>> zip(*T) #用*号解开传给zip，相当于 zip((1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12))
[(1, 2, 3, 4), (5, 6, 7, 8), (9, 10, 11, 12)] #得到再次转置的矩阵

zip函数的一个用处，是将key和value两个list构造成dict函数所需的(key, value)对，以生成一个字典。就像把拉链两边拉起来成为一个完整的拉链一样。

>>> name = ['Alice', 'Ben', 'Celia', 'Dave']
>>> age = [12, 13, 14, 15]
>>> keypair = zip(name, age)
>>> keypair
[('Alice', 12), ('Ben', 13), ('Celia', 14), ('Dave', 15)]
>>> dict(keypair)
{'Dave': 15, 'Ben': 13, 'Alice': 12, 'Celia': 14}

如果您对本文有疑问或者寻求合作，欢迎 联系邮箱 。邮箱已到剪贴板

精彩评论