numpy.tri#

numpy.tri(N, M=None, k=0, dtype=<class 'float'>, *, like=None)[source]#

一个在给定对角线及其以下位置为 1、其他位置为 0 的数组。

参数:

Nint: 数组的行数。
Mint, optional: 数组的列数。默认情况下，M 等于 N。
kint, optional: 数组在其下方（包括）填充 1 的子对角线。k = 0 表示主对角线，k < 0 表示位于主对角线之下，k > 0 表示位于主对角线之上。默认值为 0。
dtypedtype, optional: 返回数组的数据类型。默认是 float。
likearray_like, optional: 用于创建非 NumPy 数组的引荐对象。如果传入的 like 支持 __array_function__ 协议，则结果将由它定义。在这种情况下，它确保创建与通过此参数传入的数组兼容的数组对象。

版本 1.20.0 中新增。

返回:

tri形状为 (N, M) 的 ndarray: 数组的下三角填充为 1，其他位置为 0；即当 j <= i + k 时 T[i,j] == 1，否则为 0。

示例

>>> import numpy as np
>>> np.tri(3, 5, 2, dtype=np.int_)
array([[1, 1, 1, 0, 0],
       [1, 1, 1, 1, 0],
       [1, 1, 1, 1, 1]])

>>> np.tri(3, 5, -1)
array([[0.,  0.,  0.,  0.,  0.],
       [1.,  0.,  0.,  0.,  0.],
       [1.,  1.,  0.,  0.,  0.]])