如何在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]]