替换列表中的特定元素

Question

码：

>>> mylist = ['abc','def','ghi']
>>> mylist
['abc', 'def', 'ghi']
>>> for i,v in enumerate(mylist):
...     if v=='abc':
...             mylist[i] = 'XXX'
... 
>>> mylist
['XXX', 'def', 'ghi']
>>> 

在这里，我尝试用'XXX'替换所有出现的'abc' 'XXX' 。 有没有更短的方法来做到这一点？

Answer 1

您可以使用列表推导，而不是使用显式的for循环。 这允许您迭代列表中的所有元素并过滤它们或将它们映射到新值。

在这种情况下，您可以使用条件表达式 。 它类似于(v == 'abc') ? 'XXX' : v (v == 'abc') ? 'XXX' : v用其他语言。

将它们放在一起，您可以使用以下代码：

mylist = ['XXX' if v == 'abc' else v for v in mylist]

Answer 2

使用带有三元运算/ 条件表达式的列表推导：

['XXX' if item == 'abc' else item for item in mylist]