파이썬 알고리즘-108 (프로그래머스) 행렬의 곱셈

jiffydev·2021년 2월 6일
0

Algorithm

목록 보기
115/134

코드

def solution(arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        lst=[]
        for j in range(len(arr2[0])):
            part_sum=0
            for k in range(len(arr1[i])):
                part_sum+=arr1[i][k]*arr2[k][j]
            lst.append(part_sum)
        answer.append(lst)
    return answer

다른 사람의 풀이

def productMatrix(A, B):
    return [[sum(a*b for a, b in zip(A_row,B_col)) for B_col in zip(*B)] for A_row in A]
profile
잘 & 열심히 살고싶은 개발자

0개의 댓글