欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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)]