Python numpy.triu()用法及代码示例

numpy.triu(a，k = 0)：返回三角形上部为w.r.t k的数组的副本
参数：

a : input array
k : [int, optional, 0 by default]
          Diagonal we require; k>0 means diagonal above main diagonal or vice versa.

返回值：

Upper triangle of a, having same shape and data-type as a.

# Python Programming illustrating 
# numpy. triu method 
  
import numpy as geek 
  
# string input 
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))

输出：

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]]