[프로그래머스] 행렬의 덧셈

코딩테스트 연습 - 행렬의 덧셈

# 2개의 행렬의 각 행과 각 렬을 더하는 문제이다. 

# zip 함수를 이용하여 각 배열의 인덱스 끼리 매칭시켰다.

a = [1,2,3]
b = [4,5,6]
for i in zip(a,b):
    print(i)
   
(1, 4)
(2, 5)
(3, 6)

 

def solution(arr1, arr2):
    array = []
    answer = []
    for i,j in zip(arr1,arr2):  # 자료형을 튜플로 묶는 함수
        #print(i,j)
        for k,l in zip(i,j):
            #print(k,l)

            array.append(k+l)

        answer.append(array)
        array = []
    return answer

# zip 함수를 이용하면 간편하게 출력값을 묶 것이 가능했다.

 

 

 

* 한 줄로 작성할 수도 있어서 확인했다.

def solution(arr1, arr2):
    return [[k+l for k,l  in zip(i,j)] for i,j in zip(arr1,arr2)]