python 행렬의 덧셈

생각보다 풀기가 어려웠던 문제

def solution(arr1, arr2):
    answer=[]
    for i in range(len(arr1)):
        tmp = []
        for j in range(len(arr1[0])):
            tmp.append(arr1[i][j]+arr2[i][j])
        answer.append(tmp)
    return answer

잘푼분 코드 zip활용을 잘하셨다

def solution2(A,B):
    return [[i+j for i,j in zip(a,b)] for a,b in zip(A,B)]