python - 重新排列二维 numpy 数组的每一行，并为每行指定一个索引

Question

这个问题已经有答案了:

Indexing one array by another in numpy (4 个回答)

已关闭 4 年前。

我有一个二维 numpy 数组 x

x = np.array([[6,7,8],                                             
              [9,10,11]])

我想用索引重新排列数组 x 的每一行

index = np.array([[2,0,1],                                                  
                  [1,2,0]])

这样

result[i,:] = x[i,index[i,:]]

result = np.array([[8,6,7],                                                  
                  [10,11,9]])

我有一个很大的数组 x，所以我不想用循环来执行此操作。

Answer 1

您可以将 0 范围内的数字广播到数组的形状，并以与现在相同的方式对列进行索引:

x[np.arange(x.shape[0])[:, None], index]

array([[ 8,  6,  7],
       [10, 11,  9]])