numpy.tri

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

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

Параметры

Nint

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

Mint, optional

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

kint, optional

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

dtypedtype, optional

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

likearray_like

Объект-ссылка, позволяющий создавать массивы, которые не являются массивами NumPy. Если переданный в качестве like массив-подобный объект поддерживает протокол __array_function__, результат будет определён им. В этом случае это гарантирует создание объекта массива, совместимого с переданным через этот аргумент.

Примечание

Ключевое слово like — это экспериментальная функция, ожидает принятия NEP 35.

Добавлен в версии 1.20.0.

Возвращает

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