(Python)矩阵运算

190 阅读2分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

【问题描述】 编写一个程序,输入两个矩阵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()

运行结果:

在这里插入图片描述