做leetcode时,发现官方给出的答案中用的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行或列数不同时,取两者中的最小的行列数。
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
原文链接:https://blog.youkuaiyun.com/wyisfish/article/details/79242387