numpy.diagflat() in Python (original) (raw)

Last Updated : 04 Aug, 2021

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

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" )

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.