Python numpy.tril函数方法的使用

手机查看 2024-07-10

Python 中，numpy.tril 函数用于从一个给定的数组中提取其下三角部分。下三角部分是指矩阵中位于主对角线及其下方的元素，其余元素被设为零。这个函数对于处理矩阵相关的计算非常有用。本文主要介绍一下NumPy中tril方法的使用。

numpy.tril

numpy.tril(m, k=0) [source]

数组的下三角形。

返回数组的副本，其中第k个对角线上方的元素为零。

参数：

m ：array_like, shape (M, N)

输入数组。

k ：int, 可选

对角线上方为零元素。 k = 0（默认值）是主对角线，

k<0在其下方，k>0在其上方。

返回值：

tril ：ndarray, shape (M, N)

m的下三角形，形状和数据类型与m相同。

例子

import numpy as np

# 定义一个 4x3 矩阵
A = np.array([[1, 2, 3],
              [4, 5, 6],
              [7, 8, 9],
              [10, 11, 12]])

print("原矩阵：")
print(A)

# 使用 np.tril 提取主对角线下方的元素，k=-1
L = np.tril(A, -1)
print("下三角矩阵 (k=-1)：")
print(L)

# 提取下三角矩阵
L = np.tril(A)
print("下三角矩阵 (k=0)：")
print(L)

# 示例 2：k 参数的使用
L_k1 = np.tril(A, k=1)
print("下三角矩阵 (k=1)：")
print(L_k1)

L_k_1 = np.tril(A, k=-1)
print("下三角矩阵 (k=-1)：")
print(L_k_1)

文档：numpy.tril.html