Python 中的 numpy.tri 方法

numpyserver side programmingprogramming更新于 2025/8/14 12:37:17

numpy.tri 方法可用于获取一个数组,该数组在给定对角线及其下方为 1,在其他位置为 0。

语法

numpy.tri(N, M=None, k=0, dtype=<class 'float'>)

参数

numpy.tri 接受以下参数 −

  • N - 定义数组的行数。

  • M - 定义数组的列数。默认情况下,它是 None。

  • k - 对于主对角线,使用 k = 0,k < 0 表示在其下方,k > 0 表示在其上方。

  • dtype - 返回数组的数据类型。默认情况下,它是浮点型。

示例 1

让我们考虑以下示例 −

# 导入 numpy 库
import numpy as np

# numpy.tri() 函数
y = np.tri(5, 3, 0)

# 显示 Tri 值
print("Y 的 Tri 函数:
", y)

输出

它将生成以下输出 −

Y 的 Tri 函数:
[[1. 0. 0.]
[1. 1. 0.]
[1. 1. 1.]
[1. 1. 1.]
[1. 1. 1.]]

示例 2

我们再举一个例子 −

#导入 numpy 库
导入 numpy 作为 np

# numpy.tri() 函数
x = np.tri(2, 4, -5, dtype=int)

# 显示三元组值
print("X 的三元组函数:
", x)

输出

它将生成以下输出 −

Tri function of X:
[[0 0 0 0]
[0 0 0 0]]

相关文章