替換列表中的特定元素

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]