Python numpy.argmax()
-
La sintaxis de
numpy.argmax()
: -
Códigos de ejemplo:
numpy.argmax()
Método para encontrar índices de los valores más altos en un array -
Códigos de ejemplo: Establecer el parámetro
axis
ennumpy.argmax
(Método para encontrar índices de los valores más grandes en un array) -
Códigos de ejemplo: Establece el parámetro
out
ennumpy.argmax
(Método para encontrar los índices de los valores más altos en un array)
![Python numpy.argmax()](/img/Numpy/feature-image---Python-Numpy-argmax.webp)
La función Python Numpy numpy.argmax()
devuelve los índices de valores con los valores más altos de el array NumPy dada.
La sintaxis de numpy.argmax()
:
numpy.argmax(a, axis=None, out=None)
Parámetros
a |
Matriz u objeto que puede ser convertido en matriz en la que necesitamos encontrar índices de valores más altos. |
axis |
encontrar el índice del mayor valor a lo largo de la fila (axis=0) o columna (axis=1). Por defecto, el índice del mayor valor se encuentra aplanando el array. |
out |
marcador de posición para el resultado del método np.argmax . Debe ser del tamaño apropiado para guardar el resultado. |
Retorna
Un conjunto de índices de elementos con el valor más alto de todo el conjunto.
Códigos de ejemplo: numpy.argmax()
Método para encontrar índices de los valores más altos en un array
Encuentra el índice del valor más alto en un conjunto de 1-D
import numpy as np
a=np.array([2,6,1,5])
print("Array:")
print(a)
req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)
print("\nThe largest value in the array:")
print(a[req_index])
Producción:
Array:
[2 6 1 5]
Index with the largest value:
1
The largest value in the array:
6
Da el índice del elemento con el mayor valor en el array de entrada dada.
También podemos encontrar el mayor valor de el array usando el índice devuelto por el método np.argmax()
.
Si tenemos dos valores mayores en el array, el método devuelve el índice del elemento más grande que aparece primero en el array.
import numpy as np
a=np.array([2,6,1,6])
print("Array:")
print(a)
req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)
print("\nThe largest value in the array:")
print(a[req_index])
Producción:
Array:
[2 6 1 5]
Index with the largest value:
1
The largest value in the array:
6
Aquí, 6
es el valor más alto que aparece dos veces en el array, pero el np.argmax()
devuelve el índice de 6
en el índice 1
ya que aparece primero en el array.
Encuentra el índice del valor más alto en un array 2-D
import numpy as np
a=np.array([[2,1,6],
[7,4,5]])
print("Array:")
print(a)
req_index=np.argmax(a)
print("\nIndex with the largest value:")
print(req_index)
Producción:
Array:
[[2 1 6]
[7 4 5]]
Index with the largest value:
3
Aquí, como no se da el parámetro axis
, el array se aplana y se devuelve el índice del elemento más grande de el array aplanada.
Códigos de ejemplo: Establecer el parámetro axis
en numpy.argmax
(Método para encontrar índices de los valores más grandes en un array)
Encuentra índices del elemento más alto a lo largo del eje de la columna
import numpy as np
a=np.array([[2,1,6],
[7,4,5]])
print("Array:")
print(a)
req_index=np.argmax(a,axis=0)
print("\nIndices with the largest value along column axis:")
print(req_index)
Producción:
Array:
[[2 1 6]
[7 4 5]]
Index with the largest value:
[1 1 0]
Si ponemos axis=0
, da índices del valor más alto a lo largo de cada columna. Aquí la primera columna tiene el valor más alto en el índice 1
, la segunda columna tiene el valor más alto en el índice 1
y la tercera columna tiene el valor más alto en el índice 0
.
Encontrar los índices de los elementos más altos a lo largo del eje de la fila
import numpy as np
a=np.array([[2,1,6],
[7,4,5]])
print("Array:")
print(a)
req_index=np.argmax(a,axis=1)
print("\nIndices with the largest value along row axis:")
print(req_index)
Producción:
Array:
[[2 1 6]
[7 4 5]]
Indices with the largest value along the row axis:
[2 0]
Si ponemos axis=1
, da índices de los valores más altos a lo largo de cada fila.
Códigos de ejemplo: Establece el parámetro out
en numpy.argmax
(Método para encontrar los índices de los valores más altos en un array)
import numpy as np
a=np.array([[2,1,6],
[7,4,5]])
req_index=np.array(0)
print("Array:")
print(a)
np.argmax(a,out=req_index)
print("\nIndex with the largest value:")
print(req_index)
Producción:
Array:
[[2 1 6]
[7 4 5]]
Index with the largest value:
3
Aquí, la variable req_index
actúa como un marcador de posición para el valor requerido del índice y debemos asegurarnos de que el tamaño de la salida debe ser igual a req_index
.
Suraj Joshi is a backend software engineer at Matrice.ai.
LinkedIn