python 列表生成式

2019-08-11  本文已影响0人  狗嗨旺
[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处理,最后统一生成为一个数据列表

L = 5
data = [1,3,5,7,9]
flag = [1 if i in data else 0 for i in range(L) ]

[0, 1, 0, 1, 0, 1, 0, 1, 0, 1]

L = 5
data = [1,3,5,7,9]
flag = [1 for i in range(L) if i in data]

[1, 1, 1, 1, 1]

上一篇 下一篇

猜你喜欢

热点阅读