numpy.triu

numpy.triu(m, k=0)

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

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


Примеры

>>> 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.triu(a)
array([[1, 2, 3, 4],
       [0, 1, 2, 3],
       [0, 0, 1, 2],
       [0, 0, 0, 1]])
>>> 
>>> np.triu(a, k=1)
array([[0, 2, 3, 4],
       [0, 0, 2, 3],
       [0, 0, 0, 2],
       [0, 0, 0, 0]])
>>> 
>>> np.triu(a, k=-1)
array([[1, 2, 3, 4],
       [4, 1, 2, 3],
       [0, 4, 1, 2],
       [0, 0, 4, 1]])