将自定义比较与python 3中的键选择相结合

Question

我正在将一堆 python 2.7 代码迁移到 python 3，它在 python 2.7 中具有这样的模式：

sorted(some_list, cmp=custom_compare, key=custom_key_getter)

我能想到的最好的迁移方法如下：

sorted(some_list, 
       key=functools.cmp_to_key(
             lambda x,y: custom_compare(custom_key_getter(x), 
                                        custom_key_getter(y)))

有没有更好的方法？还是符合python 3的更惯用的方式？

Answer 1

作为权宜之计，我觉得还可以。

如果您有 custom_compare 和 custom_key_getter 的多个组合，我会将 lambda 分解为

def compare_using(custom_compare, custom_key_getter):
   def comparator(x, y):
     return custom_compare(custom_key_getter(x), custom_key_getter(y))
   return comparator

如果您只有几个组合，我可以想象将它们预先计算，例如

compare_names_special = compare_using(compare_special, itemgetter('name'))
...
customers_by_name = sorted(customers, compare_names_special)

性能不应该受到影响，因为这段代码只是展示了 Python 2 的排序所做的事情。

从长远来看，我会考虑重写自定义比较函数，这样它们就不需要cmp_to_key 处理；在大多数情况下，它应该是微不足道的。