python - 如何在 Python 中获取原始列表的排序列表的索引？

Question

numpy 的argsort 函数返回排序列表的原始索引。现在我想要原始列表的排序列表的索引。有没有一个函数或一种优雅的方式来做到这一点？

例如:

>>> import numpy as np
>>> a = np.array([2, 8 , 5, 6])
>>> np.argsort(a)
array([0, 2, 3, 1])
>>> function(a)
array([0, 3, 1, 2])

Answer 1

你总是可以调用 argsort 两次:

>>> a.argsort().argsort()
array([0, 3, 1, 2])

据我所知，NumPy 中没有可用的“double-argsort”函数，但是对数组应用 argsort 两次是计算值排名的常用方法(参见 here ).