[Python / 프로그래머스 level 2] 행렬의 곱셈

import numpy as np

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