（Python）矩阵运算

【问题描述】
编写一个程序，输入两个矩阵AN×N与BN×N（2≤N≤10）进行如下运算：
1.加
2.减
3.乘
【输入形式】
标准输入上依次输入以下内容：
1.输入第一个矩阵AN×N。形式是N个输入行，每行有分别N个整数（可能为负数），代表矩阵AN×N的所有元素。各整数之间以一个或多个空格分隔。
2.输入运算符，是一个字符，可以是"+"、"-"、"*"，分别代表加、减、乘三种 *** 作。
3.输入第二个矩阵BN×N。形式与AN×N相同。
4.以空行表示所有输入的结束。
【输出形式】
在标准输出上打印运算结果，同样是输出N行，每行分别输出该行的N个元素。每个整数占据5个字符宽度（包括负号），向右对齐，不足部分补以空格。
【样例输入】
（下图中"#"代表一个空格）
5##7#12
3###-15# 4
2#6#7
+
41#8#9
7#6#12
2#4#6
【样例输出】
##46##15##21
##10##-9##16
###4##10##13
【样例说明】
输入的A和B分别是两个三阶矩阵，要求进行加法运算，结果如上例所示。

实现代码：

list1 = []
list2 = []
list3 = []
result1 = []
result2 = []
result3 = []
flag = 0
def MatrixMultiplication(A, B):
    if len(A[0]) == len(B):
        res = [[0] * len(B[0]) for i in range(len(A))]
        for i in range(len(A)):
            for j in range(len(B[0])):
                for k in range(len(B)):
                    res[i][j] += A[i][k] * B[k][j]
        return res
while 1:
    list1 = (input().split())
    n = len(list1)
    if n > 1:
        list1 = [int(x) for x in list1]
        result1.append(list1)
        flag += 1
    else:
        c = list1[0]
        break
for i in range(flag):
    list2 = (input().split())
    list2 = [int(y) for y in list2]
    result2.append(list2)
for i in range(flag):
    for j in range(flag):
        list3.append(0)
    result3.append(list3)
    list3 = []
if c == '+':
   for i in range(flag):
       for j in range(flag):
           result3[i][j] = result1[i][j] + result2[i][j]
elif c == '-':
     for i in range(flag):
         for j in range(flag):
             result3[i][j] = result1[i][j] - result2[i][j]
else:
     result3 = MatrixMultiplication(result1,result2)
for i in result3:
    for j in i:
        print("%5d" % j, end="")
    print()

运行结果：