[프로그래머스] 행렬의 곱셈

https://programmers.co.kr/learn/courses/30/lessons/12949

코딩테스트 연습 - 행렬의 곱셈

 

numpy 가 통하는지 확인해봤다. 된다.

 

import numpy as np

def solution(arr1, arr2):
    answer = np.array(arr1).dot(np.array(arr2))
    
    return answer.tolist()
    
print(solution([[1, 4], [3, 2], [4, 1]], [[3, 3], [3, 3]])) # [[15, 15], [15, 15], [15, 15]]