[Python] 정렬 (Sort) : sort(), sorted(), reverse, key

<목차>

📁 파이썬의 정렬 라이브러리

✔️ 병합 정렬(merge sort)을 기반으로 만들어졌으며, 퀵 정렬보다는 느리지만 최악의 경우에도 시간 복잡도 O(NlogN)을 보장한다는 특징을 가진다. 

 

📁 sort() 와 sorted()

✔️ iterable 객체를 정렬해주는 파이썬의 내장함수(built-in function)

 

✔️ reverse 속성 : False - 오름차순 / True - 내림차순 

✔️ 오름 차순 : 숫자의 크기가 커진다. ex. 1, 2, 3, 4

array = [9, 1, 8, 5, 4]

array.sort() 

array
Out[4]: [1, 4, 5, 8, 9]

array = [9, 1, 8, 5, 4]

sorted(array)
Out[6]: [1, 4, 5, 8, 9]

# 리스트에 반영되어 있지 않다. 
array
Out[7]: [9, 1, 8, 5, 4]

# 다른 리스트에 저장을 해서 사용해야한다.
result = sorted(array)

result
Out[9]: [1, 4, 5, 8, 9]

 

✔️ 내림 차순 : 숫자의 크기가 작아진다. ex. 4, 3, 2, 1

array.sort(reverse=True)

array
Out[11]: [9, 8, 5, 4, 1]

 

📁 튜플을 원소로 하는 리스트 정렬

✔️ 파이썬에서 튜플을 원소로 하는 리스트가 있을 때, 그 리스트를 정렬하면 기본적으로 각 튜플을 구성하는 원소의 순서에 따라 정렬된다. 

     → 첫 번째 원소의 순서에 맞게 정렬하고, 첫 번째 원소의 값이 같은 경우 두 번째 원소의 순서에 맞게 정렬하며, 이 과정을 모든 원소에 반복 적용한다. 

a = [(5, 1, 5), (3, 1, 9), (3, 5, 5), (5, 1, 5)]

a.sort()

a
Out[12]: [(3, 1, 9), (3, 5, 5), (5, 1, 5), (5, 1, 5)]

 

✔️ key 속성에 값을 대입하여 내가 원하는 '조건'에 맞게 튜플을 정렬시킬 수 있다. 

✔️ key 속성 : 정렬 기준을 명시

 

💡 두 번째 원소 내림차순 → 세 번째 원소 오름차순 → 첫 번째 원소 오름차순

내림차순으로 정렬하기 위해 '-'를 붙여줄 수 있지만 이는 int 형에만 한함에 주의하자. 

a.sort(key = lambda x: (-x[1], x[2], x[0]))

a
Out[15]: [(3, 5, 5), (5, 1, 5), (5, 1, 5), (3, 1, 9)]

 

💡 두 번째 값을 기준으로 오름차순 정렬

sorted(a, key = lambda x: x[1])
Out[16]: [(5, 1, 5), (5, 1, 5), (3, 1, 9), (3, 5, 5)]

 

---

Ref.

- 이것이 코딩테스트다 with 파이썬