numpy.diagflat() in Python
The diagflat() function of Python numpy class creates a two-dimensional array with the array_like input as a diagonal to the new output array.
Syntax
|
1 2 3 |
numpy.diagflat (a, k = 0) |
Parameter
a : It represents the array_like input data with diagonal elements
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 2-D output array as a diagonal to the new array(ndarray).
Example 1
|
1 2 3 4 5 6 7 8 9 10 |
# Python Programming explaining # numpy.diagflat() function import numpy as np print("diagflat use on main diagonal : \n", np.diagflat([14, 71])) print("diagflat use on main diagonal : \n", np.diagflat([15, 27, 46])) # Diagnol above main diagnol print("diagflat above main diagonal : \n", np.diagflat([51, 67, 69], 1)) print("diagflat above main diagonal : \n", np.diagflat([51, 67, 69], 2)) |
Output
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
diagflat use on main diagonal : [[14 0] [ 0 71]] diagflat use on main diagonal : [[15 0 0] [ 0 27 0] [ 0 0 46]] diagflat above main diagonal : [[ 0 51 0 0] [ 0 0 67 0] [ 0 0 0 69] [ 0 0 0 0]] diagflat above main diagonal : [[ 0 0 51 0 0] [ 0 0 0 67 0] [ 0 0 0 0 69] [ 0 0 0 0 0] [ 0 0 0 0 0]] |