numpy.tri

numpy.tri(N, M=None, k=0, dtype=<type 'float'>) [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) Массив, заполненный единицами в нижнем треугольнике и нулями в остальных местах; другими словами, 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.]])