numpy.tri
numpy.tri(N, M=None, k=0, dtype=) [source]
在给定对角线处及以下且在其他位置为零的数组。
参数 ： | N ： 数组中的行数。 M ： 数组中的列数。 默认情况下，M等于N。 k ： 填充数组并在其下的对角线。 而 dtype ： 返回数组的数据类型。 默认为 |
返回值 ： | tri ：ndarray of shape (N, M) 下三角的数组由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)