Верхний треугольник в NumPy
Функция triu симметрична
нижнему треугольнику, но обнуляет
ячейки ниже диагонали. При k=0
сохраняются главная диагональ и всё
над ней.
Применим к той же таблице 3×3
и выведем результат:
import numpy as np
square = np.arange(1, 10).reshape(3, 3)
upper = np.triu(square)
print(upper) # выведет [[1 2 3] [0 5 6] [0 0 9]]
Для таблицы np.arange(9).reshape(3, 3)
оставьте верхний треугольник и
выведите элемент в строке 0,
столбце 2.
Для таблицы np.ones((3, 3))
оставьте только ячейки строго выше
главной диагонали и выведите сумму
полученной таблицы.