python求一个输入矩阵主对角线和副对角线所有元素和

可以使用NumPy库来计算一个输入矩阵的主对角线和副对角线所有元素和。具体实现方式如下：

import numpy as np

# 获取用户输入的矩阵
n = int(input("请输入矩阵的行数和列数："))
matrix = np.zeros((n, n))
for i in range(n):
    for j in range(n):
        matrix[i][j] = int(input("请输入第{}行第{}列的元素：".format(i+1, j+1)))

# 计算矩阵主对角线元素的和
diag_sum = np.trace(matrix)

# 计算矩阵副对角线元素的和
rev_diag_sum = np.trace(np.fliplr(matrix))

print(diag_sum) # 输出主对角线元素的和
print(rev_diag_sum) # 输出副对角线元素的和

以上代码中，首先通过`input()`函数获取用户输入的矩阵，并使用NumPy库将其转换为矩阵形式。然后使用`np.trace()`函数分别计算矩阵主对角线和副对角线所有元素的和。其中，对于副对角线，我们需要先使用`np.fliplr()`函数将矩阵左右翻转，然后再计算它的主对角线即可。最后输出结果即可。