Open In App

numpy.diagflat() in Python

Improve
Improve
Like Article
Like
Save
Share
Report

numpy.diagflat (a, k = 0): Create a two-dimensional array with the array_like input as a diagonal to the new output array.

Parameters : 

a : array_like input data with diagonal  elements
strong>k : [int, optional, 0 by default]
          Diagonal we require; k>0 means diagonal  above main diagonal  or vice versa.

Returns :  

array with the array_like input as a diagonal to the new output array.

Python




# Python Program illustrating
# numpy.diagflat method
 
import numpy as geek
 
print("diagflat use on main diagonal : \n", geek.diagflat([1, 7]), "\n")
 
print("diagflat use on main diagonal : \n", geek.diagflat([1, 7, 6]), "\n")
 
# Diagonal above main diagonal
print("diagflat above main diagonal : \n", geek.diagflat([1, 7, 6], 1), "\n")


Output : 

diagflat use on main diagonal  : 
 [[1 0]
 [0 7]] 

diagflat use on main diagonal  : 
 [[1 0 0]
 [0 7 0]
 [0 0 6]] 

diagflat above main diagonal  : 
 [[0 1 0 0]
 [0 0 7 0]
 [0 0 0 6]
 [0 0 0 0]] 

Note : 
These NumPy-Python programs won’t run on onlineID, so run them on your systems to explore them.

 

 


Last Updated : 04 Aug, 2021
Like Article
Save Article
Previous
Next
Share your thoughts in the comments
Similar Reads