在numpy中逐个索引一个数组

Question

假设我有一个矩阵a具有任意值：

array([[ 2, 4, 5, 3],
       [ 1, 6, 8, 9],
       [ 8, 7, 0, 2]])

矩阵B包含a中元素的指数：

^{pr2}$

如何从A中选择由B指向的值，即：

A[B] = [[2, 2, 4, 5],
        [1, 9, 8, 6],
        [2, 0, 7, 8]]

Answer 1

最近的版本添加了一个take_along_axis函数来完成此任务：

In [203]: A = np.array([[ 2, 4, 5, 3], 
     ...:        [ 1, 6, 8, 9], 
     ...:        [ 8, 7, 0, 2]])                                                
In [204]: B = np.array([[0, 0, 1, 2], 
     ...:        [0, 3, 2, 1], 
     ...:        [3, 2, 1, 0]])                                                 
In [205]: np.take_along_axis(A,B,1)                                             
Out[205]: 
array([[2, 2, 4, 5],
       [1, 9, 8, 6],
       [2, 0, 7, 8]])

还有一个put_along_axis。在