Python列表推导式if else及一些应用

两种形式的列表推导式

1. 只有if…版

模板：
[item for item in data if condition]
示例：

# 剔除data中的奇数
data = [1, 2, 3, 4, 5, 6, 7, 8, 9]
test = [item for item in data if item % 2 == 0]
print(test)
# 结果
[2, 4, 6, 8]

此处if主要起条件判断作用，data数据中只有满足if条件的才会被留下，最后统一生成为一个数据列表

2. if…else… 版

模板：
[exp1 if condition else exp2 for x in data]
示例：

# 一行代码实现data中偶数位的元素加2
data = [1,2,3,4,5,6,7,8,9]
test = [data[i]+2 if (i+1)%2 == 0 else data[i] for i in range(len(data))]
print(test)
# 结果
[1, 4, 3, 6, 5, 8, 7, 10, 9]

此处if…else…主要起赋值作用，当data中的数据满足if条件时将其做exp1处理，否则按照exp2处理，最后统一生成为一个数据列表

参考：

https://blog.csdn.net/jasonLee_lijiaqi/article/details/79305779

广告时间：

Github：https://github.com/Prolht 点个star哟！

个人博客：https://www.lex666.online

新浪微博： https://weibo.com/u/2415026333

微信公众号：

欢迎关注！