Python zip() 函数
描述
zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表。如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同。
利用 * 号操作符,可以将元组解压为列表
a = [1,2,3]
b = [4,5,6]
c = [4,5,6,7,8]
zipped = zip(a,b) # 打包为元组的列表
# [(1, 4), (2, 5), (3, 6)]
zip(a,c) # 元素个数与最短的列表一致
# [(1, 4), (2, 5), (3, 6)]
zip(*zipped) # 与 zip 相反,*zipped 可理解为解压,返回二维矩阵式
# [(1, 2, 3), (4, 5, 6)]
zip(*)用途
获取矩阵每列的元素
a = [[1,2,3],[4,5,6],[7,8,9]]
for col in zip(*a):
print(col)
'''
(1, 4, 7)
(2, 5, 8)
(3, 6, 9)
'''