Numpy NP.argmax

A função aceita os seguintes parâmetros:

• Array - refere -se à matriz de entrada na qual a função é aplicada.
• Eixo -especifica ao longo de qual eixo da matriz a função é aplicada -este é um parâmetro de tipo inteiro opcional. A função achatará a matriz se o valor do eixo não for definido.
• Fora - especifica uma matriz na qual o resultado é inserido. A matriz especificada deve ser do tipo e forma adequados.
• KeepDims - este é um parâmetro booleano que leva verdadeiro ou falso. Se definido como true, o eixo reduzido é deixado como dimensões com o tamanho um.

Valor de retorno da função

A função retorna uma matriz de índices para a matriz com a mesma forma que a matriz.forma.

Exemplo

Vamos ilustrar como usar o Numpy.função argmax com um exemplo.

Comece importando Numpy como:

Em seguida, crie uma matriz 2D como mostrado:

O acima deve criar uma matriz 2D com os elementos como mostrado:

Para usar a função argmax () em toda a matriz (achatada), podemos executar o código como mostrado:

O código acima retorna o índice máximo do valor máximo, como mostrado:

Você também pode usar a função Argmax ao longo de um eixo específico, como mostrado:

Nota: Como estamos trabalhando com uma matriz 2D, a matriz tem dois eixos possíveis. O primeiro vá verticalmente através das linhas como eixo 0, e o outro corre horizontalmente através das colunas como eixo 1.
Portanto, o código acima deve retornar:

O mesmo caso se aplica ao longo do eixo 1. Um exemplo é como mostrado:

Isso deve retornar:

Exemplo 2

Também podemos usar a função argmax com uma matriz de dimensão n. Por exemplo, o código abaixo ilustra como usar a função com uma matriz 3D.

Isso deve gerar uma matriz como:

Ao longo dos eixos, podemos obter o Argmax como mostrado:

O código acima deve retornar:

Fechamento

Este artigo explora como usar a função Numpy argmax () para buscar os índices dos valores max, juntamente com um índice específico.

Obrigado pela leitura e fique atento!!