Python numpy.argmax()

1. La sintaxis de numpy.argmax():
2. Códigos de ejemplo: numpy.argmax() Método para encontrar índices de los valores más altos en un array
3. 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)
4. 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)

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

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.