python - 如何使用存储在另一个 numpy 数组中的一组索引访问 numpy 数组？

Question

我有一个 numpy 数组，它存储一组索引，我需要访问另一个 numpy 数组。

我尝试使用 for 循环，但它没有像我预期的那样工作。

情况是这样的:

>>> a
array([[1, 2],
       [3, 4]])
>>> c
array([[0, 0],
       [0, 1]])
>>> a[c[0]]
array([[1, 2],
       [1, 2]])
>>> a[0,0]         # the result I want
1

上面是我实际代码的简化版本，其中 c 数组要大得多，所以我必须使用 for 循环来获取每个索引。

Answer 1

将其转换为元组:

>>> a[tuple(c[0])]
1

因为 list 和 array 索引触发 advanced indexing . 元组(大部分)是基本切片。