NumPy - Funciones de clasificación, búsqueda y conteo

Una variedad de funciones relacionadas con la clasificación están disponibles en NumPy. Estas funciones de clasificación implementan diferentes algoritmos de clasificación, cada uno de ellos caracterizado por la velocidad de ejecución, el peor rendimiento del caso, el espacio de trabajo requerido y la estabilidad de los algoritmos. La siguiente tabla muestra la comparación de tres algoritmos de clasificación.

tipo velocidad peor de los casos espacio de trabajo estable
'ordenación rápida' 1 O (n ^ 2) 0 No
'mergesort' 2 O (n * log (n)) ~ n / 2 si
'heapsort' 3 O (n * log (n)) 0 No

numpy.sort ()

La función sort () devuelve una copia ordenada de la matriz de entrada. Tiene los siguientes parámetros:

numpy.sort(a, axis, kind, order)

Dónde,

No Señor. Descripción de parámetros
1

a

Matriz a ordenar

2

axis

El eje a lo largo del cual se ordenará la matriz. Si no hay ninguno, la matriz se aplana, ordenando en el último eje

3

kind

El valor predeterminado es ordenación rápida

4

order

Si la matriz contiene campos, el orden de los campos a ordenar

Ejemplo

import numpy as np  
a = np.array([[3,7],[9,1]]) 

print 'Our array is:' 
print a 
print '\n'

print 'Applying sort() function:' 
print np.sort(a) 
print '\n' 
  
print 'Sort along axis 0:' 
print np.sort(a, axis = 0) 
print '\n'  

# Order parameter in sort function 
dt = np.dtype([('name', 'S10'),('age', int)]) 
a = np.array([("raju",21),("anil",25),("ravi", 17), ("amar",27)], dtype = dt) 

print 'Our array is:' 
print a 
print '\n'  

print 'Order by name:' 
print np.sort(a, order = 'name')

Producirá la siguiente salida:

Our array is:
[[3 7]
 [9 1]]

Applying sort() function:
[[3 7]
 [1 9]]

Sort along axis 0:
[[3 1]
 [9 7]]

Our array is:
[('raju', 21) ('anil', 25) ('ravi', 17) ('amar', 27)]

Order by name:
[('amar', 27) ('anil', 25) ('raju', 21) ('ravi', 17)]

numpy.argsort ()

los numpy.argsort()La función realiza una ordenación indirecta en la matriz de entrada, a lo largo del eje dado y utilizando un tipo específico de ordenación para devolver la matriz de índices de datos. Esta matriz de índices se utiliza para construir la matriz ordenada.

Ejemplo

import numpy as np 
x = np.array([3, 1, 2]) 

print 'Our array is:' 
print x 
print '\n'  

print 'Applying argsort() to x:' 
y = np.argsort(x) 
print y 
print '\n'  

print 'Reconstruct original array in sorted order:' 
print x[y] 
print '\n'  

print 'Reconstruct the original array using loop:' 
for i in y: 
   print x[i],

Producirá la siguiente salida:

Our array is:
[3 1 2]

Applying argsort() to x:
[1 2 0]

Reconstruct original array in sorted order:
[1 2 3]

Reconstruct the original array using loop:
1 2 3

numpy.lexsort ()

La función realiza una clasificación indirecta mediante una secuencia de teclas. Las claves se pueden ver como una columna en una hoja de cálculo. La función devuelve una matriz de índices, mediante los cuales se pueden obtener los datos ordenados. Tenga en cuenta que la última clave resulta ser la clave principal de clasificación.

Ejemplo

import numpy as np 

nm = ('raju','anil','ravi','amar') 
dv = ('f.y.', 's.y.', 's.y.', 'f.y.') 
ind = np.lexsort((dv,nm)) 

print 'Applying lexsort() function:' 
print ind 
print '\n'  

print 'Use this index to get sorted data:' 
print [nm[i] + ", " + dv[i] for i in ind]

Producirá la siguiente salida:

Applying lexsort() function:
[3 1 0 2]

