Python 表格对角线求和

matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]

sum1 = 0
sum2 = 0

# 计算从左上角到右下角的对角线和
for i in range(len(matrix)):
    sum1 += matrix[i][i]

# 计算从右上角到左下角的对角线和
for i in range(len(matrix)):
    sum2 += matrix[i][len(matrix) - i - 1]

print("左上角到右下角的对角线和为：", sum1)
print("右上角到左下角的对角线和为：", sum2)

左上角到右下角的对角线和为： 15
右上角到左下角的对角线和为： 15

import numpy as np

# 创建一个3x3矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 计算从左上角到右下角的对角线和
sum1 = np.trace(matrix)

# 计算从右上角到左下角的对角线和
sum2 = np.trace(np.fliplr(matrix))

print("左上角到右下角的对角线和为：", sum1)
print("右上角到左下角的对角线和为：", sum2)

左上角到右下角的对角线和为： 15
右上角到左下角的对角线和为： 15