Numpy中的一个函数用法:np.triu() & np.tril()

2021-01-04 10:11:30 浏览数 (1)

参考链接: Python中的numpy.triu

今天帮朋友看一个代码,刚好里边有一个函数不太明白,因此看了源码之后在此做以记录: 

#取上三角阵

def triu(m, k):

m:表示一个矩阵 

K:表示对角线的起始位置(k取值默认为0) 

以一个5*5的矩阵举例说明: 

#k=0表示正常的上三角矩阵

upper_triangle = np.triu(data, 0)

[[1 2 3 4 5]

 [0 5 6 7 8]

 [0 0 7 8 9]

 [0 0 0 7 8]

 [0 0 0 0 5]]

#k=-1表示对角线的位置下移1个对角线

upper_triangle = np.triu(data, -1)

[[1 2 3 4 5]

 [4 5 6 7 8]

 [0 7 7 8 9]

 [0 0 6 7 8]

 [0 0 0 4 5]]

#k=1表示对角线的位置上移1个对角线

upper_triangle = np.triu(data, 1)

[[0 2 3 4 5]

 [0 0 6 7 8]

 [0 0 0 8 9]

 [0 0 0 0 8]

 [0 0 0 0 0]] 

#取下三角阵

def tril(m, k):

lower_triangle = np.tril(data, 0)

[[1 0 0 0 0]

 [4 5 0 0 0]

 [6 7 7 0 0]

 [4 5 6 7 0]

 [1 2 3 4 5]]

lower_triangle = np.tril(data, -1)

[[0 0 0 0 0]

 [4 0 0 0 0]

 [6 7 0 0 0]

 [4 5 6 0 0]

 [1 2 3 4 0]]

lower_triangle = np.tril(data, 1)

[[1 2 0 0 0]

 [4 5 6 0 0]

 [6 7 7 8 0]

 [4 5 6 7 8]

 [1 2 3 4 5]]

0 人点赞