← К задачам
Отследить эту матрицу.
Дана квадратная матрица (т.е. количество строк и столбцов одинаково), ее след - это сумма элементов на главной диагонали (т.е. диагональная линия от верхнего левого угла до нижнего правого).
Например, для:
[
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
... след равен 1 + 5 + 9 = 15.
Напишите функцию, которая принимает квадратную матрицу и вычисляет ее след.
Примеры
trace([
[1, 4],
[4, 1]
]) ➞ 2
# 1 + 1 = 2
trace([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]) ➞ 15
# 1 + 5 + 9 = 15
trace([
[1, 0, 1, 0],
[0, 2, 0, 2],
[3, 0, 3, 0],
[0, 4, 0, 4]
]) ➞ 10
# 1 + 2 + 3 + 4 = 10
Примечание
Как в примерах, размер матриц будет разным (но они всегда будут квадратными).
def trace(lst):
Для запуска тестов необходима авторизация.