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行或列数不同时，取两者中的最小的行列数。

# zip()函数有两个参数
In [5]: m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
In [6]: n = [[1, 1, 1], [2, 2, 3], [3, 3, 3]]
In [7]: p = [[1, 1, 1], [2, 2, 2]]
In [8]: list(zip(m,n))
Out[8]: [([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 3]), ([7, 8, 9], [3, 3, 3])]
In [9]: list(zip(m,p))
Out[9]: [([1, 2, 3], [1, 1, 1]), ([4, 5, 6], [2, 2, 2])]

3、zip()函数的应用

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

In [10]: m = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
In [11]: n = [[1, 1, 1], [2, 2, 3], [3, 3, 3]]
# 矩阵的点乘
In [13]: [x*y for a, b in zip(m, n) for x, y in zip(a, b)]
Out[13]: [1, 2, 3, 8, 10, 18, 21, 24, 27]
# 矩阵相加
In [14]: [x+y for a, b in zip(m, n) for x, y in zip(a, b)]
Out[14]: [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