알고리즘 연습 [행렬의 곱셈]

23 Jan 2018

알고리즘

행렬의 곱셈

문제

행렬의 곱셈은, 곱하려는 두 행렬의 어떤 행과 열을 기준으로, 좌측의 행렬은 해당되는 행, 우측의 행렬은 해당되는 열을 순서대로 곱한 값을 더한 값이 들어갑니다. 행렬을 곱하기 위해선 좌측 행렬의 열의 개수와 우측 행렬의 행의 개수가 같아야 합니다. 곱할 수 있는 두 행렬 A,B가 주어질 때, 행렬을 곱한 값을 출력하는 productMatrix 함수를 완성해 보세요.

나의 풀이

def productMatrix(A, B):
    n = len(A)
    m = len(B)
    l = len(B[0])
    a = []
    b = []
    result = []
    for i in range(n):
        for j in range(l):
            for k in range(m):
                a.append(A[i][k] * B[k][j])
            b.append(sum(a))
            a = []
        result.append(b)
        b = []
    return result

n*l 크기의 행렬이 결과로 나와야 함

다른 풀이

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]

zip(*B) B의 원소끼리 zip()를 적용한 것과 같음
- B= [[1,2,],[3,4]]일 때 zip(*B)는 zip([1,2],[3,4])로 결과는 [(1,3),(2,4)]가 된다.
A_row 당 [B_col을 돌려서 원소 하나씩 곱한값을 합한 값] 을 return