[Programmers] LV 1. 행렬의 덧셈 (Python/파이썬)

---

▶ 문제 설명

행렬의 덧셈은 행과 열의 크기가 같은 두 행렬의 같은 행, 같은 열의 값을 서로 더한 결과가 됩니다. 2개의 행렬 arr1과 arr2를 입력받아, 행렬 덧셈의 결과를 반환하는 함수, solution을 완성해주세요.

---

▶ 제한 조건

• 행렬 arr1, arr2의 행과 열의 길이는 500을 넘지 않습니다.

---

▶ 입출력 예

arr1	arr2	return
[[1,2],[2,3]]	[[3,4],[5,6]]	[[4,6],[7,9]]
[[1],[2]]	[[3],[4]]	[[4],[6]]

---

▶  Code

 

def solution(arr1, arr2):
    answer = arr1
    
    for i in range(len(arr1)):
        for j in range(len(arr1[i])):
            answer[i][j] = arr1[i][j] + arr2[i][j]
    return answer
                                        
print(solution([[1,2],[2,3]],[[3,4],[5,6]]))

 

▶  Point

리스트 두 개가 주어지고 각 리스트의 같은 번쨰 수를 더한 리스트를 출력해 주면 된다.

for i in range(len(arr1)):

반복해서 더해줘야 한다는 개념을 생각하여 for문을 사용함. for문을 통해 처음 받은 리스트의 길이만큼 반복을 실행해 준다. 2중 리스트를 쓰고 있으므로 리스트 안에 리스트가 몇 개가 있는지 확인하고 그 수 만큼 반복한다.

for j in range(len(arr1[i])):

2중 리스트에서 2번째 리스트 값 만큼 반복을 해주기 위해 for 문을 만들어 준다.

arr1[i][j] + arr2[i][j]

각 자리수 별로 더해주고

return answer

리턴을 해주면 끝이다.