행렬의 덧셈

나의 풀이 

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        a1 = arr1.pop(0)
        a2 = arr2.pop(0)
        num = [x + y for x, y in zip(a1, a2)]
        answer.append(num)
    return answer

 

 

다른 사람 풀이

def sumMatrix(A,B):
    answer = [[c + d for c, d in zip(a, b)] for a, b in zip(A,B)]
    return answer

이중 for문 컴프리헨션으로 끝냈다.