python压缩函数--------zip

1.描述

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

如果各个可迭代对象的元素个数不一致,则返回的对象长度与最短的可迭代对象相同。

利用 * 号操作符,与zip相反,进行解压。

2.语法

zip(iterable1,iterable2, ...)

iterable是可迭代的对象,如字符串、列表、元祖、字典

解压用法:

'''假设arr是一个10行两列的矩阵'''

import numpy as np

arr = np.ones((10,2))
a,b = zip(*arr)

'''此时a,b均为包含10个元素的列表'''

注:Python2中直接返回一个由元组组成的列表,Python3中返回的是一个对象,如果想要得到列表,可以用 list() 函数进行转换

3.用法

python3中

>>> a = [1,2,3] #此处可迭代对象为列表
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b)
>>> zipped
<zip object at 0x02B01B48> #返回的是一个对象
>>> list(zipped)
[(1, 4), (2, 5), (3, 6)] #使用list()函数转换为列表
>>> list(zip(a,c))
[(1, 4), (2, 5), (3, 6)]
>>> zipped = zip(a,b)
>>> list(zip(*zipped)) #解压也使用list进行转换
[(1, 2, 3), (4, 5, 6)]

python2中

>>>a = [1,2,3] #此处可迭代对象为列表
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b)     # 打包为元组的列表
>>> zipped
[(1, 4), (2, 5), (3, 6)]
>>> zip(a,c)              # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
>>> zip(*zipped)          # 与 zip 相反,可理解为解压
[(1, 2, 3), (4, 5, 6)]

最最重要的用法,与for循环搭配用来同时取多个迭代器中同一位置的数:

list1 = [2,3,4]
list2 = [4,5,6]
 
for x,y in zip(list1,list2):
    print(x,y,'--',x*y)

结果:

猜你喜欢

转载自blog.csdn.net/goodxin_ie/article/details/89318704