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

def matrix_diagonal_sum(num):
    a = []
    total = 0
    for i in range(num):
        a.append([])
        for j in range(num):
            a[i].append(int(input("Please input a integer number:")))
            if i == j:
                total += a[i][j]
    return total


if __name__ == '__main__':
    print(matrix_diagonal_sum(3))

！＃的/ usr /斌/蟒＃ - * -编码：UTF-8 - * - 如果__name__ == ' __main__ ' ：
     一个= [ ] 总和= 0.0 为我在范围（3 ）：
         一个。追加（[ ] ）为Ĵ 在范围（3 ）：
             一个[ 我] 。append （float （raw_input （“

 
 
    
       
           输入民：\ n “ ）））为我在范围（3 ）：
         总和+ = 一个[ 我] [ 我] 打印总和

以上实例输出结果为：

输入数字：
78
输入数字：
34
输入数字：
23
输入数字：
34
输入数字：
56
输入数字：
33
输入数字：
12
输入数字：
21
输入数字：
2
136.0