วันอาทิตย์ที่ 25 ตุลาคม พ.ศ. 2558

Lab 6 - Matrix

def setup():
    matrix1 = [[15,50],[4,16]]
    matrix2 = [[15,10],[10,12]]
    print("Matrix One")
    display(matrix1)
    print("\nMatrix Two")
    display(matrix2)
    print("\nAdd Two Matrices")
    display(addMatrix(matrix1,matrix2))
    print("\nSubtract Two Matrices")
    display(subtractMatrix(matrix1,matrix2))
    print("\nMultiply Two Matrices")
    display(multiplyMatrix(matrix1,matrix2))

   
def display(matrix):
    i = 0
    while(i < len(matrix)):
        j = 0
        while(j < len(matrix[i])):
            print(matrix[i][j],end = ' ')
            j += 1
        print()
        i += 1

def addMatrix(matrix1,matrix2):
    answer = [[0,0],[0,0]]
    i = 0
    while(i < len(matrix1)):
        j = 0
        while(j < len(matrix1[i])):
            answer[i][j] = matrix1[i][j]+matrix2[i][j]
            j += 1
        i += 1
    return answer

def subtractMatrix(matrix1,matrix2):
    answer = [[0,0],[0,0]]
    i = 0
    while(i < len(matrix1)):
        j = 0
        while(j < len(matrix1[i])):
            answer[i][j] = matrix1[i][j]-matrix2[i][j]
            j += 1
        i += 1
    return answer

def multiplyMatrix(matrix1,matrix2):
    answer = [[0,0],[0,0]]
    i = 0
    while(i < len(matrix1)):
        j = 0
        k = 0
        while(j < len(matrix1[i])):
            answer[i][j] = (matrix1[i][k]*matrix2[k][j])+(matrix1[i][k+1]*matrix2[k+1][j])
            j += 1
        i += 1
    return answer

setup()

ไม่มีความคิดเห็น:

แสดงความคิดเห็น