在python中搜索和替换2D数组中的元素

Question

我试图找出一种方法来搜索2D数组以找到某个单词然后替换该单词。

例如：

pets = [['I', 'have', 'a', 'cat'], ['She', 'has', 'a', 'pet', 'cat']]

我需要一种搜索​​单词“cat＆＃39;并将其替换为“狗”字样。

Answer 1

您可以使用列表推导检查所有元素，并将'cat'替换为'dog'：

pets = [['I', 'have', 'a', 'cat'], ['She', 'has', 'a', 'pet', 'cat']]

new_pets = [[p if p.lower()!='cat' else 'dog' for p in s] for s in pets]
print(new_pets) # => [['I', 'have', 'a', 'dog'], ['She', 'has', 'a', 'pet', 'dog']]

Answer 2

使用list comprehension：

pets = [[val.replace('cat', 'dog') for val in row] for row in pets]