프로그래머스 Lv.1 행렬의 덧셈

1. 문제

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

• 제한사항
  행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

2. 풀이

1차 시도(성공)

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

• append를 사용하는 방법과 zip사용법을 다시한번 보았다.

3. Lv.up

1. zip쓰고, 각자리 요소들끼리 더하기

A = [1, 2, 3]
B = [4, 5, 6]

result = [x + y for x, y in zip(A, B)]

4. Ref.

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

• zip을 겹쳐서 두개 쓴것 같은데 어렵다..