python - 在python中从矩阵中提取子元素

我有这个矩阵：

0   0   0   138
0   8   0   0
0   1   0   0
131 0   0   138
0   0   138 0
0   0   0   0
0   115 0   8

这个索引向量：

idx = [2,4,5]

我需要从矩阵中获取138的所有条目的行索引和列索引，但是，只针对idx中的行。

最佳答案

如果这样存储矩阵：
matrix = [ [0, 0, 0, 138], [0, 8, 0, 0], ... ]
你的工作很简单：

result = []
for i in idx:
    row = matrix[i]
    for j in range(len(row)):
        if row[j] == 138:
            result.append((i, j))
return result

关于python - 在python中从矩阵中提取子元素，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/49817025/