numpy.tri

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

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

参数:

Nint

数组中的行数.

M整数,可选

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

k整数,可选

填充数组的子对角线及以下位置. 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.

示例

Try it in your browser!

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

Go BackOpen In Tab