코딩테스트 연습 - 행렬덧셈

2911 단어 numpy행렬numpy

코딩테스트 연습 - 행렬덧셈

2개의 행렬을 입력받아 행렬 덧셈의 결과를 반환하라.

# 정답코드

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

*numpy

행렬연산을 할 때 numpy 모듈을 이용하면 훨씬 더 쉽게 계산할 수 있다.
numpy.matrix() 또는 numpy.array() 를 사용하면 타입이 다르기 때문에 tolist()를 통해 list 타입으로 바꿔줘야 한다.

import numpy as np
def sumMatrix(A,B):
    A=np.array(A)
    B=np.array(B)
    answer=A+B
    return answer.tolist()

좋은 웹페이지 즐겨찾기