Python 2.7转3：sorted函数自定义比较与键选择的迁移优化咨询

嘿，这个迁移场景我之前在项目里也遇到过！你的思路方向是对的——Python 3确实移除了sorted()的cmp参数，必须用functools.cmp_to_key()来适配自定义比较逻辑，但你的写法有可以优化的地方，尤其是在效率和Python 3风格上。

首先，先明确原Python 2代码的逻辑：它会先给每个元素调用一次custom_key_getter生成键，再用custom_compare比较这些键来排序。但你的迁移写法里，每次比较两个元素时都会重复调用custom_key_getter(x)和custom_key_getter(y)，如果列表很大，这会导致大量重复计算，效率拉胯。

更优且符合Python 3风格的实现有两种方向，取决于你的custom_compare是否能被简化：

1. 如果custom_compare可以用key逻辑替代（优先选这个！） #

如果你的custom_compare只是简单的大小/顺序比较（比如数值升序降序、字符串字典序等），那完全不需要用cmp_to_key，直接用key参数就够了，这是Python 3最推荐的高效写法：

# 比如custom_compare是普通的升序比较，直接用key即可
sorted(some_list, key=custom_key_getter)
# 如果是降序，加reverse=True
sorted(some_list, key=custom_key_getter, reverse=True)

这种方式每个元素只计算一次key，排序效率最高，也是Python 3设计时鼓励的写法。

2. 必须用自定义custom_compare的情况 #

如果custom_compare是复杂的自定义逻辑（比如多条件优先级比较、非标准排序规则），那我们可以先预处理生成带key的元素对，确保每个key只计算一次，再用cmp_to_key比较：

import functools

# 用生成器表达式预处理，避免额外内存开销
sorted_with_keys = sorted(
    ((custom_key_getter(item), item) for item in some_list),
    key=functools.cmp_to_key(lambda a, b: custom_compare(a[0], b[0]))
)
# 提取排序后的原元素
result = [item for _, item in sorted_with_keys]

这里用生成器表达式而不是列表推导式，是因为生成器不需要一次性把所有(key, item)对存入内存，对于超大列表更友好。而且每个元素的custom_key_getter只调用一次，比你的原迁移写法效率高很多。

补充：你的原写法没问题，但效率有短板 #

你的迁移代码逻辑是正确的，能得到和Python 2一致的结果，但每次比较都会重复计算key，当列表元素多或者custom_key_getter计算成本高时，性能差异会很明显。

内容的提问来源于stack exchange，提问作者Jason Mond