Функция tri
Функция tri создаёт двумерный массив с единицами на главной диагонали и ниже (нижняя треугольная матрица) или на главной диагонали и выше (верхняя треугольная матрица), в зависимости от параметра k.
Первым параметром она принимает количество строк (N).
Вторым параметром можно указать количество столбцов (M), по умолчанию равно N.
Параметр k задаёт смещение диагонали (по умолчанию 0).
Возвращает массив из единиц и нулей.
Синтаксис
np.tri(N, [M], [k], [dtype])
Пример
Создадим нижнюю треугольную матрицу 3x3:
import numpy as np
res = np.tri(3)
print(res)
Результат выполнения кода:
[[1. 0. 0.]
[1. 1. 0.]
[1. 1. 1.]]
Пример
Создадим треугольную матрицу 3x4:
import numpy as np
res = np.tri(3, 4)
print(res)
Результат выполнения кода:
[[1. 0. 0. 0.]
[1. 1. 0. 0.]
[1. 1. 1. 0.]]
Пример
Создадим матрицу со смещением k=1 (включаем диагональ и одну выше):
import numpy as np
res = np.tri(4, k=1)
print(res)
Результат выполнения кода:
[[1. 1. 0. 0.]
[1. 1. 1. 0.]
[1. 1. 1. 1.]
[1. 1. 1. 1.]]
Пример
Создадим матрицу с отрицательным смещением k=-2:
import numpy as np
res = np.tri(5, k=-2)
print(res)
Результат выполнения кода:
[[0. 0. 0. 0. 0.]
[0. 0. 0. 0. 0.]
[1. 0. 0. 0. 0.]
[1. 1. 0. 0. 0.]
[1. 1. 1. 0. 0.]]
