求一个3*3矩阵对角线元素之和代码

以下是一个求3*3矩阵对角线元素之和的代码：

matrix = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
sum = 0
for i in range(3):
    sum += matrix[i][i]
print(sum)

这个代码会输出对角线元素之和，即 1+5+9=15。