python - 从列表中查找矩阵中的字符串元素

Question

我想搜索 n x n 矩阵中的字符串值，看看它们是否存在于列表中。输出应该是一个 n x n 矩阵， bool 值 True 或 False 取决于是否找到了字符串。矩阵和列表如下所示:

matrix = [['aa', 'ba', 'ca'], ['ab', 'bb', 'cb'], ['ac', 'bc', 'cc']]
list = ['ba','cb','dg'] 

我有这个代码:

matrixFound = [[for x in matrix] for y in matrix]

我需要以某种方式在元素中包含这个语句:

matrix in list

输出应该是这样的矩阵:

[[False,True,False],[False,False,True],[False,False,False]]

我正在考虑使用 map 或 lambda 来解决这个问题，但无法正确编码。 应该怎么做？可以使用常规的 python 或 numpy。

Answer 1

你可以使用 numpy 的 isin:

np.isin(matrix,list2)                                                                                                                                                               

# array([[False,  True, False],
#       [False, False,  True],
#       [False, False, False]])