计算二维数组对角线元素之和 python
时间: 2023-06-06 08:07:35 浏览: 291
python实战:矩阵对角线之和.zip
可以的,可以使用python的for循环和切片来计算二维数组对角线元素之和。代码如下:
```python
def diagonal_sum(arr):
n = len(arr)
sum = 0
for i in range(n):
sum += arr[i][i] + arr[i][n-i-1]
if n%2 == 1:
sum -= arr[n//2][n//2]
return sum
arr = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
print(diagonal_sum(arr)) #输出15
```
该代码首先计算了从左上角到右下角的对角线上所有元素之和,然后再计算从右上角到左下角的对角线上的所有元素之和,最后把这两个和相加即可得到二维数组的对角线元素之和。
阅读全文