Clasifique elementos en una matriz usando Python/NumPy, sin ordenar la matriz dos veces
Tengo una matriz de números y me gustaría crear otra matriz que represente la clasificación de cada elemento en la primera matriz. Estoy usando Python y NumPy.
Por ejemplo:
array = [4,2,7,1]
ranks = [2,1,3,0]
Este es el mejor método que se me ocurrió:
array = numpy.array([4,2,7,1])
temp = array.argsort()
ranks = numpy.arange(len(array))[temp.argsort()]
¿Existe algún método mejor/más rápido que evite ordenar la matriz dos veces?
Utilice argsort dos veces, primero para obtener el orden de la matriz y luego para obtener la clasificación:
array = numpy.array([4,2,7,1])
order = array.argsort()
ranks = order.argsort()
Cuando trabaje con matrices 2D (o de dimensiones superiores), asegúrese de pasar un argumento de eje a argsort para ordenar sobre el eje correcto.
Esta pregunta tiene algunos años y la respuesta aceptada es excelente, pero creo que aún vale la pena mencionar lo siguiente. Si no le importa depender de scipy, puede utilizar scipy.stats.rankdata:
In [22]: from scipy.stats import rankdata
In [23]: a = [4, 2, 7, 1]
In [24]: rankdata(a)
Out[24]: array([ 3., 2., 4., 1.])
In [25]: (rankdata(a) - 1).astype(int)
Out[25]: array([2, 1, 3, 0])
Una característica interesante rankdataes que el methodargumento proporciona varias opciones para manejar los vínculos. Por ejemplo, hay tres apariciones de 20 y dos apariciones de 40 en b:
In [26]: b = [40, 20, 70, 10, 20, 50, 30, 40, 20]
El valor predeterminado asigna la clasificación promedio a los valores empatados:
In [27]: rankdata(b)
Out[27]: array([ 6.5, 3. , 9. , 1. , 3. , 8. , 5. , 6.5, 3. ])
method='ordinal'asigna rangos consecutivos:
In [28]: rankdata(b, method='ordinal')
Out[28]: array([6, 2, 9, 1, 3, 8, 5, 7, 4])
method='min'asigna el rango mínimo de los valores empatados a todos los valores empatados:
In [29]: rankdata(b, method='min')
Out[29]: array([6, 2, 9, 1, 2, 8, 5, 6, 2])
Consulte la cadena de documentación para obtener más opciones.
