优化这段代码:n=int(input()) list_a=[] sum=0 for i in range(n): row = input().split( ) list_a.append([int(x) for x in row]) for j in range(n): for k in range(n): if j==k or j+k==n-1: sum+=list_a[j][k] print("{:.2f}".format(sum))
时间: 2024-01-26 13:04:59 浏览: 77
翻转数列python实现,求前n项和,并能输出整个数列的案例
5星 · 资源好评率100%
可以使用 NumPy 库来优化这段代码,NumPy 是 Python 的一个科学计算库,它提供了高效的多维数组对象和相关工具。
以下是使用 NumPy 库优化后的代码:
```python
import numpy as np
n = int(input())
list_a = np.zeros((n, n), dtype=int)
for i in range(n):
row = list(map(int, input().split()))
list_a[i] = row
sum = np.sum(list_a.diagonal()) + np.sum(np.fliplr(list_a).diagonal()) - list_a[n//2][n//2]
print("{:.2f}".format(sum))
```
首先,我们使用 NumPy 创建一个 $n \times n$ 的全零矩阵 `list_a`。然后,我们使用 `map` 函数和 `list` 函数将输入的字符串转换为整数列表,并将它们赋值给 `list_a` 的每一行。
接下来,我们使用 NumPy 的 `diagonal` 函数来计算矩阵的对角线元素之和。由于矩阵有两条对角线,我们需要计算它们的和。我们还需要减去中心元素的值,以避免重复计算。
最后,我们使用 `format` 函数来格式化输出结果。
阅读全文