计算矩阵对角线元素之和的Python程序

矩阵是一种常见的数学结构，由行和列组成的二维数组。在矩阵中，对角线是从左上角到右下角的连续线，包括沿途的元素。本文将介绍如何使用Python编写一个程序来计算矩阵的对角线元素之和。

首先，让我们定义一个示例矩阵来演示程序的工作原理。假设我们有一个3x3的矩阵：

matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]

要计算对角线元素之和，我们需要遍历矩阵的行和列，并将对角线上的元素相加。在这个例子中，我们可以看到对角线上的元素是1、5和9。

接下来，我们可以使用以下Python程序来计算矩阵对角线元素之和：

def diagonal_sum(matrix):
    n = len(matrix)
    diagonal_sum = 0
    for i