행렬 더하기

# -*- coding: utf-8 -*-

def sum(Mat_a, Mat_b):

   s = [] # temp list

   for row in range(2):

      L = [] # empty list

      for col in range(2):

         L.append(Mat_a[row][col] + Mat_b[row][col])

      s.append(L)

      del L

   return s

def MatPrintf(I, M):

   if (I == 0): 

      print("m1: _________________")

   elif (I == 1):

      print("m2: _________________")

   else:

      print("MatSum: _____________")

   for row in range(2):

      for col in range(2):

         print ("{element}".format(element = M[row][col]), end=" ")

            

         if col == 1:

            print()

def main():

   m1 = [[1,2], [3,4]]  # 2by2

   m2 = [[4,5], [6,7]]  # 2by2

   MatSum = sum(m1, m2) # 2by2

   total = [m1, m2, MatSum]

   for i in range(len(total)):

      MatPrintf(i, total[i])

if __name__ == "__main__":

   main()