numpy.tri

麻木的。tri ( N , M=None , k=0 , dtype=<class 'float'> , * , like=None )

一个数组，其中 1 位于给定对角线及其下方，其他位置为零。

参数：

整数​

数组中的行数。

M整数，可选

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

k整数，可选

数组在其及其下方填充的次对角线。 k = 0 为主对角线，k < 0 为主对角线，k > 0 为主对角线。默认值为 0。

数据类型数据类型，可选

返回数组的数据类型。默认为浮动。

类似array_like，可选

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

1.20.0 版本中的新增功能。

返回：

形状为 (N, M) 的三元数组

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

例子

>>> 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.]])