[Python/Algorithm] 행렬의 덧셈

Q.

 

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬을 입력받는 sumMatrix 함수를 완성하여 행렬 덧셈의 결과를 반환해 주세요.

예를 들어 2x2 행렬인 A = ((1, 2), (2, 3)), B = ((3, 4), (5, 6)) 가 주어지면, 같은 2x2 행렬인 ((4, 6), (7, 9))를 반환하면 됩니다.(어떠한 행렬에도 대응하는 함수를 완성해주세요.)

 

 

A.

 

def sumMatrix(A,B):
    answer = []
   
    m = len(A)
    if m > 0: n = len(A[0])
       
    for i in range(m):
        tmp = []
        for j in range(n):
            tmp.append(A[i][j] + B[i][j]) # append() - 배열의 끝에 값을 붙이는 함수
        answer.append(tmp)
   
    return answer

print(sumMatrix([[1,2], [2,3]], [[3,4],[5,6]])) # 이차원 배열 선언 참고