정렬 순서를 커스터마이징: cmp_to_key

compare to key의 약자로 커스터마이징 된 비교 함수를 정렬에 사용할 수 있게 해주는 도구입니다.

 

Python에서는 기본적으로 sorted() 함수와 list.sort() 메서드에 key 인자를 사용하여 정렬 기준을 지정할 수 있는데, cmp_to_key를 사용하면 이 정렬 기준을 비교 함수 기반으로 정의할 수 있습니다.

 

from functools import cmp_to_key

 

사용법

from functools import cmp_to_key

# 문자열 길이를 비교하는 함수
def compare_len(s1, s2):
    return len(s1) - len(s2)

# 비교 함수를 key로 변환
key_func = cmp_to_key(compare_len)

# 문자열 리스트 정렬
words = ["apple", "banana", "pear", "kiwi", "grape"]
sorted_words = sorted(words, key=key_func)

print(sorted_words)
# ['kiwi', 'pear', 'apple', 'grape', 'banana']

 

이렇게 원하는 비교로 정렬을 할 수 있습니다