numpy 返回函数的上三角矩阵实例

  • Post category:Python

当我们需要对一个矩阵进行上三角矩阵的操作时,可以使用Numpy中的triu函数。本文将详细介绍如何使用triu函数,并提供两个示例来说明它的用法。

triu函数的语法

triu函数的语法如下:

numpy.triu(m, k=0)

其中,参数m是要进行操作的矩阵,参数k是指定对角线的偏移量。当k=0时,表示对角线上的元素也包含在上三角矩阵中;当k>0时,表示对角线上方k个元素不包含在上三角矩阵中;当k<0时,表示对角线下方-k个元素不包含在上三角矩阵中。

triu函数的示例

下面是两个示例,演示如何使用triu函数返回一个矩阵的上三角矩阵。

示例一

import numpy as np

# 创建一个矩阵
m = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 返回矩阵的上三角矩阵
result = np.triu(m)

print(result)

在上面的示例中,我们创建了一个矩阵m,然后使用triu函数返回了该矩阵的上三角矩阵。由于没有指定参数k的值,因此对角线上的元素也包含在上三角矩阵中。输出结果如下:

[[1 2 3]
 [0 5 6]
 [0 0 9]]

示例二

import numpy as np

# 创建一个矩阵
m = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 返回矩阵的上三角矩阵,但不包含对角线上的元素
result = np.triu(m, k=1)

print(result)

在上面的示例中,我们创建了一个矩阵m,然后使用triu函数返回了该矩阵的上三角矩阵,但是不包含对角线上的元素。由于指定了参数k的值为1,因此对角线上的元素不包含在上三角矩阵中。输出结果如下:

[[0 2 3]
 [0 0 6]
 [0 0 0]]

在这两个示例中,我们都使用了triu函数对一个矩阵进行了上三角矩阵的操作。需要注意的是,triu函数可以通过指定参数的值来控制上三角矩阵中是否包含对角线上的元素。