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

最新推荐文章于 2025-03-02 16:58:04 发布

BlackMan_阿伟

最新推荐文章于 2025-03-02 16:58:04 发布

阅读量4.8k

点赞数 7

分类专栏： numpy 文章标签： numpy tril（） triu()

本文链接：https://blog.csdn.net/weixin_37724529/article/details/102881776

版权

numpy 专栏收录该内容

2 篇文章

订阅专栏

本文详细解析了numpy库中np.triu和np.tril函数的使用方法，通过实例展示了如何通过参数k控制上三角矩阵和下三角矩阵的生成，适合初学者和需要复习相关知识点的读者。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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