본문 바로가기
[업무 지식]/Algorithm

[array] 행렬의 덧셈

by 에디터 윤슬 2024. 12. 27.

링크

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

 

문제

정답 1

# numpy 활용

import numpy as np

def solution(arr1, arr2):
	return (np.array(arr1) + np.array(arr2)).tolist()

정답 2

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