Python的列表推导是通过已经存在的列表去生成另一个经过加工的列表,比如:
生成基础列表
a=list(range(1,11))
print(a)
现在要生成一个列表,是a列表里面所有元素的平方
基础用法
使用列表的[] 将表达式括起来,然后里面写上跟for循环相关的表达式,具体如下:
b=[x**2 for x in a]
print(b)
加if的列表推导
列表推导可以加筛选,比如将a中所有偶数的元素做平方,并剔除奇数
c=[x**2 for x in a if x%2==0]
print(c)
加if else的列表推导
可以加if的话,基本也可以加if else,只是不能像上面c那样是放在if语句后面。具体例子如下(偶数做平方,奇数为三次方):
d=[x**2 if x%2==0 else x**3 for x in a]
print(d)