[프로그래머스 파이썬] 행렬의 덧셈

일단 해볼게·2023년 4월 19일
0

프로그래머스

목록 보기
68/106

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

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

for문으로 하나하나 더한다.

다른 사람 답

import numpy as np

def sumMatrix(A,B):
    A_np = np.array(A)
    B_np = np.array(B)
    result = A_np + B_np
    return result.tolist()

numpy를 이용해 행렬을 계산하고 list로 변환한다.

profile
시도하고 More Do하는 백엔드 개발자입니다.

0개의 댓글