Функция tri

Функция tri создает двумерный массив (матрицу) с единицами на главной диагонали и ниже нее, и нулями выше главной диагонали. Первым параметром функция принимает количество строк N, вторым параметром можно указать количество столбцов M (по умолчанию равно N). Третьим параметром задается смещение диагонали k (по умолчанию 0).

Синтаксис

np.tri(N, [M], [k], [dtype])

Пример

Создадим треугольную матрицу размером 3x3 со стандартной диагональю:

import numpy as np

arr = np.tri(3)
print(arr)

Результат выполнения кода:

[[1. 0. 0.]
 [1. 1. 0.]
 [1. 1. 1.]]

Пример

Создадим матрицу с другим количеством столбцов (3 строки, 4 столбца):

import numpy as np

arr = np.tri(3, 4)
print(arr)

Результат выполнения кода:

[[1. 0. 0. 0.]
 [1. 1. 0. 0.]
 [1. 1. 1. 0.]]

Пример

Используем смещение диагонали k=1, чтобы включить единицы на первой наддиагонали:

import numpy as np

arr = np.tri(4, k=1)
print(arr)

Результат выполнения кода:

[[1. 1. 0. 0.]
 [1. 1. 1. 0.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]]

Пример

Используем отрицательное смещение k=-1, чтобы исключить главную диагональ (строгая нижняя треугольная матрица):

import numpy as np

arr = np.tri(4, k=-1)
print(arr)

Результат выполнения кода:

[[0. 0. 0. 0.]
 [1. 0. 0. 0.]
 [1. 1. 0. 0.]
 [1. 1. 1. 0.]]

Пример

Укажем тип данных элементов матрицы:

import numpy as np

arr = np.tri(3, dtype=int)
print(arr)

Результат выполнения кода:

[[1 0 0]
 [1 1 0]
 [1 1 1]]

Смотрите также

функцию tril,
которая создает нижнюю треугольную матрицу из существующего массива
функцию triu,
которая создает верхнюю треугольную матрицу из существующего массива
функцию diag,
которая извлекает диагональ из массива или создает диагональный массив
функцию diagflat,
которая создает диагональный массив из плоских входных данных