[파이썬] 문제 : 3x3 A,B 행렬 입력 받고, 행렬 곱 계산(for문)

 

내용으로 보아 numpy를 사용한 계산이 아닌 for문을 사용한 계산을 해야하는 것으로 보입니다.

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

# A = [[2,3,2], [4,2,4], [3,1,4]]
# B = [[5,4,3], [2,4,1], [3,1,1]]
A = []
for i in range(3):
    tmp = []
    for j in range(3):
        tmp.append(int(input()))
    A.append(tmp)

B = []
for i in range(3):
    tmp = []
    for j in range(3):
        tmp.append(int(input()))
    B.append(tmp)
    
rtn = solution(A, B)
print(A)
print(B)
print(rtn)