Extragerea și Construirea Diagonalelor cu np.diag în NumPy

Funcția versatilă np.diag() are un comportament dublu intuitiv:

• Dacă primește un vector 1D, construiește o matrice pătratică având elementele vectorului pe diagonala principală și zerouri în rest.
• Dacă primește o matrice 2D, extrage elementele de pe diagonala principală sub forma unui vector 1D.

import numpy as np

# 1. Crearea unei matrice diagonale:
mat_diag = np.diag([5, 10, 15])
print("Matrice diagonală:
", mat_diag)

# 2. Extragerea diagonalei dintr-o matrice oarecare:
A = np.array([[1, 2], [3, 4]])
print("Diagonala extrasă:", np.diag(A))