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])