上一主题

numpy.diagflat

下一主题

numpy.tril

numpy.tri

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

数组,其中一个在给定的对角线和在其他地方零。

参数:

N:int

数组中的行数。

M:int,可选

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

k:int,可选

数组被填充的子对角线。k = 0 is the main diagonal, while k < 0 is below it, and k > 0 is above. 默认值为0。

dtype:dtype,可选

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

返回:

tri:形状的引号(N,M)

Array with its lower triangle filled with ones and zero elsewhere; in other words T[i,j] == 1 for i <= j + k, 0 otherwise.

例子

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