[프로그래머스] 행렬의 곱셈 - 파이썬

행렬의 곱셈

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

 

문제

코드

• for문을 3개 활용하여 행렬의 곱셈을 계산
• 곱해진 행렬의 크기는 (A x B) x (B x C) = (A x C)

def solution(arr1, arr2):

    answer = [[0 for _ in range(len(arr2[0]))] for _ in range(len(arr1))]

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

    return answer