Numpy NP.argmin

Numpy NP.argmin
O pacote Python Numpy nos fornece a função argmin (), que nos permite obter o índice do elemento min em uma matriz em um eixo específico.

Vamos discutir.

Sintaxe da função numpy argmin

A função fornece uma sintaxe minimalista como mostrado:

Numpy.argmin (a, eixo = nenhum, out = Nenhum, *, keepDims =)

Parâmetros

Os parâmetros da função são os mostrados abaixo:

  1. a - refere -se à matriz de entrada. Este é um parâmetro não opcional.
  2. Eixo - Especifica ao longo de qual eixo aplicar a função argmin (). Se definido como nenhum, a função achatará a matriz e usará a função em todos os elementos.
  3. OUT - Especifica uma matriz de saída alternativa. A matriz de saída deve ter a mesma forma que o valor de saída.
  4. KeepDims - um valor booleano que permite preservar os eixos reduzidos no resultado como dimensões com um tamanho de um.

Resultado da função

A função retornará uma variedade de índices com a mesma forma que um.forma e as dimensões ao longo do eixo especificado removido.

Exemplo 1

A seguir, é apresentado um exemplo que ilustra como usar a função argmin () com uma matriz 1D em Python.

# importar numpy
importar numpy como np
arr = np.matriz ([1,2,3,4,5,6,7,8])
imprimir (f "Índice de Min Element -> np.argmin (arr) ")

No código acima, temos uma matriz 1D segurando elementos de 1 a 8. Em seguida, verificamos o elemento mínimo na matriz usando a função argmin () e retornamos seu índice.

A saída é como mostrado:

ÍNDICE DE MIN ELEMENTO -> 0

Exemplo 2

Vamos ver o que acontece ao aplicar a mesma operação em uma matriz 2D.

arr_2d = np.Array ([[[1,2,3,4], [5,6,7,8]]]))
imprimir (f "Índice de Min Element -> np.argmin (arr_2d) ")

No código acima, aplicamos a função argmin () a uma matriz 2D sem especificar o eixo. Isso nutre a matriz e aplica a função.

O valor resultante é como mostrado:

ÍNDICE DE MIN ELEMENTO -> 0

Exemplo 3

Para agir ao longo de um eixo específico, podemos definir o parâmetro do eixo como mostrado:

arr_2d = np.Array ([[[1,2,3,4], [5,6,7,8]]]))
Imprimir (f "índices de elementos min -> np.argmin (arr_2d, eixo = 0) ")

O código acima deve aplicar a função argmin () ao longo do eixo 0 e retornar os índices dos elementos Min, como mostrado na matriz de saída:

Índices de elementos Min -> [[0 0 0]
[0 0 0]]

Exemplo 4

Para aplicar a função no último eixo, podemos definir o valor do eixo como -1 como mostrado abaixo:

arr_2d = np.Array ([[[1,2,3,4], [5,6,7,8]]]))
Imprimir (f "índices de elementos min -> np.argmin (arr_2d, eixo = -1) ")

O código acima deve retornar:

Índices de elementos Min -> [[0 0]]

Conclusão

Ao longo deste artigo, exploramos a função Numpy argmin, sua sintaxe, parâmetros e valores de retorno. Também fornecemos vários exemplos ilustrando como a função funciona em vários cenários.

Codificação feliz!!