[연습] 코딩테스트 - 행렬의 덧셈

코딩테스트 연습

• programmers.co.kr
• 행렬의 덧셈

def solution(arr1, arr2):
    answer = []
    for i,x in enumerate(arr1):
        for j, y in enumerate(arr2):
            if  i == j:
                tmp = []
                for z in range(len(arr1[i])):
                    tmp.append(arr1[i][z] + arr2[j][z])
                answer += [tmp]
    return answer