从排序列表中提取索引

Question

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

Answer 1

最简单的方法是使用位置索引扩充数组，然后对2-D数组进行排序。这样可以同时为您提供排序数据和原始位置指示。

如果您只想要标记（不是排序数据），请使用 argsort ：

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