[Python] sort()에서의 key lambda 사용하기

 

파이썬의 sort() 함수는 오름차순으로 정렬해주는 함수입니다. 문자열 또한 오름차순으로 정렬이 됩니다.

sort(reverse=True) 또는 reverse()를 이용하여 내림차순으로도 구현이 가능합니다.

 

이 글에서는 key 매개변수를 가지는 sort()에 대해 알아보겠습니다.

 

key 매개변수를 가지는 sort() 함수는 key 값을 기준으로 정렬되고 기본값은 오름차순입니다. 또한 lambda식을 사용할 수 있습니다.

 

백준 1181번 문제를 예시로 들어보겠습니다.

 

 

이 문제 같은 경우 sort(key)를 이용하여 쉽게 구할 수 있습니다.

 

data_list = ['but','i','wont','hesitate','no','more','no','more','it','cannot','wait','im','yours']

#중복 제거
data_list = list(set(data_list))

data_list.sort()
data_list.sort(key=lambda x : len(x))

print(data_list)

 

출력 값에 대해선 for 문 돌려주시면 됩니다.

 

lambda식을 이용하여 2차원 리스트에 대해서도 가능합니다.

 

data_list = ['but','i','wont','hesitate','no','more','no','more','it','cannot','wait','im','yours']

for index in range(len(data_list)) : 
    data_len = len(data_list[index])
    data_list[index] = (data_list[index], data_len)

data_list.sort(key = lambda x :(x[1], x[0]))
print(data_list)

이런 식으로도 문제 풀이가 가능합니다.

 

https://velog.io/@aonee/Python-%EC%A0%95%EB%A0%AC-sort-sorted-reverse