Numpy NP.diagramme

La fonction Numpy Diag () vous permet d'extraire ou de créer un nouveau tableau diagonal. C'est une fonction pratique mais simple.

Suivez ce tutoriel pour explorer cette fonction plus loin.

Syntaxe de fonction

La syntaxe de fonction est comme indiqué ci-dessous:

nombant.Diag (v, k = 0)

La fonction ne prend que deux paramètres, comme discuté ci-dessous:

Paramètres de fonction

1. V - le tableau d'entrée ou l'objet array_.
2. K - définit la diagonale à extraire.

Remarque: si la valeur de k est supérieure à 0, cela signifie la diagonale au-dessus de la diagonale principale. S'il est négatif, cela signifie la diagonale en dessous de la diagonale principale.

Valeur de retour de fonction

La fonction renvoie le réseau diagonal extrait ou un réseau diagonal nouvellement construit.

Exemple 1

Considérez l'exemple de code ci-dessous:

Importer Numpy comme NP
arr = np.Arange (6).Reshape (2,3)
imprimer (arr)
Imprimer (NP.diag (arr, k = 0))

Nous utilisons la fonction diag () pour extraire la diagonale principale du tableau fourni dans le code ci-dessus.

La sortie résultante est montrée:

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

Exemple 2

Pour extraire la diagonale au-dessus de la principale, définissez la valeur de K comme indiqué:

arr = np.Arange (6).Reshape (2,3)
print (f "Original: arr")
imprimer (f "Extrait: np.diag (arr, k = 1) ")

Cela revient:

Original: [[0 1 2]
[3 4 5]]
Extrait: [1 5]

Exemple 3

Si la valeur de K est négative, elle doit revenir:

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

Sortir:

Original: [[0 1 2]
[3 4 5]]
Extrait: [3]

Conclusion

Vous connaissez maintenant la fonction Diag dans Numpy et comment l'utiliser pour extraire ou construire un nouveau tableau diagonal.

Merci d'avoir lu!!