AI 프로그래밍

행렬의 덧셈

import numpy as np
def solution(arr1, arr2):
    answer=np.array(arr1)+np.array(arr2)
    return answer.tolist()

행렬의 곱셈

def solution(arr1, arr2):
    answer=[[0]*len(arr2[0]) for _ in range(len(arr1))]
    for i in range(len(arr1)): 
        for j in range(len(arr2[0])):
            for k in range(len(arr1[0])):
                answer[i][j]+=(arr1[i][k]*arr2[k][j])
    return answer
    
    #i*k+k*j=arr1*arr1[0]+arr1[0]+arr2[0]=>행렬의 최종 크기(arr1,arr2[0])