python之矩陣相加:提示使用者輸入矩陣的行數n,再提示使用者輸入矩陣的列數m,接下來,提示使用者輸入 2*n*m 個數字(每次輸入 一個數字)。輸出 C=A+B。

QI_writing碼狗發表於2020-11-20
程式碼:
def func40():
    n = int(input('Please input the number of rows:'))
    m = int(input('Please input the number of columns:'))
    lis1, lis2, lis3 = [], [], []
    t1 = 0
    while t1 < n:
        for j in range(0, m):
            a = int(input('Please input A[%d,%d]:' % (t1, j)))
            lis1.append(a)
        t1 += 1
    t2 = 0
    while t2 < n:
        for j in range(0, m):
            b = int(input('Please input B[%d,%d]:' % (t2, j)))
            lis2.append(b)
        t2 += 1
#按輸入條件,上面兩個while迴圈可以進行整合
    tem1 = 0
    while 0 <= tem1 < (n * m):
        a = lis1[tem1] + lis2[tem1]
        lis3.append(a)
        tem1 += 1
    return [lis3[i:i + m] for i in range(0, len(lis3), m)] #列表拆分成m(列數)個小列表


if __name__ == '__main__':
    print(func40())

相關文章