numpy.tril

numpy.tril(m, k=0)

Функция tril() преобразует указанный массив в треугольный у которого все элементы выше указанной диагонали равны 0.

Параметры:
m - подобный массиву объект
Двумерные, матрицы, списки или кортежи, а так же лбая функция или объект с методом, возвращаеющие список или кортеж.
k - индекс диагонали (необязательный)
По умолчанию k = 0, что соответствует главной диагонали. Положительное значение k смещает диагональ вверх, отрицательное - вниз.
Возвращает:
ndarray - массив NumPy
Возвращает копию указанного массива, с той же формой и типом данных, у которого все элементы выше указанной диагонали равны 0.
Смотрите так же: eye, diag, diagflat, tri, triu


Примеры

>>> import numpy as np
>>> 
>>> a = np.array([[1, 2, 3, 4], [4, 1, 2, 3], [3, 4, 1, 2], [2, 3, 4, 1]])
>>> a
array([[1, 2, 3, 4],
       [4, 1, 2, 3],
       [3, 4, 1, 2],
       [2, 3, 4, 1]])
>>> 
>>> np.tril(a)
array([[1, 0, 0, 0],
       [4, 1, 0, 0],
       [3, 4, 1, 0],
       [2, 3, 4, 1]])
>>> 
>>> np.tril(a, k=1)
array([[1, 2, 0, 0],
       [4, 1, 2, 0],
       [3, 4, 1, 2],
       [2, 3, 4, 1]])
>>> 
>>> np.tril(a, k=-1)
array([[0, 0, 0, 0],
       [4, 0, 0, 0],
       [3, 4, 0, 0],
       [2, 3, 4, 0]])