Python求3*3矩阵对角线元素之和

#!/usr/bin/env python
# -*- coding:utf-8 -*-

"""
# 3*3矩阵对角线元素之和
    1   2   3
    4   5   6
    7   8   9
# array = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]
"""

###########################################################################
###########################################################################

a = []
sum = 0.0
for i in range(3):
    a.append([])
    for j in range(3):
        a[i].append(float(input('请输入第%s行第%s列的元素：' % (i + 1, j + 1))))
    print('数组：',a)
for i in range(3):
    sum += a[i][i]
print('数组之和：',sum)
###########################################################################
###########################################################################


"""
请输入第1行第1列的元素：1
请输入第1行第2列的元素：2
请输入第1行第3列的元素：3
数组： [[1.0, 2.0, 3.0]]

请输入第2行第1列的元素：4
请输入第2行第2列的元素：5
请输入第2行第3列的元素：6
数组： [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0]]

请输入第3行第1列的元素：7
请输入第3行第2列的元素：8
请输入第3行第3列的元素：9
数组： [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]]

数组之和： 15.0
"""

# 作业：求4*4矩阵对角线之和
'''
1   2   3   4
5   6   7   8
9   10  11  12
13  14  15  16  
'''