본문 바로가기
알고리즘 문제 풀이

프로그래머스 행렬의 덧셈 (Python)

by 로널드 피셔 2022. 10. 19.

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

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

numpy 라이브러리를 사용하면 간단하지만, 프로그래머스에서는 사용할 수 없다.

따라서 행과 열로 쪼개서 더해야 한다.

행 번호(i)가 같을 때 열 번호(j)가 같은 성분끼리 더해 answer_r 이라는 리스트를 만들고,

answer_r을 합쳐 answer를 생성한다.

댓글