Python numpy.argmax()

1. Sintaxe de numpy.argmax():
2. Códigos de exemplo: numpy.argmax() Método para encontrar índices de valores mais elevados num array
3. Códigos de exemplo: Definir o parâmetro axis em numpy.argmax()Método para Encontrar Índices de Valores Maiores num Array
4. Códigos de exemplo: Definir out Parâmetro em numpy.argmax() Método para encontrar índices dos valores mais altos de um array

Python Numpy numpy.argmax() retorna os índices de valores com os valores mais altos no dado array NumPy.

Sintaxe de numpy.argmax():

numpy.argmax(a, axis=None, out=None)

Parâmetros

Devolver

Um array de índices de elementos com o maior valor em todo o array.

Códigos de exemplo: numpy.argmax() Método para encontrar índices de valores mais elevados num array

Encontrar o Índice do Valor Mais Elevado num Array 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])

Resultado:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Fornece o índice do elemento com o maior valor na array de entrada dada.

Podemos também encontrar o maior valor no array utilizando o índice devolvido pelo método np.argmax().

Se tivermos dois valores maiores no array, o método devolve o índice do elemento maior que aparece primeiro no 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])

Resultado:

Array:
[2 6 1 5]

Index with the largest value:
1

The largest value in the array:
6

Aqui, 6 é o valor mais alto que ocorre duas vezes no array, mas o np.argmax() devolve o índice de 6 no índice 1 uma vez que vem em primeiro lugar no array.

Encontrar o índice do valor mais alto de um 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)

Resultado:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Aqui, como o parâmetro “eixo” não é dado, a matriz é achatada e o índice do maior elemento do array achatada é devolvido.

Códigos de exemplo: Definir o parâmetro axis em numpy.argmax()Método para Encontrar Índices de Valores Maiores num Array

Encontrar Índices do Elemento Mais Elevado ao longo do Eixo da Coluna

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)

Resultado:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
[1 1 0]

Se definirmos axis=0, dá índices do valor mais alto ao longo de cada coluna. Aqui a primeira coluna tem o valor mais alto no índice 1, a segunda coluna tem o valor mais alto no índice 1 e a terceira coluna tem o valor mais alto no índice 0.

Encontrar Índices dos Elementos Mais Altos ao longo do Eixo da Fileira

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)

Resultado:

Array:
[[2 1 6]
 [7 4 5]]

Indices with the largest value along the row axis:
[2 0]

Se definirmos axis=1, dá índices dos valores mais altos ao longo de cada linha.

Códigos de exemplo: Definir out Parâmetro em numpy.argmax() Método para encontrar índices dos valores mais altos de um 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)

Resultado:

Array:
[[2 1 6]
 [7 4 5]]

Index with the largest value:
3

Aqui, a variável req_index actua como um espaço reservado para o valor requerido do índice e temos de garantir que o tamanho da saída deve ser igual a req_index.