functools.cmp_to_key로 사용자 지정 순서 만들기

어떤 정렬 순서는 각 항목의 값 하나가 아니라 두 항목을 함께 비교해야 결정됩니다. 이런 경우에는 key 대신 비교 함수를 직접 작성합니다.

비교 함수는 두 항목 a와 b를 받습니다. a가 먼저 와야 하면 음수를, b가 먼저 와야 하면 양수를, 동률이면 0을 반환하십시오.

Python의 sort는 key만 받으므로, 비교 함수를 사용할 수 있는 key로 바꾸려면 functools.cmp_to_key로 감싸야 합니다.

숫자를 이어 붙여 가장 큰 문자열을 만들려면 어떤 두 숫자를 이어 붙인 결과가 더 큰지 비교해야 합니다. 단순한 key로는 이를 표현할 수 없습니다.

a+b를 이어 붙인 결과가 더 크면 이 표현식은 -1을 반환하므로 a가 b보다 앞에 정렬됩니다. 이렇게 해서 만들 수 있는 가장 큰 수를 구성합니다.

정렬은 결과의 부호만 확인합니다. -2나 -100을 반환해도 -1을 반환한 것과 같으므로 정확한 크기에 신경 쓰지 않아도 됩니다.

비교 함수는 일관적이어야 합니다. a가 b보다 앞서고 b가 c보다 앞선다면 a도 c보다 앞서야 합니다. 모순이 생기면 정렬 순서가 엉망이 됩니다.

key는 항목마다 한 번 계산되지만, 비교 함수는 여러 쌍에 대해 실행되므로 더 느립니다. key로 해결할 수 없는 경우에만 cmp_to_key를 사용하십시오.

숫자 비교에서는 a - b를 그대로 반환해도 됩니다. 그 부호에 이미 순서 정보가 담겨 있기 때문입니다. 정수에서는 짧고 안전한 방법입니다.

실수에서는 a - b가 아주 작은 값이나 0으로 반올림될 수 있습니다. 잘못된 동률 판정을 피하려면 명시적인 분기문으로 비교하십시오.

이어 붙이기 순서, 사용자 지정 순위, 규칙에 따른 동률 처리까지 모두 구현할 수 있습니다. 까다로운 정렬에서는 비교 함수가 해결책이 됩니다.

비교 함수가 cmp(a, b)에 대해 음수를 반환합니다.

key로 표현할 수 없는 순서는 비교 함수를 cmp_to_key로 감싸서 구현할 수 있습니다. 부호를 반환하고 일관성을 유지하며, 가능하면 key를 우선하십시오. 🧩