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

微信公衆號：

歡迎關注！