如何在python中添加矩陣

Question

如何在python中添加矩陣（列表的列表）？我知道我能做到這一點：

X = [[12,7,3], 
    [4 ,5,6], 
    [7 ,8,9]] 

Y = [[5,8,1], 
    [6,7,3], 
    [4,5,9]] 

result = [[0,0,0], 
     [0,0,0], 
     [0,0,0]] 

# iterate through rows 
for i in range(len(X)): 
    # iterate through columns 
    for j in range(len(X[0])): 
     result[i][j] = X[i][j] + Y[i][j] 

for r in result: 
    print(r) 

但是，如果在兩個表中輸入具有行或列數不同？我需要先檢查一下，看看在添加之前行和列的數量是否相同，我不知道如何執行此操作。

Answer 1

進行大小檢查，然後直接使用雙嵌套列表理解來生成結果。 Numpy更好，但如果你想避免導入，這會做。

X = [[12,7,3], 
    [4 ,5,6], 
    [7 ,8,9]] 

Y = [[5,8,1], 
    [6,7,3], 
    [4,5,9]] 

if len(X) == len(Y) and len(X[0]) == len(Y[0]): 
    result = [[X[i][j] + Y[i][j] for j in range(len(X[0]))] 
       for i in range(len(X))] 

print result 

輸出：

[[17, 15, 4], [10, 12, 9], [11, 13, 18]]