Python 列表理解 - 简单

Question

我有一个列表，我只想对满足特定条件的那些条目使用特定函数 - 其他条目保持不变。

示例:假设我只想将那些偶数的元素乘以 2。

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]

但无论如何我更喜欢第一个选项；)