[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]가 나온다.
2023. 1. 25.
