python题目：实现矩阵运算

题目

• 分别构造2*2-2*5，3*2~3*5，4*2-4*5，5*2-5*5共16个矩阵
• 在屏幕端可以选择打印相关矩阵
• 程序运行后，只需提示有哪些类型的矩阵即可
• 实现矩阵运算，包括矩阵的乘法和矩阵的转置，在屏幕端选择运算类型以及需要进行运算的矩阵
• 单独定义并实现一个函数，要求可以判断两个矩阵之间是否可以进行乘法运算，返回布尔值（0或1），要求该函数呗矩阵的乘法功能所调用
• 需要构造类，并实现初始化函数及其成员

说明

• 2*3矩阵是指矩阵的行数是2行，列数是3列

库

• 用numpy库来实现该功能

代码

#2019.1.12 liu先僧

from numpy import *

#创建矩阵类，包含矩阵的行、列和内容以及操作
class Matrix:
    def __init__(self,row,column,ele):
        self.row = row
        self.column = column
        self.ele = ele
    def Multiply(m1,m2):
        if judge(m1,m2) == 1:
            print(m1.ele*m2.ele)
        else:
            print("Can not Multiply!")
    def Transpose(m):
        print(m.ele.T)

#判断两个矩阵是否可以相乘的函数
def Judge(m1,m2):
    if m1.column == m2.row:
        return 1
    else:
        return 0

#创建一个储存矩阵的字典，例如matrixs['2*2'] = [[1,2],[3,4]]
matrixs = {}

#初始化，输入矩阵
print("Please input 16 matrices one by one, for example [[1,2],[3,4]] is a matrix of 2 rows and 2 columns.")
for i in range(2,6):
    for j in range(2,6):
        print("Please enter the matrix of the {0} row {1} column".format(i,j))
        row = i
        column = j
        ele = mat(eval(input()))
        name = str(i) + '*' + str(j)
        matrixs[name] = Matrix(row,column,ele)

#菜单，可以进行矩阵的打印、乘法以及转置
while(True):
    print("Input:\n1:to Print\n2:to Multiply or Transpose\n3:to Quit")#菜单
    op = int(input())
    if op == 1:#矩阵的打印
        print("Input the matrix you want to output like 2*2")
        s = input()
        print(matrixs[s].ele)
        while(True):#输入 1 继续
            print("Input 1 to continue!")
            a = int(input())
            if a == 1:
                break
    elif op == 2:#矩阵的运算
        print("Input:\n1:to Multiply\n2:to Transpose\n3:to Quit")
        b = int(input())
        if b == 1:#矩阵的乘法
            print("Input the first Matrix")
            s1 = input()
            print("Input the other Matrix")
            s2 = input()
            m1 = matrixs[s1]
            m2 = matrixs[s2]
            Matrix.Multiply(m1,m2)
        elif b == 2:#矩阵的转置
            print("Input the Matrix")
            s = input()
            m = matrixs[s]
            Matrix.Transpose(m)
        elif b == 3:#退出
            print("Quit Successfully!")
            continue
        else:
            print("INPUT ERROR!")
    elif op == 3:
        print("Quit Successfully!")
        break
    else:
        print("INPUT ERROR!")