使用列表访问 numpy 矩阵中的元素（python）

Question

我有一个 10x10 numpy 矩阵，我有一个包含要查询的元素索引的列表。

matrix = np.zeros((10, 10), dtype=int)

indices = [[2,3], [3,4]]

我面临的问题是，我真正想要的是元素matrix[2,3]但matrix[indices[0]]给出不同的 output 因为后者实际上意味着matrix[[2,3]] 。 output 是 2 个不同的行。

我该如何解决这个问题？

Answer 1

我最终找到了解决方案。

将索引存储为列表中的元组有效

indices = [(2,3), (3,4)]然后matrix[indices[0]]给出所需的 output