排序使用自定义比较的键功能

时间:2016-10-15 00:08:52

标签: python python-3.x sorting functional-programming comparison

Python 3 removes the cmp parameter to sorting functions

  

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函数替换,可以吗?

1 个答案:

答案 0 :(得分:4)

使用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']

Python 3 sorting functions take a ‘key’ parameter

  

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

functools.cmp_to_key助手为designed to help you transition to that style

  

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

有关此操作和其他技巧,请参阅Sorting HOWTO