Obtenga los índices originales de una matriz Numpy ordenada

Question

Tengo una matriz de distancias a = np.array([20.5 ,5.3 ,60.7 ,3.0 ], 'double') y necesito los índices de la matriz ordenada (por ejemplo [3, 1, 0, 2], para a.sort()). ¿Hay una función en Numpy para hacer eso?

Answer 1

Sí, existe la función x = numpy.argsort(a) o el método x = numpy.ndarray.argsort(a). Hace exactamente lo que estás pidiendo. También puede llamar al argsort como un método en un objeto ndarray como lo siguiente: a.argsort().

Aquí hay un enlace a la documentación: http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html#numpy.argsort

Answer 2

He aquí un ejemplo, para referencia y conveniencia:

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