python - numpy.sort 的逆映射

Question

我想获取np.sort逆向映射的索引

换句话说，如果我有密码

x=[[4,3],[2,3],[4,2]]
ind1=np.lexsort((x[:,1],x[:,0]))
y=x[ind1] #y=[[2,3],[4,2],[4,3]]

那么我想要一个这样的索引

y[ind]

返回原始函数 x。有没有办法得到这个索引？

Answer 1

在 ind1 上使用 np.argsort:

idx = np.argsort(ind1)
y[idx]
# array([[4, 3],
#        [2, 3],
#        [4, 2]])