Python列表理解 - 簡單

Question

我有一個列表，我希望只對那些符合某些條件的條目使用某個函數 - 使其他條目保持不變。

例如：假設我想只乘以那些偶數的元素。

a_list = [1, 2, 3, 4, 5] 

求購結果：

a_list => [1, 4, 3, 8, 5] 

但是[elem * 2 for elem in a_list if elem %2 == 0]收率[4, 8]（它充當另外的過濾器）。

什麼是正確的方式去呢？

Answer 1

使用conditional expression：

[x * 2 if x % 2 == 0 else x 
for x in a_list] 

（數學怪才注：您還可以解決這種特殊情況下與

[x * (2 - x % 2) for x in a_list] 

但我想反正喜歡的第一個選項;）

Answer 2

a_list = [1, 2, 3, 4, 5] 

print [elem*2 if elem%2==0 else elem for elem in a_list ] 

或者，如果您有很長的列表要修改：

a_list = [1, 2, 3, 4, 5] 

for i,elem in enumerate(a_list): 
    if elem%2==0: 
     a_list[i] = elem*2 

所以，只有偶數元素被修改

Answer 3

你可以使用拉姆達：

>>> a_list = [1, 2, 3, 4, 5] 
>>> f = lambda x: x%2 and x or x*2 
>>> a_list = [f(i) for i in a_list] 
>>> a_list 
[1, 4, 3, 8, 5] 

編輯 - 約agf此言思考我做了我的代碼第2個版本：

>>> a_list = [1, 2, 3, 4, 5] 
>>> f = lambda x: x if x%2 else x*2 
>>> a_list = [f(i) for i in a_list] 
>>> a_list 
[1, 4, 3, 8, 5]