[알고리즘] 특정 기준으로 리스트 정렬하기 (python) - key=lambda

요즘 정렬 문제를 많이 풀다보면, 리스트를 특정 기준으로 정렬해야할 때가 많다. 단순히 .sort(), sorted()로의 오름차순 내림차순으로는 풀 수 없는 문제도 많이 만났고, 이럴때는 key=lambda 함수를 사용하면 된다! 

오늘은 이러한 용법에 대해 포스팅 해보도록 하겠다. :) 이 게시글은 여기를 참고하였다. 

 

# key=lambda 용법 

우선 key=lambda 용법은 sorted 함수의 인자로 쓰인다. 그리고 다음과 같이 사용한다. 

# key가 하나일 때 

모든 정렬의 기반은 오름차순 정렬이다. 

ex 1) 리스트의 각 원소를 기준으로 정렬하기 (일반 오름차순) 

참고로 문자열을 다음과 같이 정렬하면 사전순으로 정렬이 된다. 

arr = ['abc', 'bac', 'bca']
sorted(arr, key=lambda x : x)

 

ex 2) 리스트 각 원소의 첫글자를 기준으로 정렬하기

리스트의 각 원소를 x로 받는다고 가정하면 다음과 같이 정렬할 수 있다. 

arr = ['abc', 'bac', 'bca']
sorted(arr, key=lambda x : x[0])

 

ex 3) 리스트 내 요소의 인덱스를 원소의 크기순으로 정렬하기 

즉, 리스트 내 원소를 크기순으로 정렬했을 때의 인덱스가 어떻게 정렬되는지가 필요할 때가 있다. 예를들어 A리스트의 크기 순서대로 똑같이 맞추어 B 리스트를 정렬하고자 할 때는, A 리스트의 인덱스 정렬 정보가 필요할 것이다. 

이때는 다음과 같이 해주면 된다. 

b = [12, 14, 23, 24, 16]
b_idx = sorted(range(len(b)), key = lambda k: b[k]) 

# 여기서 b_idx는 [0, 1, 4, 2, 3]이 된다.

 

 

# key가 두개일 때 

정렬할 때 특정 정렬 기준으로 같을경우, 다른 기준을 적용하고 싶을 때가 있다. 이럴때는 다음과 같이 하면 된다. 

ex1) x[0]를 기준으로 정렬하고 같을 경우 x[1]를 기준으로 정렬하기

arr = ['abb', 'acc', 'bcd']
sorted(arr, key=lambda x : (x[0], x[1]))

 

ex2) x[0]는 내림차순, x[1]는 오름차순을 기준으로 정렬하기

arr = ['abb', 'acc', 'bcd']
sorted(arr, key=lambda x : (-x[0], x[1]))

 

 

# 관련 문제 

1026 보물 https://www.acmicpc.net/problem/1026

1181 단어 정렬 https://www.acmicpc.net/problem/1181