numpy.diag() in Python
Last Updated :
09 Mar, 2022
numpy.diag(a, k=0) : Extracts and construct a diagonal array
Parameters :
a : array_like
k : [int, optional, 0 by default]
Diagonal we require; k>0 means diagonal above main diagonal or vice versa.
Returns :
ndarray
Python
import numpy as geek
a = geek.matrix([[ 1 , 21 , 30 ],
[ 63 , 434 , 3 ],
[ 54 , 54 , 56 ]])
print ( "Main Diagonal elements : \n" , geek.diag(a), "\n" )
print ( "Diagonal above main diagonal : \n" , geek.diag(a, 1 ), "\n" )
print ( "Diagonal below main diagonal : \n" , geek.diag(a, - 1 ))
|
Output :
Main Diagonal elements :
[ 1 434 56]
Diagonal above main diagonal :
[21 3]
Diagonal below main diagonal :
[63 54]
References :
https://docs.scipy.org/doc/numpy/reference/generated/numpy.diagflat.html#numpy.diagflat
Note :
These NumPy-Python programs won’t run on online IDE’s, so run them on your systems to explore them
.
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...