[프로그래머스] Level 2_행렬의 곱셈(연습문제)

📌문제 유형

연습문제

 

📌문제

프로그래머스

 

📌나의 문제풀이

- numpy 모듈 사용, @(행렬 곱)연산 수행

- 행렬 곱

[Numpy]행렬곱(@)과 내적(dot) 그리고 별연산(*)

- numpy to list, list to numpy

• list to numpy : np.array(list)
• numpy to list : list.tolist()

list를 numpy array로 바꾸고 numpy array를 list로 바꾸는 방법을 알아봅시다.

import numpy as np

def solution(arr1, arr2): 
    arr1 = np.array(arr1)
    arr2 = np.array(arr2)
    
    answer = arr1 @ arr2
    
    return answer.tolist()

 

📌다른 사람의 문제풀이

1) zip 이용

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]

2) 직관적인 코드

def productMatrix(A, B):
    answer = []
    for y1 in range(len(A)):
        a=[]
        for x2 in range(len(B[0])):
            n = 0
            for x1 in range(len(A[0])):
                n += A[y1][x1] * B[x1][x2]
            a.append(n)
        answer.append(a)
    return answer