python sorted函数自定义排序

一般使用sorted函数的时候,直接会将key设置为一个lambda表达式的匿名函数。这样一般可以直接对字典的key或者value进行排序。

但是今天刷leetcode的时候,需要对字典value的数组进行逐一排序,当value一致时,按key的字母顺序排序。这时候就需要自定义比较方法了。

通过functools中的com_to_key将一个函数转变为参数key的值。

https://leetcode-cn.com/problems/rank-teams-by-votes/

def cmp(self, x, y):
		# x, y 为items中的单个item, x[0]为key, x[1]为数组
		# 对x和y的数组进行逐位比较,当value完全一致时,再比较key的字母顺序
        for a,b in zip(x[1],y[1]):
            if a > b:
                return 1
            elif b > a:
                return -1
        if x[0] < y[0]:
            return 1
        else:
            return -1

from functools import cmp_to_key
s = sorted(dic.items(), key=cmp_to_key(cmp), reverse=True)


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章