numpy.tri#

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

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

参数:
N整型

数组中的行数。

M整型, 可选

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

k整型, 可选

填充数组的次对角线及以下位置。k = 0 是主对角线，而 k < 0 在主对角线下方，k > 0 在主对角线上方。默认值为 0。

dtype数据类型, 可选

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

like类数组对象, 可选

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

1.20.0 版新增。

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

其下三角填充为一，其他位置为零的数组；换句话说，当 j <= i + k 时 T[i,j] == 1，否则为 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.]])
在此页