Python sorted()函数的高级使用（多条件）

2024-07-12 21:26| 来源: 网络整理| 查看: 265

Python中字典根据多项规则排序

在SQL中或者用pandas可以轻松地对一列数据按照多个规则进行排序，那么有没有办法直接用python的sorted实现呢？答案是肯定的。

mydict = {'b':10, 'c':10, 'a':10, 'd':20} 一、排序常规操作字典根据key排序 sorted(mydict.iteritems(), key=lambda x:x[0]) # [('a', 10), ('b', 10), ('c', 10), ('d', 20)] 字典根据value排序 sorted(mydict.iteritems(), key=lambda x:x[1]) # [('a', 10), ('c', 10), ('b', 10), ('d', 20)] 字典先按key排序，然后按value排序 sorted(mydict.iteritems(), key=lambda x:(x[0],x[1])) # [('a', 10), ('b', 10), ('c', 10), ('d', 20)] 字典先按value排序，然后按key排序 sorted(mydict.iteritems(), key=lambda x:(x[1],x[0])) # [('a', 10), ('b', 10), ('c', 10), ('d', 20)] 字典先按value升序排序，然后按key降序排序 sorted(mydict.iteritems(), key=lambda x:(x[1],-ord(x[0]))) # [('c', 10), ('b', 10), ('a', 10), ('d', 20)] 字典先按value降序排序，然后按key升序排序 sorted(mydict.iteritems(), key=lambda x:(x[1],-ord(x[0])), reverse=True) # [('d', 20), ('a', 10), ('b', 10), ('c', 10)] 二、使用itemgetter()

itemgetter()有时也可以使用lambda表达式来替代

from operator import itemgetter sorted(mydict.iteritems(), key=lambda x:(x[1],x[0])) sorted(mydict.iteritems(), key=itemgetter(x[1],x[0])) # [('c', 10), ('b', 10), ('a', 10), ('d', 20)]

根据以上多条件排序可以发现，条件函数将两个条件使用tuple进行传递（条件的括号是不是tuple我没有细细研究，但是可以这样理解），所以根据这一特性可以实现列表中嵌套字典进行排序

三、列表字典嵌套排序 rows = [ {'fname': 'Brian', 'lname': 'Jones', 'uid': 1003}, {'fname': 'David', 'lname': 'Beazley', 'uid': 1002}, {'fname': 'John', 'lname': 'Cleese', 'uid': 1001}, {'fname': 'Big', 'lname': 'Jones', 'uid': 1004} ] 根据fname进行排序 rows_by_fname = sorted(rows, key=itemgetter('fname')) [ {'fname': 'Big', 'uid': 1004, 'lname': 'Jones'}, {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'}, {'fname': 'David', 'uid': 1002, 'lname': 'Beazley'}, {'fname': 'John', 'uid': 1001, 'lname': 'Cleese'} ] itemgetter()函数也支持多个keys rows_by_lfname = sorted(rows, key=itemgetter('lname','fname')) [ {'fname': 'David', 'uid': 1002, 'lname': 'Beazley'}, {'fname': 'John', 'uid': 1001, 'lname': 'Cleese'}, {'fname': 'Big', 'uid': 1004, 'lname': 'Jones'}, {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'} ] 使用lambda的写法 rows_by_lfname = sorted(rows, key=lambda item:(item['uid'],item['lname'])) [ {'fname': 'David', 'uid': 1002, 'lname': 'Beazley'}, {'fname': 'John', 'uid': 1001, 'lname': 'Cleese'}, {'fname': 'Big', 'uid': 1004, 'lname': 'Jones'}, {'fname': 'Brian', 'uid': 1003, 'lname': 'Jones'} ]

sorted() 会遍历传入的列表，所以key根据遍历的元素item进行排序判断

四、扩展

使用itemgetter()方式会快一些，如果对性能要求比较高，可以优先使用itemgetter()方式

itemgetter() 同样使用于min()和max()等函数 min(rows, key=itemgetter('uid')) {'fname': 'John', 'lname': 'Cleese', 'uid': 1001} max(rows, key=itemgetter('uid')) {'fname': 'Big', 'lname': 'Jones', 'uid': 1004}

参考：https://blog.csdn.net/yxjwhhhh/article/details/103957414

【本文地址】

公司简介

联系我们