numpy.diag() in Python
numpy.diag() in Python
The diag() function of Python numpy class extracts and construct a diagonal array.
Syntax
numpy.diag(v, k=0)
Parameter
a: It represents the array_like.
k: It represents the diagonal value that we require. It is an optional parameter and its default value is 0. If k>0, the diagonal is above the main diagonal or vice versa.
Return
This function returns the extracted diagonal or constructed diagonal array.
Example 1
# Python Programming explaining # numpy.diag() function import numpy as np # matrix creation by array input num = np.matrix([[11, 121, 130], [613 ,34, 13], [514, 50, 16]]) print("Main Diagnol elements : \n", np.diag(num)) print("\nDiagnol elements above main diagnol : \n", np.diag(num, 1) ) print("\nDiagnol elements below main diagnol : \n", np.diag(num, -1))
Output
Main Diagnol elements : [11 34 16] Diagnol elements above main diagnol : [121 13] Diagnol elements below main diagnol : [613 50]