NP NP.argmax
Numpy le proporciona la función ArgMax () que le permite obtener el índice del elemento MAX dentro de una matriz en un eje específico.
Este tutorial intenta explicar cómo el Numpy.La función argMax () funciona y cómo usarla.
Sintaxis de funciones
La sintaxis de la función es como se muestra a continuación:
numpy.ArgMax (Array, Axis = None, Out = None, *, KeepDims =La función acepta los siguientes parámetros:
- Array: se refiere a la matriz de entrada en la que se aplica la función.
- El eje -especifica a lo largo de qué eje de la matriz se aplica la función: este es un parámetro de tipo entero opcional. La función aplanará la matriz si el valor del eje no está configurado.
- Out: especifica una matriz en la que se inserta el resultado. La matriz especificada debe ser del tipo y la forma adecuados.
- Keepdims: este es un parámetro booleano que toma verdadero o falso. Si se establece en True, el eje reducido se deja como dimensiones con el tamaño uno.
Valor de retorno de la función
La función devuelve una matriz de índices en la matriz con la misma forma que la matriz.forma.
Ejemplo
Ilustramos cómo usar el Numpy.Función ArgMax con un ejemplo.
Comience por importar Numpy como:
# importy numpyimportar numpy como np
A continuación, cree una matriz 2D como se muestra:
# Matriz 2dmatriz = np.Arange (6).remodelar ((2,3))
Imprimir (matriz)
Lo anterior debe crear una matriz 2D con los elementos como se muestra:
[[0 1 2][3 4 5]]
Para usar la función ArgMax () en toda la matriz (aplanada), podemos ejecutar el código como se muestra:
# elemento máximoimprimir (f "valor máximo: np.argMax (array, axis = none) ");
El código anterior devuelve el índice máximo del valor máximo como se muestra:
También puede usar la función ArgMax a lo largo de un eje específico como se muestra:
# elemento máximo a lo largo de 0 ejeImprimir (NP.argMax (matriz, axis = 0));
Nota: Dado que estamos trabajando con una matriz 2D, la matriz tiene dos ejes posibles. El primero va verticalmente a través de las filas como el eje 0, y el otro corre horizontalmente a través de las columnas como el eje 1.
Por lo tanto, el código anterior debe devolver:
El mismo caso se aplica a lo largo del eje 1. Un ejemplo es como se muestra:
# elemento máximo a lo largo del eje 1Imprimir (NP.ArgMax (Array, eje = 1))
Esto debería volver:
[2 2]Ejemplo 2
También podemos usar la función ArgMax con una matriz N-Dimension. Por ejemplo, el siguiente código ilustra cómo usar la función con una matriz 3-D.
# Matriz 3Dmatriz = np.Arange (24).remodelar (2, 3, 4)
Imprimir (matriz)
Esto debería generar una matriz como:
[[[0 1 2 3][4 5 6 7]
[8 9 10 11]]
[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]
A lo largo de los ejes, podemos obtener el argmax como se muestra:
# elemento máximoimprimir (f "max: np.ArgMax (Array, Axis = None) ")
# elemento máximo a lo largo del eje 0
imprimir (f "índices max: np.ArgMax (Array, axis = 0) ")
# elemento máximo a lo largo del eje 1
imprimir (f "índices max: np.ArgMax (Array, axis = 1) ")
# elemento máximo a lo largo del eje 2
imprimir (f "índices max: np.ArgMax (Array, axis = 2) ")
El código anterior debería devolver:
Max: 23Índices máximos: [[1 1 1 1]
[1 1 1 1]
[1 1 1 1]]
Índices máximos: [[2 2 2 2]
[2 2 2 2]]
Índices máximos: [[3 3 3]
[3 3 3]]
Clausura
Este artículo explora cómo usar la función numpy argMax () para obtener los índices de los valores máximos junto con un índice específico.
Gracias por leer y estad atentos!!