[Programmers][Python] Lv.2 행렬의 곱셈

Overview

• 체감 난이도: ★☆☆☆☆
• 문제 레벨: Lv.2 
• 문제 유형: 구현
• 풀이 상태: 답안참고 / 스스로 해결
• 추후: 다시 풀어보기 / 간단 복습 / 완벽 이해 

---

[문제]

https://school.programmers.co.kr/learn/courses/30/lessons/12949

프로그래머스

 

 

[코드]

(처음 풀이한 코드)

def solution(A, B):
    answer = []
    for i in range(len(A)):
        
        row = []
        for j in range(len(B[0])):
            
            tmp = 0
            for k in range(len(A[0])):
                tmp += A[i][k] * B[k][j]
            row.append(tmp)
            
        answer.append(row)
    return answer

 

(다른 방법의 코드)

list comprehension

def solution(A, B):
    return [[sum(a * b for a, b in zip(row, col)) 
             for col in zip(*B)] 
             for row in A]

 

map, zip 

def solution(A, B):
    return list(map(
        lambda row: list(map(
            lambda col: sum(map(lambda x: x[0] * x[1], zip(row, col))),
            zip(*B)
        )),
        A
    ))

 

NumPy 

import numpy as np

def solution(A, B):
    return np.dot(A, B).tolist()