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

프로그래머스_행렬의 곱셈

arr1, arr2의 길이가 100이 넘지 않기 때문에 3중 for문으로도 가능한 문제였다

def solution(arr1, arr2):
    answer = []
    n1 = len(arr1)
    m1 = len(arr1[0])
    n2 = len(arr2)
    m2 = len(arr2[0])
    
    for i in range(n1):
        tmp_list = [0 for _ in range(m2)]
        for j in range(m1):
            for k in range(m2):
                tmp_list[k] += (arr1[i][j]*arr2[j][k])
        answer.append(tmp_list)
            
    return answer