Numpy np.argmin

Numpy np.argmin
Il pacchetto Python Numpy ci fornisce la funzione Argmin (), che ci consente di ottenere l'indice dell'elemento Min in un array su un determinato asse.

Discutiamo.

Sintassi della funzione Numpy Argmin

La funzione fornisce una sintassi minimalistica come mostrato:

numpy.argmin (a, axis = nessuna, out = nessuno, *, keepDims =)

Parametri

I parametri della funzione sono come mostrato di seguito:

  1. A - si riferisce all'array di input. Questo è un parametro non opzionale.
  2. Asse - Specifica lungo quale asse per applicare la funzione Argmin (). Se impostato su nessuno, la funzione appiattirà l'array e utilizzerà la funzione su tutti gli elementi.
  3. Out - Specifica un array di output alternativo. L'array di output deve avere la stessa forma del valore di output.
  4. KeepDims - Un valore booleano che consente di preservare gli assi ridotti nel risultato come dimensioni con una dimensione di uno.

Risultato della funzione

La funzione restituirà un array di indici con la stessa forma di a.forma e le dimensioni lungo l'asse specificato rimosso.

Esempio 1

Quello che segue è un esempio che illustra come usare la funzione Argmin () con un array 1D in Python.

# Importa Numpy
Importa Numpy come NP
arr = np.Array ([1,2,3,4,5,6,7,8])
print (f "indice di min elemento -> np.argmin (arr) ")

Nel codice sopra, abbiamo un array 1D di detenzione di elementi da 1 a 8. Controlliamo quindi l'elemento minimo nell'array usando la funzione argmin () e restituiamo il suo indice.

L'output è come mostrato:

indice dell'elemento min -> 0

Esempio 2

Vediamo cosa succede quando si applica la stessa operazione su un array 2D.

arr_2d = np.array ([[[1,2,3,4], [5,6,7,8]]])
print (f "indice di min elemento -> np.argmin (arr_2d) ")

Nel codice sopra, applichiamo la funzione argmin () su un array 2D senza specificare l'asse. Questo appiattisce l'array e applica la funzione.

Il valore risultante è come mostrato:

indice dell'elemento min -> 0

Esempio 3

Per agire lungo un asse specifico, possiamo impostare il parametro dell'asse come mostrato:

arr_2d = np.array ([[[1,2,3,4], [5,6,7,8]]])
print (f "indici di Min Elements -> np.argmin (arr_2d, axis = 0) ")

Il codice sopra dovrebbe applicare la funzione argmin () lungo l'asse 0 e restituire gli indici degli elementi MIN come mostrato nell'array di output:

Indici di Min Elements -> [[0 0 0 0]
[0 0 0 0]]

Esempio 4

Per applicare la funzione sull'ultimo asse, possiamo impostare il valore dell'asse come -1 come mostrato di seguito:

arr_2d = np.array ([[[1,2,3,4], [5,6,7,8]]])
print (f "indici di Min Elements -> np.argmin (arr_2d, axis = -1) ")

Il codice sopra dovrebbe restituire:

Indici di Min Elements -> [[0 0]]

Conclusione

In questo articolo, abbiamo esplorato la funzione Numpy Argmin, la sua sintassi, i parametri e i valori di restituzione. Abbiamo anche fornito vari esempi che illustrano come funziona la funzione in più scenari.

Codice felice!!