[python] sort, sorted 함수에서 람다 표현식으로 조건 지정하기

람다 표현식이란?

람다 표현식은 식 형태로 되어 있다고 해서 람다 표현식(lambda expression)이라고 부른다. 특히 람다 표현식은 함수를 간편하게 작성할 수 있어서 다른 함수의 인수로 넣을 때 주로 사용한다.

 

---

정렬 함수에서 사용하기

 

s = ['2 A', '1 A', '4 C', '1 B']

라는 배열 s가 있을 때, sorted()로 정렬한 결과는 다음과 같을 것이다.

 

>> sorted(s)
['1 A', '1 B', '2 A', '4 C']

 

 

첫 번째 문자인 숫자가 아닌 두 번째 문자 ABC를 우선으로 기준으로 정렬하고 싶을 때는 sorted 함수 내부에 람다 표현식을 지정하여 정렬하면 된다.

 

>> sorted(s, key=lambda x: (x.split()[1], x.split()[0]))
['1 A', '2 A', '1 B', '4 C']