Open In App

numpy.diagflat() in Python

Last Updated : 04 Aug, 2021
Comments
Improve
Suggest changes
Like Article
Like
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.

 
 


Next Article
numpy.diagflat() in Python

M

Mohit Gupta_OMG
Improve
Article Tags :
Practice Tags :

Similar Reads

We use cookies to ensure you have the best browsing experience on our website. By using our site, you acknowledge that you have read and understood our Cookie Policy & Privacy Policy
Lightbox