numpy.triu() in Python
Last Updated :
09 Mar, 2022
numpy.triu(a, k = 0) : Returns copy of array with upper part of the triangle w.r.t k
Parameters :
a : input array
k : [int, optional, 0 by default]
Diagonal we require; k>0 means diagonal above main diagonal or vice versa.
Returns :
Upper triangle of a, having same shape and data-type as a.
import numpy as geek
a = geek.matrix([[ 1 , 21 , 30 ],
[ 63 , 434 , 3 ],
[ 54 , 54 , 56 ]])
print ( "Main Diagonal elements : \n" , geek. triu(a), "\n" )
print ( "Diagonal above main Diagonal elements : \n" , geek. triu(a, 1 ), "\n\n" )
print ( "Main Diagonal elements : \n" , geek. triu(a, - 1 ))
|
Output :
Main Diagonal elements :
[[ 1 21 30]
[ 0 434 3]
[ 0 0 56]]
Diagonal above main Diagonal elements :
[[ 0 21 30]
[ 0 0 3]
[ 0 0 0]]
Main Diagonal elements :
[[ 1 21 30]
[ 63 434 3]
[ 0 54 56]]
Reference :
https://docs.scipy.org/doc/numpy/reference/generated/numpy.triu.html#numpy.triu
Note :
These NumPy-Python programs won’t run on online IDE’s, so run them on your systems to explore them.
Share your thoughts in the comments
Please Login to comment...