NumPy: вычислить сумму диагонального элемента данного массива

NumPy: упражнение 15 по линейной алгебре с решением

Напишите программу NumPy, вычисляющую сумму диагонального элемента данного массива.

Пример решения :

Код Python:

import numpy as np
m = np.arange(6).reshape(2,3)
print("Original matrix:")
print(m)
result =  np.trace(m)
print("Condition number of the said matrix:")
print(result)

Пример вывода:

 Исходная матрица:
[[0 1 2]
 [3 4 5]]
Номер условия указанной матрицы:
4

Редактор кода Python:

Есть другой способ решить это решение? Внесите свой код (и комментарии) через Disqus.

Предыдущий: Напишите программу NumPy для вычисления номера условия данной матрицы.
Далее: Напишите программу NumPy, чтобы получить нижний треугольник L в разложении Холецкого данного массива.