python中的zip()函数

一、zip(*iterables)函数详解

1、zip()函数的定义

• 从参数中的多个迭代器取元素组合成一个新的迭代器
• 传入参数：列表、元组、字典
• 返回值：一个zip对象，其内部元素为元组；可以转化成列表或者元组

2、zip()函数的用法

• 当zip()函数中只有一个参数时，zip(iterable)从迭代器中依次选取一个元组，组成一个元组。

# python 3
# zip()函数单个参数
In [1]: list1 = [1, 2, 3, 4]
In [2]: tuple1 = zip(list1)
In [3]: type(tuple1)
Out[3]: zip
In [4]: list(tuple1)
Out[4]: [(1,), (2,), (3,), (4,)]

• 当zip()函数有两个参数时，zip(a,b)函数分别从a和b中取一个元素组成元组，再次将组成的元组组合成一个新的迭代器。a与b的维数相同时，正常组合对应位置的元素。当a和b行或列数不同时，取两者中的最小的行列数。

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

n=[[1,1,1],[2,2,3],[3,3,3]]

p=[[1,1,1],[2,2,2]]

print(list(zip(m,n)))
print(list(zip(m,p)))

输出结果：

[([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 3]), ([7, 8, 9], [3, 3, 3])]
[([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 2])]

3、zip()函数的应用

• 矩阵的加减、点乘（也可以用for循环+列表推导式实现)

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

n=[[1,1,1],[2,2,3],[3,3,3]]

p=[[1,1,1],[2,2,2]]

multiply_result=[x*y for a,b in zip(m,n) for x,y in zip(a,b)]
plus_result=[x+y for a,b in zip(m,n) for x,y in zip(a,b)]

print(multiply_result)
print(plus_result)

输出结果：

[1, 2, 3, 8, 10, 18, 21, 24, 27]
[2, 3, 4, 6, 7, 9, 10, 11, 12]

*zip(*iterables)函数详解

*zip()函数是zip()函数的逆过程，将zip对象编程原先组合前的数据。

In [24]: m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
In [25]: n = [[1, 1, 1], [2, 2, 3], [3, 3, 3]]
In [26]: print(*zip(m, n))
([1, 2, 3], [1, 1, 1]) ([4, 5, 6], [2, 2, 3]) ([7, 8, 9], [3, 3, 3])
In [27]: m2, n2 = zip(*zip(m,n))
In [28]: m == list(m2) and n == list(n2)
Out[28]: True