獲取排序的Numpy數組的原始索引

Question

我有一個距離爲a = np.array([20.5 ,5.3 ,60.7 ,3.0 ], 'double')的陣列，我需要排序數組的索引（例如[3, 1, 0, 2]，對於a.sort()）。 Numpy有沒有這樣的功能？

Answer 1

是的，有x = numpy.argsort(a)函數或x = numpy.ndarray.argsort(a)方法。它完全符合你的要求。您也可以撥打argsort作爲ndarray對象的方法，如下所示：a.argsort()。

下面是對文檔的鏈接：http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html#numpy.argsort

Answer 2

下面是一個例子，以供參考和方便：

# create an array 
a = np.array([5,2,3]) 

# np.sort - returns the array, sorted 
np.sort(a) 
>>> array([2, 3, 5]) 

# argsort - returns the original indexes of the sorted array 
np.argsort(a) 
>>> array([1, 2, 0])