求一个5*5矩阵主对角线元素之和。

在一个n阶方阵中，从左上角到右下角这一斜线方向上的n个元素所在的对角线，叫做n阶方阵的主对角线。

程序示例

输入：
随机5*5矩阵：
[92,62,98,100,35]
[23,22,55,12,11]
[12,13,14,15,16]
[17,18,19,20,21]
[22,23,24,25,26]

输出：
主对角线元素之和：
92+22+13+18+23=161

完整代码和注释：

import random
n = 5  # 定义矩阵的大小为5*5
arr = []  # 初始化一个空列表来存放矩阵
x = 0  # 初始化主对角线元素之和为0

# 生成一个5*5的随机矩阵
for i in range(n):
    arr.append([])
    for j in range(n):
        arr[i].append(random.randint(0, 100))
print(f'随机{5}*{5}矩阵:')
# 打印生成的随机矩阵
[print(arr[i]) for i in range(n)]

# 计算矩阵主对角线元素之和
for i in range(n):
    x += arr[i][i]
print(f'矩阵主对角线元素之和：{x}')