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]