numpy.tri

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

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

例子

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