(Python)矩阵运算

@[TOC]

【问题描述】

编写一个程序，输入两个矩阵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分别是两个三阶矩阵，要求进行加法运算，结果如上例所示。

【题解】

def Multiple(A, B):
    if len(A[0]) == len(B):
        result = [[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)):
                    result[i][j] += A[i][k] * B[k][j]
        return result
list_one = []
list_str = input()
list_one = list(map(int,list_str.split()))
for i in range(len(list_one)-1):
    list_str = input()
    list_new = list(map(int,list_str.split()))
    list_one.extend(list_new)
list_three = list_one.copy()
num = len(list_one)
count = int(len(list_one)**0.5)
flag = input()
list_two = []
list_str = input()
list_two = list(map(int,list_str.split()))
for i in range(len(list_two)-1):
    list_str = input()
    list_new = list(map(int,list_str.split()))
    list_two.extend(list_new)

result_one = []
for y in range(0, count):
    for x in range(0, count):
        if x == 0:
            result_one.append([])
        result_one[y].append(list_one[x + y * count])
result_two = []
for y in range(0, count):
    for x in range(0, count):
        if x == 0:
            result_two.append([])
        result_two[y].append(list_two[x + y * count])
result_three = []
list_three = []
for i in range(count):
    for j in range(count):
        list_three.append(0)
    result_three.append(list_three)
    list_three = []
if flag == '+':
   for i in range(count):
       for j in range(count):
           result_three[i][j] = result_one[i][j] + result_two[i][j]
elif flag == '-':
     for i in range(count):
         for j in range(count):
             result_three[i][j] = result_one[i][j] - result_two[i][j]
else:
     result_three = Multiple(result_one,result_two)
for i in result_three:
    for j in i:
        print("%5d" % j, end="")
    print()