用python，对一个四行四列的矩阵，分别计算主对角线和副对角线元素之和

Question

用python，对一个四行四列的矩阵，分别计算主对角线和副对角线元素之和

Answer 1

arr = np.arange(16).reshape(4,4)

np.diagonal(arr).sum()  # 主对角线和

np.flipud(arr).diagonal().sum()   # 副对角线