如何索引numpy数组？

Question

一个简单的例子：我有一个numpy数组（[[5,4,2]，[1,8,8]，[4,5,6]]）和另一个数组（[2,2,1] ）。第二个数组包含第一个数组中每行的感兴趣的索引。我希望我的最终数组只是一个矢量数组（[2,8,5]）。我正在寻找与first_array [second_array]类似的东西，但事实证明这是难以捉摸的。谢谢！

Answer 1

您可以尝试以下内容：

>>> arr
array([[5, 4, 2],
       [1, 8, 8],
       [4, 5, 6]])
>>> idx
array([2, 2, 1])
>>> arr[np.arange(arr.shape[0]), idx]
array([2, 8, 5])