菜鸟教程 练习实例38 （python3）

题目：求一个3*3矩阵主对角线元素之和。
程序分析：利用双重for循环控制输入二维数组，再将a[i][i]累加后输出。

# -*- coding: UTF-8 -*-


def main():
    array = [[0 for i in range(3)] for i in range(3)]
    for i in range(3):  # 行
        for j in range(3):  # 列
            print('第{0}行，第{1}列：'.format(i + 1, j + 1), end='')
            array[i][j] = int(input())
    print('您输入的二维数组如下：')
    for i in range(3):
        for j in range(3):
            print('  %d' % array[i][j], end='')
        print()
    sum_array = 0
    for i in range(3):
        sum_array += array[i][i]
    print('合为：%d' % sum_array)


main()