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

 

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

#取上三角阵
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]]