我有这个矩阵:
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 中的行。
最佳答案
如果你像这样存储矩阵:
矩阵 = [ [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/