Numpy np.diag

Numpy np.diag
La funzione Numpy Diag () consente di estrarre o creare un nuovo array diagonale. È una funzione conveniente ma semplice.

Segui insieme questo tutorial per esplorare ulteriormente questa funzione.

Sintassi della funzione

La sintassi della funzione è come mostrato di seguito:

numpy.Diag (V, K = 0)

La funzione richiede solo due parametri, come discusso di seguito:

Parametri della funzione

  1. V - L'array di input o l'array_like oggetto.
  2. K - definisce la diagonale da estrarre.

Nota: se il valore di k è maggiore di 0, significa diagonale sopra la diagonale principale. Se è negativo, significa diagonale al di sotto della diagonale principale.

Valore di ritorno della funzione

La funzione restituisce l'array diagonale estratto o un array diagonale di nuova costruzione.

Esempio 1

Considera il codice di esempio mostrato di seguito:

Importa Numpy come NP
arr = np.Arange (6).Reshape (2,3)
Stampa (ARR)
Stampa (NP.diag (arr, k = 0))

Usiamo la funzione Diag () per estrarre la diagonale principale dall'array fornito nel codice sopra.

Viene mostrato l'output risultante:

[[0 1 2]
[3 4 5]]
--> estratto: [0 4]

Esempio 2

Per estrarre la diagonale sopra il principale, impostare il valore di K come mostrato:

arr = np.Arange (6).Reshape (2,3)
print (f "originale: arr")
print (f "estratto: np.diag (arr, k = 1) ")

Questo ritorna:

Originale: [[0 1 2]
[3 4 5]]
Estratto: [1 5]

Esempio 3

Se il valore di K è negativo, dovrebbe restituire:

arr = np.Arange (6).Reshape (2,3)
print (f "originale: arr")
print (f "estratto: np.diag (arr, k = -1) ")

Produzione:

Originale: [[0 1 2]
[3 4 5]]
estratto: [3]

Conclusione

Ora hai familiarità con la funzione Diag in Numpy e come usarla per estrarre o costruire un nuovo array diagonale.

Grazie per aver letto!!