在python中搜索和替换二维数组中的元素

Question

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

例如：

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

我需要一种方法来搜索“猫”这个词并将其替换为“狗”这个词。

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']]