矩阵加法需要矩阵同型
矩阵乘法A·B需要满足A的列等于B的行
代码如下
class MatrixCalc: A = [] B = [] def __init__(self, shapeA, shapeB): self.shapeA = shapeA self.shapeB = shapeB def input_(self): print("Please input the Matrix A and the Matrix B below:") MatrixCalc.A = [] MatrixCalc.B = [] for i in range(self.shapeA[0]): MatrixCalc.A.append([int(e) for e in input().split()]) print("") for i in range(self.shapeA[1]): MatrixCalc.B.append([int(e) for e in input().split()]) def add(self): if self.shapeA[0] != self.shapeB[0] or self.shapeA[1] != self.shapeB[1]: print("Error!") else: MatrixCalc.input_(self) print("The result is:") for i in range(self.shapeA[0]): print([MatrixCalc.A[i][j] + MatrixCalc.B[i][j] for j in range(self.shapeA[1])]) def multiply(self): if self.shapeA[1] != self.shapeB[0]: print("Error!") else: MatrixCalc.input_(self) print("The result is:") C = [] for i in range(self.shapeA[0]): tmp = [] for k in range(self.shapeB[1]): sum = 0 for j in range(self.shapeA[1]): sum += MatrixCalc.A[i][j] * MatrixCalc.B[j][k] tmp.append(sum) C.append(tmp) for i in C: for j in i: print(j, end=' ') print()if __name__ == '__main__': shapeA = [int(i) for i in input("Please input the shape of Matrix A:").split()] shapeB = [int(i) for i in input("Please input the shape of Matrix B:").split()] cal_mode = input("How do you want to calculate it? Choose one:n[1] '+'n[2] '*'n") M = MatrixCalc(shapeA, shapeB) if cal_mode is '+': M.add() else: M.multiply()