zip() 函数用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表。如果各个迭代器的元素个数不一致,则返回列表长度与最短的对象相同,利用 * 号操作符,可以将元组解压为列表
>>>a = [1,2,3]
>>> b = [4,5,6]
>>> c = [4,5,6,7,8]
>>> zipped = zip(a,b) # 打包为元组的列表
[(1, 4), (2, 5), (3, 6)]
>>> zip(a,c) # 元素个数与最短的列表一致
[(1, 4), (2, 5), (3, 6)]
>>> zip(*zipped) # 与 zip 相反,可理解为解压,返回二维矩阵式
[(1, 2, 3), (4, 5, 6)]
列表推到式可以使用函数或复杂表达式。
def f(v):
if v%2==0:
v=v**2
else:
v=v+1
return v
>>> print([f(v) for v in [2,3,4,-1] if v>0])
[4, 4, 16]
print([v**2 if v%2==0 else v+1 for v in [2,3,4,-1] if v>0])
[4, 4, 16]
列表推导式支持文件对象迭代:(r 错误 :https://blog.youkuaiyun.com/tengqingyong/article/details/75307016)
fp=open(r'C:\Users\Sky\Desktop\one.txt','r')
>>> print([line for line in fp])
['11111111111111111111111111111111\n', '11111111111111111111111111111111\n']
fp.close()
使用列表推导式生成100以内的所有素数
[p for p in range(2,100) if 0 not in [p%d for d in range(2,int(math.sqrt(p))+1)]]
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97]
>>>