numpy.tri

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

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

Параметры

Nint

Количество строк в массиве.

Mint, optional

Количество столбцов в массиве. По умолчанию, M принимается равным N.

kint, optional

Поддиагональ, на и ниже которой массив заполняется. k = 0 — главная диагональ, в то время как k < 0 — ниже нее, а k > 0 — выше. По умолчанию равно 0.

dtypedtype, optional

Тип данных возвращаемого массива. По умолчанию — float.

Возвращает

trindarray of shape (N, M)

Массив с заполненной нижней треугольной частью единицами и нулями в других местах; другими словами, T[i,j] == 1 для j <= i + 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.]])