Lab06 : Using 2D array as matrix

def setup():
    row1=[20,30,70]
    row2=[10,25,15]
    row3=[30,15,35]
    matrix1=[row1,row2,row3]
    arow1=[40,20,23]
    arow2=[10,15,70]
    arow3=[23,15,27]
    matrix2=[arow1,arow2,arow3]
    opear="+"
    display_2D(matrix1)
    print(opear)
    display_2D(matrix2)
    print("=")
    result=do_two_2D(matrix1,matrix2,opear)
    display_2D(result)
    transpose(result)
    print("after transpose")
    display_2D(result)
def transpose(aa):
    i=0
    j=0
    while i<len(aa):
        j = i+1
        while j<len(aa[0]):
            temp = aa[i][j]
            aa[i][j]=aa[j][i]
            aa[j][i]=temp
            j+=1
        i+=1

def do_two_2D(aa,bb,mode): #2D
    matrix=[]
    i=0
    while i<len(aa):
        matrix.append([]) #append row
        k=0
        while k<len(aa[0]):
            matrix[i].append(opearation(aa[i][k],bb[i][k],mode))
            k+=1
        i+=1
    return matrix
def opearation(a,b,mode):
    if mode=="+":
        return a+b
    if mode=="-":
        return a-b
    if mode=="*":
        return a*b

def display_2D(aa):
    i=0
    while i<len(aa):
        display_1D(aa[i])
        print()
        i+=1
def display_1D(a):
    i=0
    while i<len(a):
        print(a[i],end=' ')
        i+=1

  
setup()