[Level 2] 행렬의 곱셈

[문제]

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

 

 

[제한조건]

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

 

 

 

[코드]

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        arr_list = []
        for j in range(len(arr1[i])):
            arr_list.append(arr1[i][j])
        
        sum_list = []
        for x in range(len(arr2[0])):
            sum = 0
            for y in range(len(arr2)):
                sum += arr_list[y] * arr2[y][x]
            sum_list.append(sum)
        answer.append(sum_list)
    
    return answer