Python 練習實例38

Python 100例

題目：求一個3*3矩陣對角線元素之和。

程序分析：利用雙重for循環控制輸入二維數組，再將a[i][i]累加後輸出。

程序源代碼：

#!/usr/bin/python
# -*- coding: UTF-8 -*-

if __name__ == '__main__':
    a = []
    sum = 0.0
    for i in range(3):
        a.append([])
        for j in range(3):
            a[i].append(float(raw_input("input num:\n")))
    for i in range(3):
        sum += a[i][i]
    print sum

以上實例輸出結果為：

input num:
78
input num:
34
input num:
23
input num:
34
input num:
56
input num:
33
input num:
12
input num:
21
input num:
2
136.0

Python 100例