numpy.diagflat() in Python

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.