[python, 파이썬] sum 함수를 이용한 차원축소 및 병합

sum 함수를 이용한 차원축소 및 병합

반응형

• 여태까지 알던 파이썬 내장함수인 sum의 기능은 단순히 iterable한 객체들의 합을 리턴해주는 걸로만 알고있었다.

• 내부 docs를 봐도 아래처럼 start값을 지정해주면 iterable의 합과 start를 더해준다까지만 나와있을 뿐이다.

 

 

그런데 sum을 이용해 리스트의 차원을 낮추고 병합까지 하는 기능을 이용할 수 있다!

sum(다차원 리스트, [ ])를 통해 차원을 축소하고 병합까지 가능하다.

아래는 예시이다

 

arr = [[1, 0, 3], [4, 0, 6]]
print(sum(arr, []))

# 결과 : [1, 0, 3, 4, 0, 6]

arr = [[[1, 0, 3], [0, 1, 2]], [[4,5,6], [7,8,9]]]
print(sum(arr, []))

# 결과 : [[1, 0, 3], [0, 1, 2], [4, 5, 6], [7, 8, 9]]

빈리스트가 아닌 숫자를 넣을 경우 해당 값들이 제일 앞에 위치하게 된다