python列表推导式中使用if-else
程序员文章站
2022-03-09 18:21:02
...
列表推导式总共有两种形式:
①[x for x in data if condition]
此处if主要起条件判断作用,data数据中只有满足if条件的才会被留下,最后统一生成为一个数据列表
②[exp1 if condition else exp2 for x in data]
此处if...else主要起赋值作用,当data中的数据满足if条件时将其做exp1处理,否则按照exp2处理,最后统一生成为一个
数据列表
例子如下:
#1-100中3的倍数
p = [x for x in range(1,101) if x%3 == 0]
#1-100中,不是3的倍数的数去相反数,其余的数保持不变
q = [x if x%3==0 else -x for x in range(1,101)]
下一篇: Python基础--列表与元祖