import sys
import numpy as np

def Metoda_1(a, b):

    print("Suma A+B")
    print(a + b)
    print("Roznica A-B")
    print(a - b)

def Metoda_2(a, b):

    print("Iloczyn A*B")
    print(a * b)
    print("Iloczyn A*B")
    print(b * a)

def Metoda_3(a, b):
    print("Transpozycja - A")
    print(a.transpose())
    print("Transpozycja - B")
    print(b.transpose())

def Menu(n):
    A=[]
    B=[]
    while(True):
        print("Wybierz 1 - Wprowadz macierze")
        print("Wybierz 2 - Oblicz A + B oraz A - B")
        print("Wybierz 3 - Oblicz A * B oraz B * A")
        print("Wybierz 4 - Oblicz Transpozycja A oraz Transpozycja B")
        print("Wybierz 5 - Wyjscie")
        menu = int(input())
        if menu == 1:
            print("Wprowadz macierz A")
            for i in range(n):
                A.append([int(j) for j in input().split()])
            print("Wprowadz macierz B")
            for i in range(n):
                B.append([int(j) for j in input().split()])
            a = np.matrix(A)
            b = np.matrix(B)
        elif menu == 2:
            Metoda_1(a, b)
        elif menu == 3:
            Metoda_2(a, b)
        elif menu == 4:
            Metoda_3(a, b)
        elif menu == 5:
            return 0;




while(True):
    print("MENU GLOWNE")
    print("Wybierz 1 - Macierze 2x2")
    print("Wybierz 2 - Macierze 3x3")
    print("Wybierz 3 - Wyjscie")
    opcja = int(input())
    if opcja == 1:
        n=2
        Menu(n)
    elif opcja == 2:
        n=3
        Menu(3)
    elif opcja == 3:
        sys.exit(0)