[Level 2] 행렬의 곱셈

나며기·2021년 9월 20일
0

programmers

목록 보기
53/111
post-thumbnail
def solution(arr1, arr2):
    answer = [[0]*len(arr2[0]) for i in range(len(arr1))]
    arr2_t = [[0]*len(arr2) for i in range(len(arr2[0]))]
    
    for i in range(len(arr2)):
        for j in range(len(arr2[i])):
            arr2_t[j][i] = arr2[i][j]

    for i in range(len(answer)):
        for j in range(len(answer[i])):
            answer[i][j] = sum([arr1[i][k]*arr2_t[j][k] for k in range(len(arr1[i]))])
    
    return answer
profile
PLUS ULTRA

0개의 댓글