zip()是Python的一个内建函数,它接受一系列可迭代的对象作为参数,将对象中对应的元素打包成一个个tuple(元组),然后返回由这些tuples组成的list(列表)。若传入参数的长度不等,则返回list的长度和参数中长度最短的对象相同。利用*号操作符,可以将list unzip(解压)
简单实例
-
当所有参数等长时:
x = [1, 2, 3]
y = [4, 5, 6]
z = [7, 8, 9]
xyz = zip(x, y, z)
结果:[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
-
当参数不等长时:
x = [1, 2, 3]
y = [4, 5, 6, 7]
xy = zip(x, y)
结果:[(1, 4), (2, 5), (3, 6)]
-
传入单参数:
x = [1, 2, 3]
x = zip(x)
结果:[(1,), (2,), (3,)]
-
无参时:
x = zip()
结果:[]
高级应用
- 使用*:
x = [1, 2, 3]
y = [4, 5, 6]
z = [7, 8, 9]
xyz = zip(x, y, z)
u = zip(*xyz)
结果:[(1, 2, 3), (4, 5, 6), (7, 8, 9)]
在运行zip(*xyz)之前,xyz的值是:[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
那么,zip(*xyz) 等价于 zip((1, 4, 7), (2, 5, 8), (3, 6, 9))
所以,运行结果是:[(1, 2, 3), (4, 5, 6), (7, 8, 9)]
- 使用该方法可以实现矩阵转置:
x=[[1,2,3],
[4,5,6],
[7,8,9]]
y = zip(*x)
map(list, y)
- 可以实现压缩与解压
- 可以用来构造字典:
x = ['a', 'b', 'c']
y = ['x', 'y', 'z']
z = zip(x, y)
d = dict(z)
结果:{'a': 'x', 'c': 'z', 'b': 'y'}
-
x = [1, 2, 3]
r = zip(* [x] * 3)
结果:[(1, 1, 1), (2, 2, 2), (3, 3, 3)]
[x]生成一个列表的列表,它只有一个元素x
[x] * 3生成一个列表的列表,它有3个元素,[x, x, x]
zip(* [x] * 3)的意思就明确了,zip(x, x, x)
- 与
iter
合用
a = [1, 2, 3, 4, 5, 6]
b = zip(*([iter(a)] * 2))
结果:[(1, 2), (3, 4), (5, 6)]
iter()是一个迭代器,返回一个可迭代对象,可以使用next()访问
[iter(a)]表示每次取一个元素,[iter(a)] * 2表示迭代两次,每次步长为2,
也即是每次取了三个元素,迭代完就变成了[1, 3, 5], [2, 4, 6],然后
再使用zip()就变成了[(1, 2), (3, 4), (5, 6)]
网友评论