python的zip函数

描述

zip() 函数用于将可迭代的对象作为参数，将对象中对应的元素打包成一个个元组，然后返回由这些元组组成的列表。

如果各个迭代器的元素个数不一致，则返回列表长度与最短的对象相同，利用 * 号操作符，可以将元组解压为列表。

语法

zip 语法：

zip([iterable, ...])

返回值

元祖列表

如：

for i in zip(range(5)):
    print(i)

（注意：python3中zip产生的可迭代对象只能操作一次，操作完一次内存就被销毁）

实例

1. 使用zip实现矩阵各个对应元素相乘（点乘）：

m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
n = [[2, 2, 2], [3, 3, 3], [4, 4, 4]]

print([x*y for a,b in zip(m,n) for x,y in zip(a,b)])# 实现矩阵的点乘

print([x+y for a,b in zip(m,n) for x,y in zip(a,b)])# 实现矩阵的相加

print([x-y for a,b in zip(m,n) for x,y in zip(a,b)])# 实现矩阵的相减

输出：

[2, 4, 6, 12, 15, 18, 28, 32, 36]
[3, 4, 5, 7, 8, 9, 11, 12, 13]
[-1, 0, 1, 1, 2, 3, 3, 4, 5]

 

2.使用*zip还原

a = [1,2,3]
b = [4,5,6]
t =  zip(a,b)
print(zip(a,b))
for i in zip(*zip(a,b)):
    print(list(i))