numpy.tri#

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

一个在给定对角线及其下方具有 1,其他位置为 0 的数组。

参数:
Nint

数组中的行数。

Mint,可选

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

kint,可选

在该对角线及其下方填充数组的子对角线。k = 0 是主对角线,而 k < 0 在其下方,k > 0 在其上方。默认值为 0。

dtypedtype,可选

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

likearray_like,可选

参考对象,允许创建非 NumPy 数组的数组。如果作为 like 传入的类似数组支持 __array_function__ 协议,结果将由它定义。在这种情况下,它确保创建与通过此参数传入的数组对象兼容的数组对象。

版本 1.20.0 中的新增功能。

返回值:
tri形状为 (N, M) 的 ndarray

数组的左下角填充 1,其他位置为 0;换句话说,T[i,j] == 1 对于 j <= i + k,否则为 0。

示例

>>> import numpy as np
>>> np.tri(3, 5, 2, dtype=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.]])