python - 从排序列表中提取索引

Question

假设您有一个未排序的列表，并使用np.sort对其进行排序。有没有办法使用 numpy 从原始列表中获取排序列表的索引？

Answer 1

最简单的方法是使用位置索引来扩充数组，然后对二维数组进行排序。这会同时为您提供排序后的数据及其原始位置索引。

如果您只想要索引(而不是排序数据)，请使用argsort:

>>> from numpy import array
>>> arr = array([10, 5, 80, 20, 70, 18])
>>> arr.argsort()
array([1, 0, 5, 3, 4, 2])