排序数组并返回排序数组的原始索引

Question

给定一个numpy数组，如何找到其中的索引序列以便对结果进行排序？在

例如，给定x=[4,2,6]，结果将是[1,0,2]，因为[x[1],x[0],x[2]]是排序的。在

我知道有很多可用的Python函数，比如argsort()可以完成这项工作，但是我需要自己实现这个排序函数。有什么建议吗？在

Answer 1

如果我理解您的问题，您可以使用list-comprehension，并使用sorted函数。在

>>> import numpy as np
>>> np_array = np.array([4, 2, 6])
>>> sorted_index_pos = [index for index, num in sorted(enumerate(np_array), key=lambda x: x[-1])]
[1, 0, 2]