코딩테스트 연습 - 행렬덧셈
코딩테스트 연습 - 행렬덧셈
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()
Author And Source
이 문제에 관하여(코딩테스트 연습 - 행렬덧셈), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@ghenmaru/코딩테스트-연습-행렬덧셈저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)