python1-python3中sort函数key如何对两个参数做对比

  • python3之后不支持cmp,所用key函数并不直接比较任意两个原始元素,而是通过key函数把那些元素转换成一个个新的可比较对象,也就是元素的key,然后用元素的key代替元素去参与比较。如果原始元素本来就是可比较对象,比如数字、字符串,那么不考虑性能优化可以直接sort(key=lambda e: e)。不过这种基于key函数的设计倾向于每个元素的大小有个绝对标准,但有时却会出现单个元素并没有一个绝对的大小的情况,此时可以使用 functools.cmp_to_key构建基于多个元素的比较函数。

  • 以一个leetcode上的题为例:

    179. 最大数
    给定一组非负整数 nums,重新排列每个数的顺序(每个数不可拆分)使之组成一个最大的整数。
    注意:输出结果可能非常大,所以你需要返回一个字符串而不是整数。
    
    示例 1:
    输入:nums = [10,2]
    输出:"210"
    
    示例 2:
    输入:nums = [3,30,34,5,9]
    输出:"9534330"
    
    示例 3:
    输入:nums = [1]
    输出:"1"
    
    示例 4:
    输入:nums = [10]
    输出:"10"
     
    提示:
    1 <= nums.length <= 100
    0 <= nums[i] <= 109
    

    可以看到在这道题中列表nums中两个值的相对位置并不能由单一num决定,而是说 x与y拼接比y与x拼接的值大,那么就用[x,y]的顺序,否则用[y,x]的顺序。此时就是所谓的:单个元素并没有一个绝对的大小的情况

  • 要解决这道题用sort也很简单:

    from functools import cmp_to_key
    
    class Solution:
    
        def largestNumber(self, nums: List[int]) -> str:
            nums.sort(key=cmp_to_key(lambda x,y: int(str(y)+str(x)) - int(str(x)+str(y))))
            ans = ''.join([str(num) for num in nums])
            return str(int(ans))
    

    或者

    from functools import cmp_to_key
    
    def auxComp(x, y):
        if int(str(x)+str(y)) > int(str(y)+str(x)):
            return -1
        elif int(str(x)+str(y)) < int(str(y)+str(x)):
            return 1
        else:
            return 0
    
    class Solution:
    
        def largestNumber(self, nums: List[int]) -> str:
            nums.sort(key=cmp_to_key(auxComp))
            ans = ''.join([str(num) for num in nums])
            return str(int(ans))
    

    至于cmp_to_key中函数的条件可以这样理解:

    sort本身是升序,而题目要求是降序,因此需要cmp_to_key中反着写(或者加上reverse=True),也就是解法中的:

    lambda x,y: int(str(y)+str(x)) - int(str(x)+str(y))
    

    以及

    def auxComp(x, y):
        if int(str(x)+str(y)) > int(str(y)+str(x)):
            return -1
        elif int(str(x)+str(y)) < int(str(y)+str(x)):
            return 1
        else:
            return 0
    
  • 一句话说:python3中一些接受key的函数中(例如sorted,min,max,heapq.nlargest,itertools.groupby),key仅仅支持一个参数,无法实现两个参数之间的对比。采用cmp_to_key 函数,可以接受两个参数,对两个参数做处理,比如做和做差,转换成一个参数,就可以应用于key关键字了。

参考:

怎么理解Python的cmp_to_key函数?

python3 为什么取消了sort方法中的cmp参数?

一句话理解cmp_to_key函数

python sort函数内部实现原理

  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

tensor_zhang

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值