python-在numpy中按索引选择

假设两个数组

ind = 
array([[1,3,2,4,0],[0,1,4],[3,1]])

x =
array([[[24,97,28,57,59],[97,67,94,77,50],[56,89,25,55,76],[88,21,50,24]],[[54,83,64,81,12],[89,49,15,26,97],[94,32,79],[24,63,40]],[[41,99,84,21],[12,9,85,43,28],[75,98,48,10,[93,37,22,63]]])

我想根据第一个数组重新排序第二个数组(第一个数组是索引)

因此,结果可能如下所示.

array([[[97,59,24],[67,76,56],[21,24,88]],[[64,41,99],[43,12,9],[10,75,98],[22,93,94]]])
# x[0]s are reordered by ind[0] and so on.

np.take有可能吗？

最佳答案

使用take_along_axis很容易：

>>> np.take_along_axis(x,ind[:,None,:],2)
array([[[97,94]]])

如果您使用的是1.15 numpy之前的版本,则可以执行以下操作：

>>> m,n,k = x.shape
>>> m,k = np.ogrid[:m,:n,:k]
>>> x[m,:]]
array([[[97,94]]])