Programmers - 행렬의 곱셈

문제 링크

a행과 b열을 각각 list로 만들고 곱한 결과를 return하는 함수를 만들어서 처리

def solution(arr1, arr2):

    # arr1의 a-1행과 arr2의 b-1열의 곱
    def multiple(a, b):
        row = arr1[a]
        col = list(map(lambda x: x[b], arr2))
        ret = 0
        for i in range(len(row)):
            ret += row[i]*col[i]
        return ret

    answer = []

    for i in range(len(arr1)):
        answer.append([])
        for j in range(len(arr2[0])):
            answer[i].append(multiple(i, j))

    return answer