[已解决]求助

isdkz

当然可以简化，针对这个特定问题，我们可以直接计算主对角线元素之和，而不需要使用双重循环。

请参考以下代码：

1. # -*- coding: utf-8 -*-
   
2. 
   
3. def diagonal_sum():
   
4.     A = [
   
5.         [1 ** 2, 2 ** 2, 3 ** 2],
   
6.         [4 ** 2, 5 ** 2, 6 ** 2],
   
7.         [7 ** 2, 8 ** 2, 9 ** 2]
   
8.     ]
   
9.    
   
10.     # 只需计算主对角线元素之和
    
11.     rlt_sum = A[0][0] + A[1][1] + A[2][2]
    
12.     return rlt_sum
    
13. 
    
14. if __name__ == "__main__":
    
15.     print(diagonal_sum())

复制代码

这里我们直接计算了主对角线上的元素之和，而无需使用嵌套循环。这样的代码对于新手来说更易理解。