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

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

 

def solution(arr1, arr2):
    arr3=[[0]*len(arr2[0]) for _ in range(len(arr1))]

    for i in range(len(arr1)):
        for j in range(len(arr2[0])):
            for k in range(len(arr1[0])):
                arr3[i][j] += (arr1[i][k]*arr2[k][j])

    return arr3

 

프로그래머스에서는 numpy 사용 가능