[프로그래머스] 행렬의 곱셈 (파이썬)

728x90
반응형

하나를 행열 위치를 바꾼 행렬로 만들어서 각 행 별로 원소를 곱해서 더해줌 

def solution(arr1, arr2):
    r2=list(map(list,zip(*arr2)))
    answer = [[sum(a*b) for a,b in zip(row,col) for col in r2] for row in arr1]
    return answer

 

728x90
반응형
TAGS.

Comments