python中List推导式中添加条件判断

1.List推导式基本语法：

• [x for in data]        将data中的元素组成一个列表并返回

arr = [i / 2 for i in range(5)]
print(arr)

2.List推导式中添加条件判断（单条件、多条件）：

注意：表达式最终是一个值（当表达式为：i = i + 1时就犯了语法错误，编译都通不过）。

• [x for in data if condition]       （单条件）

  将data中满足condition(True)的元素组合成一个列表并返回

arr = [ i for in range( 1 , 11 )  if i % 2 ]    # 得到 [1 , 10]之间的奇数列表
print(arr)

• [exp1 if condition else exp2 for x in data]        （多条件）

根据condition的值执行相应的表达式，循环结束时返回组成的列表。表达式（exp）一般是基于对元素的处理。比如最简单的如果元素 i 是偶数则 i加1，奇数则 i减1。

flag = [1 if i % 2 else 0 for i in range(1, 11)]
print(flag)