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

문제설명

2차원 행렬 2개를 입력받아 그 행렬들의 곱을 구하는 문제입니다.

실행과정

1. 두 행렬을 입력받습니다.
2. 두 행렬을 곱하면 행렬의 i행의 j열 원소는 행렬1의 i행의 원소와 행렬2의 j열의 원소들의 합이 되므로 각 곱들을 구해서 i행 j열 원소를 구합니다.
3. 연산이 모두 완료되면 두 행렬을 곱해서 만들어진 행렬을 출력합니다.

소스코드

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        answer.append([])
        for k in range(len(arr2[0])):
            answer[i].append(0)
            for j in range(len(arr1[i])):
                answer[i][k] += arr1[i][j] * arr2[j][k]

    return answer