Нижний треугольник в NumPy
Чтобы оставить нижний треугольник квадратной таблицы,
используют функцию tril.
Она копирует таблицу
и обнуляет ячейки выше выбранной
диагонали. При k=0 на месте
остаются главная диагональ и всё
под ней, верхняя часть становится
нулём.
Применим к таблице 3×3
и выведем результат:
import numpy as np
square = np.arange(1, 10).reshape(3, 3)
lower = np.tril(square)
print(lower) # выведет [[1 0 0] [4 5 0] [7 8 9]]
Для таблицы np.arange(16).reshape(4, 4)
оставьте нижний треугольник с
главной диагональю и выведите элемент
в строке 2, столбце 1.
Для таблицы [[2, 2, 2], [2, 2, 2], [2, 2, 2]]
обнулите верх над главной диагональю
и выведите сумму всех ячеек.