Use this index to get sorted data:
['amar, f.y.', 'anil, s.y.', 'raju, f.y.', 'ravi, s.y.']

El módulo NumPy tiene una serie de funciones para buscar dentro de una matriz. Están disponibles las funciones para encontrar el máximo, el mínimo y los elementos que satisfacen una condición determinada.

numpy.argmax () y numpy.argmin ()

Estas dos funciones devuelven los índices de elementos máximos y mínimos respectivamente a lo largo del eje dado.

Ejemplo

import numpy as np 
a = np.array([[30,40,70],[80,20,10],[50,90,60]]) 

print 'Our array is:' 
print a 
print '\n' 

print 'Applying argmax() function:' 
print np.argmax(a) 
print '\n'  

print 'Index of maximum number in flattened array' 
print a.flatten() 
print '\n'  

print 'Array containing indices of maximum along axis 0:' 
maxindex = np.argmax(a, axis = 0) 
print maxindex 
print '\n'  

print 'Array containing indices of maximum along axis 1:' 
maxindex = np.argmax(a, axis = 1) 
print maxindex 
print '\n'  

print 'Applying argmin() function:' 
minindex = np.argmin(a) 
print minindex 
print '\n'  
   
print 'Flattened array:' 
print a.flatten()[minindex] 
print '\n'  

print 'Flattened array along axis 0:' 
minindex = np.argmin(a, axis = 0) 
print minindex
print '\n'

print 'Flattened array along axis 1:' 
minindex = np.argmin(a, axis = 1) 
print minindex

Producirá la siguiente salida:

Our array is:
[[30 40 70]
 [80 20 10]
 [50 90 60]]

Applying argmax() function:
7

Index of maximum number in flattened array
[30 40 70 80 20 10 50 90 60]

Array containing indices of maximum along axis 0:
[1 2 0]

Array containing indices of maximum along axis 1:
[2 0 1]

Applying argmin() function:
5

Flattened array:
10

Flattened array along axis 0:
[0 1 1]

Flattened array along axis 1:
[0 2 0]

numpy.nonzero ()

los numpy.nonzero() La función devuelve los índices de elementos distintos de cero en la matriz de entrada.

Ejemplo

import numpy as np 
a = np.array([[30,40,0],[0,20,10],[50,0,60]]) 

print 'Our array is:' 
print a 
print '\n'  

print 'Applying nonzero() function:' 
print np.nonzero (a)

Producirá la siguiente salida:

Our array is:
[[30 40 0]
 [ 0 20 10]
 [50 0 60]]

Applying nonzero() function:
(array([0, 0, 1, 1, 2, 2]), array([0, 1, 1, 2, 0, 2]))

numpy.where ()

La función where () devuelve los índices de los elementos en una matriz de entrada donde se cumple la condición dada.

Ejemplo

import numpy as np 
x = np.arange(9.).reshape(3, 3) 

print 'Our array is:' 
print x  

print 'Indices of elements > 3' 
y = np.where(x > 3) 
print y  

print 'Use these indices to get elements satisfying the condition' 
print x[y]

Producirá la siguiente salida:

Our array is:
[[ 0. 1. 2.]
 [ 3. 4. 5.]
 [ 6. 7. 8.]]

Indices of elements > 3
(array([1, 1, 2, 2, 2]), array([1, 2, 0, 1, 2]))

Use these indices to get elements satisfying the condition
[ 4. 5. 6. 7. 8.]

numpy.extract ()

los extract() La función devuelve los elementos que satisfacen cualquier condición.

import numpy as np 
x = np.arange(9.).reshape(3, 3) 

print 'Our array is:' 
print x  

# define a condition 
condition = np.mod(x,2) == 0 

print 'Element-wise value of condition' 
print condition  

print 'Extract elements using condition' 
print np.extract(condition, x)

Producirá la siguiente salida:

Our array is:
[[ 0. 1. 2.]
 [ 3. 4. 5.]
 [ 6. 7. 8.]]

Element-wise value of condition
[[ True False True]
 [False True False]
 [ True False True]]

Extract elements using condition
[ 0. 2. 4. 6. 8.]