dask.array.tri

dask.array.tri

dask.array.tri(N, M=None, k=0, dtype=<class 'float'>, chunks='auto', *, like=None)[源代码]

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

此文档字符串是从 numpy.tri 复制的。

Dask 版本可能存在一些不一致性。

参数
N整数

数组中的行数。

Mint, 可选

数组的列数。默认情况下,M 被设置为等于 N

kint, 可选

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

dtypedtype, 可选

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

喜欢类似数组, 可选

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

1.20.0 新版功能.

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

数组,其下三角部分填充为1,其他部分为0;换句话说,对于 j <= i + kT[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.]])