python 给定一个n×n的方阵，本题要求计算该矩阵主、副对角线上的所有元素之和。主对角线为从矩阵的左上角至右下角的连线,副对角线为从矩阵的右上角至左下角的连线。。

可以使用两个变量分别记录主对角线和副对角线上的元素之和，然后遍历矩阵，对于每个元素，若它在主对角线上，则将其加到主对角线和中；若它在副对角线上，则将其加到副对角线和中。最后返回两个和即可。

以下是示例代码实现：

def diagonal_sum(matrix):
    n = len(matrix)
    main_sum = 0  # 主对角线和
    sub_sum = 0  # 副对角线和
    for i in range(n):
        main_sum += matrix[i][i]
        sub_sum += matrix[i][n-i-1]
    return main_sum, sub_sum

测试代码：

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
main_sum, sub_sum = diagonal_sum(matrix)
print("主对角线和：", main_sum)
print("副对角线和：", sub_sum)

输出结果为：

主对角线和： 15
副对角线和： 15