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

반응형

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

프로그래머스

문제

두 행렬의 같은 행, 같은 열의 값을 서로 더한 값을 return 하면 됩니다.

 

코드

1. zip() 함수와 for문을 사용해서 배열 안에 있는 값을 꺼내 더한 후 배열 [] 안에 다시 넣어주고 return 헀습니다.

def solution(arr1, arr2):
    return [[x + y for x, y in zip(_arr1, _arr2)] for _arr1, _arr2 in zip(arr1, arr2)]