[Programmers] 행렬의 곱셈 (Python)

코딩 1일 1문제 오늘의 문제는 프로그래머스의 행렬의 곱셈 입니다

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

Solution

import numpy as np

def solution(arr1, arr2):
    answer = [[]]
    
    answer = (np.matrix(arr1)*np.matrix(arr2)).tolist()
    
    return answer

Solution 풀이

numpy의 matrix를 활용하여 각각의 리스트를 행렬로 만들어준 다음 이를 곱한다음 

다시 리스트로 변환한 값을 정답으로 return합니다.

SOMJANG/CODINGTEST_PRACTICE