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

在np.argsort上使用ind1：

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