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

행렬의 곱셈

문제

2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한사항

• 행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
• 행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
• 곱할 수 있는 배열만 주어집니다.

풀이

arr2의 행, 열을 바꿔서 계산한다

def solution(arr1, arr2):
    answer = []
    arr2 = [list(i) for i in zip(*arr2)]
    for arr in arr1:
        row = []
        for col in arr2:
            tmp = sum([a * b for a, b in zip(arr, col)])
            row.append(tmp)
        answer.append(row)
    return answer