[프로그래머스] 행렬의 덧셈 Lv.1 - [파이썬/python]
https://school.programmers.co.kr/learn/courses/30/lessons/12950
# 프로그래머스 | 행렬의 덧셈
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
'코딩테스트' 카테고리의 다른 글
[프로그래머스] 이상한 문자 만들기 - Lv.1 (20) | 2023.08.17 |
---|---|
[프로그래머스] 최대공약수와 최소공배수 - Lv.1 (0) | 2023.08.17 |
[프로그래머스] 문자열 나누기 - Lv.1 (15) | 2023.08.16 |
[프로그래머스] 성격 유형 검사하기 - Lv.1 (15) | 2023.08.16 |
[프로그래머스] 가장 가까운 같은 글자 - Lv.1 (15) | 2023.08.16 |