使用自定义比较的排序键函数

Question

Python3 参数到排序函数

builtin.sorted()和list.sort()不再接受提供比较函数的cmp参数。改用key参数。

对于可以通过检查序列中的单个项目(例如key=str.lower)来确定的订单来说，这是很好的。但是，如果定制订单必须有两个检查项目才能确定其订单，那又该怎么办呢？

$ python2
Python 2.7.12+ (default, Sep  1 2016, 20:27:38)
[…]

>>> digits = ['3', '30', '34', '32', '9', '5']
>>> sorted(
...         digits,
...         cmp=(lambda a, b: cmp(a+b, b+a)),
...         reverse=True)
['9', '5', '34', '3', '32', '30']

双参数cmp函数不能用单参数key函数代替，是吗？

Answer 1

使用functools.cmp_to_key助手。

>>> import functools
>>> digits = ['3', '30', '34', '32', '9', '5']
>>> sorted(
...         digits,
...         key=functools.cmp_to_key(lambda a, b: cmp(a+b, b+a)),
...         reverse=True)
['9', '5', '34', '3', '32', '30']

The Python 3排序函数采用“key”参数

key指定一个参数的函数，用于从每个列表元素中提取比较键：key=str.lower。默认值是None (直接比较元素)。

functools.cmp_to_key助手是旨在帮助您过渡到那种风格。

functools.cmp_to_key(func) 将旧式比较函数转换为关键函数.…此函数主要用作从Python 2转换的程序的转换工具，后者支持使用比较函数。

这适用于最新的Python 2和Python 3。

这个技巧是通过创建一个键函数来完成的，该函数接受一个项进行比较，并返回一个自定义对象，该对象知道如何按照您的比较函数指定的方式进行比较。

>>> key_func = functools.cmp_to_key(lambda a, b: cmp(a+b, b+a))
>>> key_func("32")
<functools.K object at 0x7f6781ce0980>
>>> key_func("32") < key_func("5")
True

有关此和其他技巧，请参见分类方法。