请编fun函数,求4×4整型数组的主对角线元素的和。 说明:如下图所示为一个4×4整型数组 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
时间: 2024-01-09 16:05:51 浏览: 98
4*4矩阵对角线之和
下面是一个求解4x4整型数组主对角线元素之和的Python函数fun:
```python
def fun(arr):
n = len(arr)
sum = 0
for i in range(n):
sum += arr[i][i]
return sum
```
其中,参数arr为一个4x4整型数组,返回值为主对角线元素之和。
使用示例:
```python
arr = [[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12], [13, 14, 15, 16]]
print(fun(arr)) # 输出:34
```
阅读全文