基于另一个相同形状的 2D 数组对 2D numpy 数组（按行）进行排序

Question

有两个数组：

X = np.array([[1,2,3], [2,3,1]])
X
array([[1, 2, 3],
       [2, 3, 1]])
Y = np.array([['A','B', 'C'], ['A','B', 'C']])
Y
array([['A', 'B', 'C'],
       ['A', 'B', 'C']], 
     dtype='|S1')

我正在尝试根据 X 的值逐行对 Y 进行排序，而不是遍历每一行，即

xord = X.argsort()

for i in range(X.shape[0]):
    print Y[i][xord[i]]

['A' 'B' 'C']
['C' 'A' 'B']

有没有更高效的方法来根据 X 对应的行顺序对数组 Y 进行排序？

Answer 1

首先可以使用np.argsort根据排序后的位置获取X元素的索引，然后可以根据X的索引使用np.take()从Y获取元素：

>>> s=np.argsort(X)
>>> np.take(Y,s)
array([['A', 'B', 'C'],
       ['C', 'A', 'B']], 
      dtype='|S1')