调用fun函数计算三维数组主对角线元素之和
时间: 2024-05-08 17:21:27 浏览: 71
C语言程序设计-在键盘上输入一个3行3列矩阵的各个元素的值(值为整数),然后输出主对角线元素的积,并在fun() 函数中输出
5星 · 资源好评率100%
可以使用以下代码实现:
```python
def fun(arr):
n = len(arr)
res = 0
for i in range(n):
res += arr[i][i][i]
return res
```
其中,`arr`是一个三维数组,`n`是数组的第一维长度,即数组的维数。在循环中,我们只需要累加主对角线上的元素,即下标三个维度相等的元素。最后返回累加结果即可。
阅读全文