numpy.diag(v, k=0)
kann die Diagonale eines Arrays extrahieren oder kann ein Diagonalarray konstruieren.
Falls v ein 2-D-Array ist, liefert die Funktion eine Kopie der k-ten Diagonale zurück.
>>> x = np.array([[11, 12, 13], [21, 22, 23], [31, 32, 33]])
>>> np.diag(x)
array([11, 22, 33])
>>> np.diag(x, k=1)
array([12, 23])
>>> np.diag(x, k=-1)
array([21, 32])
Falls für v ein 1-D-Array übergeben wird, liefert sie ein 2-D-Array mit v auf der k-ten Diagonale zurück.
>>> np.diag([3,4,5])
array([[3, 0, 0],
[0, 4, 0],
[0, 0, 5]])
>>> np.diag([3,4,5], k=1)
array([[0, 3, 0, 0],
[0, 0, 4, 0],
[0, 0, 0, 5],
[0, 0, 0, 0]])