参考链接: 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]]