Python numpy.tri函数方法的使用

numpy.tri 函数用于生成一个下三角矩阵,即矩阵中的所有元素在主对角线及其以下的位置上为 1,其余位置为 0。它可以用于生成特定形状的矩阵,常用于线性代数和科学计算。本文主要介绍一下NumPy中tri方法的使用。

numpy.tri

numpy.tri(N, M=None, k=0, dtype=)      [source]

在给定对角线处及以下且在其他位置为零的数组。

参数 :

Nint

数组中的行数。

Mint, 可选

数组中的列数。 默认情况下,M等于N。

kint, 可选

填充数组并在其下的对角线。k = 0是主对角线,

k<0在其下方,而k>0在其上方。 默认值为0

dtypedtype, 可选

返回数组的数据类型。 默认为float

返回值 :

tri :ndarray of shape (N, M)

下三角的数组由1填充,在其他地方为0;

 换句话说,对于j<= i + kT [i,j] == 1,否则为0。

例子

1)基本用法

生成一个 3x3 的下三角矩阵:

import numpy as np
# 生成一个 3x3 的下三角矩阵
matrix = np.tri(3, 3)
print("3x3 下三角矩阵:")
print(matrix)

2)指定列数

生成一个 3x5 的矩阵:

import numpy as np

# 生成一个 3x3 的下三角矩阵
matrix = np.tri(3, 3)

# 生成一个 3x5 的下三角矩阵
matrix = np.tri(3, 5)
print("3x5 下三角矩阵:")
print(matrix)

3)指定对角线位置

生成一个 4x4 的矩阵,并指定 k 参数为 1,表示对角线以上一行也是 1:

import numpy as np

# 生成一个 3x3 的下三角矩阵
matrix = np.tri(3, 3)

# 生成一个 4x4 的矩阵,并指定 k 参数为 1
matrix = np.tri(4, 4, k=1)
print("k=1 的 4x4 矩阵:")
print(matrix)

4)指定数据类型

生成一个整数类型的下三角矩阵:

import numpy as np

# 生成一个 3x3 的下三角矩阵
matrix = np.tri(3, 3)

# 生成一个 3x3 的整数类型下三角矩阵
matrix = np.tri(3, 3, dtype=int)
print("整数类型的 3x3 下三角矩阵:")
print(matrix)

5)使用示例

import numpy as np

# 生成一个 3x5 的下三角矩阵,并指定 k 参数为 2,数据类型为 int
matrix1 = np.tri(3, 5, 2, dtype=int)
print("3x5 下三角矩阵 (k=2, dtype=int):")
print(matrix1)

# 生成一个 3x5 的下三角矩阵,并指定 k 参数为 -1
matrix2 = np.tri(3, 5, -1)
print("\n3x5 下三角矩阵 (k=-1):")
print(matrix2)

文档numpy.tri.html

推荐阅读
cjavapy编程之路首页