zip函数用法

语法：zip(iter1 [,iter2 [...]]) --> zip object 

解析：zip函数接受一序列可迭代对象作为参数，将迭代对象中的元素组合成一个个元组，最后返回一个由元组组成的zip对象，若传入的迭代对象的长度不同，则最终返回的zip对象的长度以传入对象中最短的长度相同。

In [1]: a = [1,2,3,4]
   ...: b = [3,1,4,5]
   ...: c = [5,6,3,2,1,7,8]
   ...: zip(a,c)
   ...:
Out[1]: <zip at 0x4afab08>

In [2]: list(zip(a,c))
Out[2]: [(1, 5), (2, 6), (3, 3), (4, 2)]

In [3]: list(zip(a,b))
Out[3]: [(1, 3), (2, 1), (3, 4), (4, 5)]

zip函数配合*符号可以将zip过得对象解压

In [11]: a = [1,2,3,4]

In [12]: c = [5,6,3,2,1,7,8]

In [13]: d = zip(a,c)

In [14]: d
Out[14]: <zip at 0x4a91488>

In [15]: list(zip(*d))
Out[15]: [(1, 2, 3, 4), (5, 6, 3, 2)]

利用zip函数配合*符号可以将一个数组的转置

In [17]: d = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

In [18]: d
Out[18]:
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

In [19]: list(zip(*d))
Out[19]: [(1, 4, 7), (2, 5, 8), (3, 6, 9)]

In [20]: np.array(list(zip(*d)))
Out[20]:
array([[1, 4, 7],
       [2, 5, 8],
       [3, 6, 9]])