numpy.tri

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

Массив с единицами на и ниже заданной диагонали и нулями в других местах.

Параметры:	N : int Количество строк в массиве. M : int, optional Количество столбцов в массиве. По умолчанию M принимается равным N. k : int, optional Поддиагональ, на и ниже которой массив заполняется. k = 0 — главная диагональ, в то время как k < 0 — ниже неё, а k > 0 — выше. По умолчанию 0. dtype : dtype, optional Тип данных возвращаемого массива. По умолчанию float.
Возвращает:	tri : ndarray of shape (N, M) Массив, нижняя треугольная часть которого заполнена единицами, а в остальных местах — нулями; другими словами T[i,j] == 1 для i <= j + k, 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.]])