[Python] 2차원 리스트 원소 별로 합치는 방법

- 2차원 리스트를 원소 별로 합치는 방법

 

아래와 같이 (1, 3) shape의 2차원 리스트가 있다고 가정해보자.

a = [[1, 1, 1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1, 1, 1], [1, 0, 0, 1, 1, 0, 1, 0]]

 

우리가 원하는 답은 리스트 a의 원소 별로 합쳐서 

새로운 1차원 리스트를 만드는 것이다.

 

위 예제와 같은 경우에는

 

sum(a[0]) = 8
sum(a[1]) = 8
sum(a[2]) = 4

 

이므로 [8, 8, 4] 와 같은 output을 얻고 싶다.

 

이를 구하는 방법은 다음과 같다.

 

sum_a = list(map(sum, a))
print(sum_a)

>> [8, 8, 4]

 

output은 우리가 원하는 대로 [8, 8, 4]가 나온다.