[Python] 두 배열의 각 자리 합 리스트

[Python] 두 배열의 각 자리 합 리스트

Produce list which is a sum of two lists, item-wise [duplicate]

---

 

💁 하고 싶은 작업

두 리스트 같은 자리(index) 값을 합한 리스트를 구하고 싶은 경우

a=[1,2,3,4,5]
b=[5,4,3,2,1]

# result
[6,6,6,6,6]

 

🚀 해결 방법

방법 1

 

>>> a = [1,2,3,4,5]
>>> b = [5,4,3,2,1]
>>> c = [x+y for x,y in zip(a, b)]
>>> c
[6, 6, 6, 6, 6]

 

 방법 2

>>> c = [a[i] + b[i] for i in range(len(a))]
>>> c
[6, 6, 6, 6, 6]

 

---

참고

리스트 안에 for문을 포함하는 리스트 내포(List comprehension)를 이용

⇒ 1) 과 2) 는 같은 결과를 나타낸다.

1)
>>> c = [aa+bb for aa in a for bb in b]

2)
>>> for aa in a:
>>>     for bb in b:
>>>           aa+bb