如何在 Python 中获取排序数组的索引

Question

我有一个数字列表：

 myList = [1, 2, 3, 100, 5]

现在，如果我对该列表进行排序以获得 [1, 2, 3, 5, 100] 。我想要的是原始列表中按排序顺序排列的元素的索引，即 [0, 1, 2, 4, 3] ala MATLAB 的排序函数返回值和索引。

原文由 Gyan 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

如果您使用的是 numpy，则可以使用 argsort() 函数：

 >>> import numpy
>>> numpy.argsort(myList)
array([0, 1, 2, 4, 3])

http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html

这将返回对数组或列表进行排序的参数。

原文由 Matthew Lewis 发布，翻译遵循 CC BY-SA 3.0 许可协议