Saltar al contenido

numpy.argmax () en Python – Javatpoint

octubre 16, 2021
numpy

En muchos casos, cuando el tamaño de la matriz es demasiado grande, se necesita demasiado tiempo para encontrar el máximo de elementos de ellos. Para este propósito, el módulo numpy de Python proporciona una función llamada numpy.argmax (). Esta función devuelve índices de los valores máximos que se devuelven junto con el eje especificado.

numpy argmax

Sintaxis:

Parámetros

x: array_like

Este parámetro define la matriz de origen cuyo valor máximo queremos conocer.

eje: int (opcional)

Este parámetro define el eje a lo largo del cual está presente el índice y, por defecto, está en la matriz aplanada.

out: array (opcional)

Este parámetro define el ndarray en el que se insertará el resultado. Este será del mismo tipo y forma, lo que es apropiado para almacenar el resultado.

Devoluciones

Este parámetro define un ndarray, que contiene los índices de la matriz. La forma es la misma que x.shape con la cota a lo largo del eje eliminada.

Ejemplo 1:

Producción:

array([[ 7,  8,  9, 10, 11],
       	[12, 13, 14, 15, 16],
       	[17, 18, 19, 20, 21],
       	[22, 23, 24, 25, 26]])
19

En el código anterior

  • Hemos importado numpy con el nombre de alias np.
  • Hemos creado una matriz ‘X’ utilizando np.arange () función con la forma de cuatro filas y cinco columnas.
  • También hemos agregado 7 en cada elemento de la matriz.
  • Hemos declarado la variable ‘y’ y se le asignó el valor devuelto de np.argmax () función.
  • Hemos pasado la matriz ‘X’ en la función.
  • Por último, intentamos imprimir el valor de ‘y’.

En la salida, muestra los índices del elemento máximo en la matriz.

Ejemplo 2:

Producción:

array([3, 3, 3, 3, 3], dtype=int64)
array([4, 4, 4, 4], dtype=int64)

Ejemplo 3:

Producción:

Ejemplo 4:

Producción:

array([[0],
       	[2],
       	[2]])
array([5, 9, 6])

En el código anterior

  • Hemos importado numpy con el nombre de alias np.
  • Hemos creado una matriz multidimensional. ‘a‘ utilizando np.array () función.
  • Hemos declarado la variable ‘index_arr’ y se le asignó el valor devuelto de np.argmax () función.
  • Hemos pasado la matriz ‘a’ y el eje en la función.
  • Intentamos imprimir el valor de ‘index_arr’.
  • Al final, intentamos obtener el valor máximo de la matriz con la ayuda de dos formas diferentes, que son bastante similares a np.argmax ().

En la salida, muestra índices de los elementos máximos en la matriz y los valores que están presentes en esos índices.


close