코딩테스트

[프로그래머스] 행렬의 덧셈 - Lv.1

pyflu 2023. 8. 17. 04:00

[프로그래머스] 행렬의 덧셈 Lv.1 - [파이썬/python]

 

https://school.programmers.co.kr/learn/courses/30/lessons/12950

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr


 

 


# 프로그래머스 | 행렬의 덧셈
def solution(arr1, arr2):
   
    result = []  # 최종 결과를 담을 리스트 초기화

    # 두 행렬의 같은 위치에 있는 원소들을 더하기 위해 zip() 함수 사용
    # a = [1,2], b = [3,4] / a = [2,3], b = [5,6]
    for a, b in zip(arr1, arr2):
        sum = []  # 현재 행의 결과를 담을 리스트 초기화

        # 현재 행의 같은 위치에 있는 원소들을 더해서 result 리스트에 추가
        # i = 1, j = 3 / i = 2, j = 4
        for i, j in zip(a, b):
            sum.append(i + j)
       
        # 현재 행의 sum 리스트를 result 리스트에 추가
        result.append(sum)

    return result

728x90