코딩테스트
[프로그래머스] 행렬의 덧셈 - 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