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

Question

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

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

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

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

我该如何解决这个问题？

标签： pythonnumpy

Answer 1

我最终找到了解决方案。

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

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