[Python] 두 리스트(배열) 각 요소들의 값 더하기



1. 원하는 동작 및 실제 동작

list1 = [1,1,1,1,1]
list2 = [1,1,1,1,1]

list1 + list2

# 원하는 result
[2,2,2,2,2]

# 실제 결과
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1]

python 에서 두 리스트의 같은 인덱스에 있는 인자들의 값을 더해주고 싶을 때가 있습니다. 그럴 때, 위와 같이 두 리스트를 더해주면 리스트가 연장되게 됩니다.

2. 해결 방법

아래와 같이 2개의 해결 방법이 있습니다. 해당 방법을 이용하여 손쉽게 문제를 해결할 수 있습니다.

2.1. list comprehension 을 사용하기

[list1[i] + list2[i] for i in range(len(list1))]

2.2. zip 함수를 사용하기

[x+y for x,y in zip(list1, list2)]