[프로그래머스💯] 코딩테스트 연습 > 연습문제 > 행렬의 곱셈

<프로그래머스 문제풀이>

<출처 : 프로그래머스(Programmers)>

 

---

해답)

 

import numpy as np


def solution(arr1, arr2):
    answer = np.dot(np.array(arr1), np.array(arr2))
    return answer.tolist()	

 

풀이)

 

사실 직접 코딩으로도 구현할 수 있는데.

numpy라는 라이브러리는 다차원배열의 계산을 매우 편하게 해준다.

dot 혹은 matmul 을 통해 바로 곱셈을 구현할 수 있다.

tolist() 는 numpy 형태의 배열(혹은벡터) 를 다시 리스트 타입으로 만들어준다.

 

 

* 이 문제 및 로고의 저작권은 Programmers에 있습니다.

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges