След в NumPy
Функция trace для двумерной
таблицы складывает числа на главной
диагонали и возвращает одно число.
Найдём след квадратной таблицы:
import numpy as np
square = np.array([[5, 6], [7, 8]])
print(np.trace(square)) # выведет 13
Размер таблицы может быть больше.
У таблицы 3 на 3
на диагонали три ячейки:
import numpy as np
grid = np.array([[1, 0, 0], [0, 2, 0], [0, 0, 3]])
print(np.trace(grid)) # выведет 6
Задайте таблицу [[2, 1], [4, 3]].
Сложите числа на главной диагонали
и выведите результат.
Создайте таблицу 3 на 3
с целыми значениями. Выведите одно
число, равное сумме главной диагонали.