python给定一个n×n的方阵,本题要求计算该矩阵主、副对角线上的所有元素之和。主对角线为从矩阵的左上角至右下角的连线,副对角线为从矩阵的右上角至左下角的连线。。 输入格式: 输入第一行给出正整数n(1<n≤10);随后n行,每行给出n个数,其间以空格分隔。 输出格式: 在一行中给出该矩阵主、副对角线的所有元素之和,保留两位小数。 输入样例1: 在这里给出一组输入。例如: 4 2 3 4 1 5 6 1 1 7 1 8 1 1 1 1 1 输出样例1: 在这里给出相应的输出。例如: 21.00 输入样例2: 在这里给出一组输入。例如: 3 1 2 3 4 5 6 7 8 9 输出样例1: 在这里给出相应的输出。例如: 25.00
时间: 2024-02-09 16:12:32 浏览: 359
python 实现矩阵对角线之和
5星 · 资源好评率100%
好的,这个问题其实和您之前的问题一样,只需要稍微修改一下代码即可。下面是可以解决这个问题的 Python 代码:
```python
import numpy as np
# 读取矩阵大小
n = int(input())
# 读取矩阵
matrix = []
for i in range(n):
row = list(map(int, input().split()))
matrix.append(row)
# 将矩阵转换为 NumPy 数组
matrix = np.array(matrix)
# 计算主对角线和副对角线上的元素之和
main_sum = np.trace(matrix)
anti_sum = np.trace(matrix[::-1], offset=n-1)
# 计算总和并输出结果
total_sum = main_sum + anti_sum
print('%.2f' % total_sum)
```
这个程序与之前的程序类似,只是最后计算总和时要加上主对角线和副对角线的和。输出结果时使用 `%.2f` 格式化字符串将结果保留两位小数。
希望这个代码对您有帮助!
阅读全文