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