문제 설명

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

제한 조건

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

풀이

def solution(arr1, arr2):
    answer = []
    
    for elem1 in arr1:
        tmp = []
        
        for elem2 in zip(*arr2):
            result = sum([e1 * e2 for e1, e2 in zip(elem1, elem2)])
            tmp.append(result)
        answer.append(tmp)
        
    return answer