Python numpy.tri函数方法的使用

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中tri方法的使用。

numpy.tri

numpy.tri(N, M=None, k=0, dtype=) [source]

在给定对角线处及以下且在其他位置为零的数组。

参数：

N ：int

数组中的行数。

M ：int, 可选

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

k ：int, 可选

填充数组并在其下的对角线。k = 0是主对角线，

而k<0在其下方，而k>0在其上方。默认值为0。

dtype ：dtype, 可选

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

返回值：

tri ：ndarray of shape (N, M)

下三角的数组由1填充，在其他地方为0；

换句话说，对于j<= i + k，T [i，j] == 1，否则为0。

例子

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