Notice
Recent Posts
Recent Comments
Link
«   2025/12   »
1 2 3 4 5 6
7 8 9 10 11 12 13
14 15 16 17 18 19 20
21 22 23 24 25 26 27
28 29 30 31
Archives
Today
Total
관리 메뉴

보랑이의 개발일지

[python] sum함수로 2차원 배열->1차원으로 차원 축소하기 본문

Tech

[python] sum함수로 2차원 배열->1차원으로 차원 축소하기

boyeonning 2023. 9. 18. 16:51

프로그래머스 최소직사각형 풀이를 보는 도중 sum함수에 매개 변수로 []를 준 것을 보게 되었다.

solution = lambda sizes: max(sum(sizes, [])) * max(min(size) for size in sizes)

한번 찍어서 어떻게 동작하는지 보자!

sizes=[[60, 50], [30, 70], [60, 30], [80, 40]]
print(sum(sizes,[]))

>> [60, 50, 30, 70, 60, 30, 80, 40]

위 코드에서는 sum 함수를 사용하여 matrix 리스트의 모든 하위 리스트를 하나의 리스트로 합치고 있습니다. 
이렇게 하면 2차원 리스트를 1차원 리스트로 축소할 수 있습니다. 

만약 더 높은 차원의 리스트를 축소하려면 sum 함수를 중첩해서 사용하면 됩니다.

sizes=[[[60, 50], [30, 70]], [[60, 30], [80, 40]]]
print(sum(sizes,[]))

>> [[60, 50], [30, 70], [60, 30], [80, 40]]