[프로그래머스] 행렬의 덧셈

문제 링크:

https://school.programmers.co.kr/learn/courses/30/lessons/12950

python

def solution(arr1, arr2):
    answer = [[] for _ in range(len(arr1))]

    for i in range(len(arr1)):
        for j in range(len(arr1[0])):
            answer[i].append(arr1[i][j] + arr2[i][j])

    return answer


if __name__ == "__main__":
    result = solution(arr1=[[1,2],[2,3]], arr2=[[3,4],[5,6]])
    print(result)

[[4, 6], [7, 9]]