Чтобы найти диагональную сумму матрицы в Python, вы можете использовать несколько подходов. Вот несколько способов:

Метод 1: использование вложенных циклов

def diagonalSum(matrix):
    n = len(matrix)
    sum = 0
    for i in range(n):
        for j in range(n):
            if i == j:
                sum += matrix[i][j]
    return sum

Метод 2: использование понимания списка

def diagonalSum(matrix):
    n = len(matrix)
    return sum(matrix[i][i] for i in range(n))

Метод 3: использование NumPy
Если у вас установлен NumPy, вы можете использовать функцию numpy.diagonalдля извлечения диагональных элементов и последующего вычисления их суммы.

import numpy as np
def diagonalSum(matrix):
    diagonal = np.diagonal(matrix)
    return sum(diagonal)

Метод 4: использование NumPy (альтернативный вариант)
Другой способ использования NumPy — преобразовать матрицу в массив NumPy, а затем использовать индексацию массива для извлечения диагональных элементов.

import numpy as np
def diagonalSum(matrix):
    diagonal = np.array(matrix).diagonal()
    return sum(diagonal)