Python中的numpy.triu

numpy.triu是NumPy库中的一个函数，用于返回一个矩阵的上三角形部分（包括对角线）。该函数将返回一个新的数组，且原始的数组不会被更改。

语法

numpy.triu(m, k=0)

参数解释：

• m：输入的矩阵。
• k：可选参数，表示上三角返回的主对角线之上的元素个数；默认值为0。

返回值

函数将返回一个新的数组，该数组包含矩阵的上三角形部分。

示例

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
up_arr = np.triu(arr)
print(up_arr)

输出结果为：

array([[1, 2, 3],
       [0, 5, 6],
       [0, 0, 9]])

其中，原始矩阵为：

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

新的数组 up_arr 是原始矩阵的上三角形部分。

示例解析

上面的示例中，我们首先导入了NumPy库，并定义了一个二维数组 arr，该数组包含 3 行和 3 列，且其内容为：

array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])

接下来，我们调用了 numpy.triu 函数，并将 arr 作为其输入参数，此时得到一个新的数组 up_arr，为：

array([[1, 2, 3],
       [0, 5, 6],
       [0, 0, 9]])

如果我们将参数 k 设置为 1，则表示仅返回主对角线之上的元素，即：

up_arr = np.triu(arr, k=1)

此时，输出结果为：

array([[0, 2, 3],
       [0, 0, 6],
       [0, 0, 0]])

总结

numpy.triu 是NumPy库中的一个函数，用于返回一个矩阵的上三角形部分（包括对角线）。其语法为：numpy.triu(m, k=0)。参数 m 表示输入的矩阵，参数 k 表示上三角返回的主对角线之上的元素个数；默认值为0。函数将返回一个新的数组，该数组包含矩阵的上三角形部分。