習題如下：
求一個3*3矩陣對角線元素之和
利用for迴圈控制輸出二維陣列，再將a[i][j]累加後輸出

  1 a = []
  2 sum1 = 0.0
  3 for i in range(3):
  4     a.append([])
  5     for j in range(3):
  6         a[i].append(int(input("輸入數字:")))
  7 print(a)
  8 for i in range(3):
  9     for j in range(3):
 10         if i == j:
 11             sum1 += l[i][j]
 12
 
         if i != j and abs(i-j)!=1:
 13             sum1 += l[i][j]
 14 print(sum1)

執行結果：

輸入數字:1
輸入數字:2
輸入數字:3
輸入數字:4
輸入數字:5
輸入數字:6
輸入數字:7
輸入數字:8
輸入數字:9
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
25.0