Python numpy.tril函数方法的使用

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

numpy.tril

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

数组的下三角形。

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

参数 :

m :array_like, shape (M, N)

输入数组。

kint, 可选

对角线上方为零元素。 k = 0(默认值)是主对角线,

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

返回值 :

trilndarray, 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

推荐阅读
cjavapy编程之路首页