Как добавить суммирование матрицы ко всем диагональным элементам матрицы? [дубликат] ⇐ Python

[Anonymous]

Я пытаюсь добавить суммирование матрицы ко всем диагональным элементам матрицы.

У меня есть код для получения диагоналей матрицы и код для суммирования матрицы. Как изолировать диагонали каждой матрицы?

Вот мой текущий код:

diag = np.diag(A) А_сум = А.сум() Пример:

Исходная матрица:

[[0 1 2] [3 4 5] [6 7 8]] Желаемый результат:

[[36 1 2] [ 3 40 5] [ 6 7 44]] Сумма всех значений исходной матрицы равна 36. Я бы добавил 36 к диагональным значениям исходной матрицы.