행렬 곱 연산

 

import numpy as np

def solution(arr_list):
    answer = np.array([])
    A = np.array([[0]])
    for i in arr_list:
        if len(np.shape(A))==len(np.shape(i)) and np.shape(A)[1]==np.shape(i)[0]:
            A = (A+1).dot(i*2)
    return A