如何索引3D数组

Question

这个问题已经在这里有了答案： 具有（n-1）d个数组1个答案的索引n维数组

我认为这里的答案更为直接（我立即了解了问题所在）。重复的帖子很棒，我将它们与通用的索引功能相关联，但是我需要时间来理解。

我想通过沿轴查询的索引从3d数组中提取数据，但是我不知道如何编写。

#data:
arr= np.array([[[ 0.000,  0.200],
                [ 0.000,  0.100],
                [-0.932, -0.073]],

               [[ 0.000,  0.000],
                [-0.932, -0.073],
                [-1.626, -0.900]],

               [[-0.132, -0.073],
                [-1.626, -0.900],
                [-1.802, -0.688]],

               [[-1.626, -0.900],
                [-1.802, -0.688],
                [-3.059, -1.190]]])

# This is the index
idx= np.array([[0, 1],
               [1, 2],
               [0, 0],
               [0, 1]])

# Expected output
    array([[ 0.000,  0.100],
           [-0.932, -0.900],
           [-0.132, -0.073],
           [-1.626, -0.688],

# If it is 2d it is like this:
axis=0:
arr[idx,np.arange(arr.shape[1])]
axis=1:
arr[np.arange(arr.shape[0]),idx]

但是我不知道如何在3d数组上编写它，我希望得到帮助。谢谢。

Answer 1

您不清楚将idx应用于哪个轴。我有匹配的值来查看模式。

无论如何，这是索引

In [88]: arr[np.arange(4)[:,None], idx, np.arange(2)]
Out[88]: 
array([[ 0.   ,  0.1  ],
       [-0.932, -0.9  ],
       [-0.132, -0.073],
       [-1.626, -0.688]])

写了2个范围，因此它们广播成与idx相同的形状（4,2）。