python - 使用索引数组返回 'ndarray' 的值

Question

我正在尝试使用索引数组获取 ndarray 的值。

假设a是目标数组，b是索引数组。然后:

a = np.asarray([[[1,2,3]],[[2,3,4]],[[3,4,5]]])
b = np.asarray([[0,0,1],[0,0,2]])
print a[tuple(map(tuple, b))]

应该返回我想要的内容，但我收到“索引超出范围错误”。

我期望的输出是:

[2,3]

Answer 1

IIUC，您想要a[tuple(b.T)]。使数字唯一以便于验证:

>>> a = np.arange(9).reshape(3,1,3)
>>> b = np.asarray([[1,0,2],[2,0,0]])
>>> a
array([[[0, 1, 2]],

       [[3, 4, 5]],

       [[6, 7, 8]]])
>>> b
array([[1, 0, 2],
       [2, 0, 0]])
>>> a[1,0,2]
5
>>> a[2,0,0]
6
>>> a[tuple(b.T)]
array([5, 6])