欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Python排序函数的使用方法详解

程序员文章站 2022-03-08 08:05:49
python排序函数完美体现了python语言的简洁性,对于list对象,我们可以直接调用sort()函数(这里称为"方法"更合适)来进行排序,而对于其他可迭代对象(如set,dict),我们可以使用...

python排序函数完美体现了python语言的简洁性,对于list对象,我们可以直接调用sort()函数(这里称为"方法"更合适)来进行排序,而对于其他可迭代对象(如set,dict),我们可以使用更灵活的sorted()函数。

一.list的sort()函数

python源码builtins.py文件对sort()函数的定义如下

 def sort(self, key=none, reverse=false):
 """ l.sort(key=none, reverse=false) -> none -- stable sort *in place* """
 pass

可以看出:sort()函数没有返回值,有两个参数。key表示的是排依据的函数;reverse是指需不需要反转列表,默认为false表示的是升序,如果设为true表示的是降序。

sort()用法如下

l = [1, 2, 7, 4, 3]
l.sort()
print(l)
#[1, 2, 3, 4, 7]

我们加入一些参数来看看,如降序排序:

l = [1, 2, 7, 4, 3]
l.sort(reverse=true)
print(l)
#[7, 4, 3, 2, 1]

再比如,我们想通过key来实现降序排序,这里用到匿名函数lambda:

l = [1, 2, 7, 4, 3]
l.sort(key=lambda x : x*(-1))
print(l)
#[7, 4, 3, 2, 1]

当然,key还有很多用法,我们可以将其设为内置函数或者自定义函数,比如用绝对值函数:

l = [-1, 2, -7, 4, 3]
l.sort(key=abs)
print(l)
#[-1, 2, 3, 4, -7]

python3对key的还有另外一种支持,就是通过functools模块的cmp_to_key函数,将传统的cmp函数转为key。这和lambda函数用法类似,但是对于比较的逻辑比较复杂的情况,这样的方式会更清晰更便于维护。

from functools import cmp_to_key
l = [-1, 2, -7, 4, 3]
def cmp(a,b):
 if abs(a) < abs(b):
 return -1
 else:
 return 1
l.sort(key=cmp_to_key(cmp))
print(l)

二.sorted()

比起sort()函数只能排序列表,sorted函数更通用一些。官方文档对其有详细说明:

sorted(iterable, *, key=none, reverse=false)
return a new sorted list from the items in iterable.

sorted()返回值为list类型。参数列表iterable表示可迭代对象;*表示位置参数就此终结,后面的参数都必须用关键字来指定;key与reverse参数用法与sort()完全一致。

sorted()用法如下:

l = [1, 2, 7, 4, 3]
l1 = sorted(l)
print(l1)
#[1, 2, 3, 4, 7]

对于reverse和key的用法就不单独展示了,可以参考sort()。如果我们想要排序的对象是一个集合,那么用法与列表排序一样:

s = {1, 2, 7, 4, 3}
s1= sorted(s)
print(s1)
#[1, 2, 3, 4, 7]

注意这个排序对象是set对象,而结果是一个列表对象(set对象本就没有顺序一说)。如果我们要实现一个复杂一点的排序呢?比如对字典排序,排序依据是键值,排序结果是由键值对组成的列表:

dict = {'a':2, 'b':3, 'c':7, 'd':4, 'e':1}
l = sorted(dict.items(), key=lambda x : x[1])
print(l)
#[('e', 1), ('a', 2), ('b', 3), ('d', 4), ('c', 7)]

最后给一个leetcode上的题():

给定一个字符串,请将字符串里的字符按照出现的频率降序排列。
输入:
"tree"
输出:
"eert"
解释:
'e'出现两次,'r'和't'都只出现一次。因此'e'必须出现在'r'和't'之前。此外,"eetr"也是一个有效的答案。

通过使用sorted函数代码就很清晰易读了:

def frequencysort(s):
 d = {}
 for i in s:
 d[i] = d.get(i,0) + 1
 l = sorted(d.keys(), key=lambda x: d[x], reverse=true)
 s = ''
 for i in l:
 s += i * d[i]
 return s

注意这里的d.keys()生成的是一个由key组成的可迭代对象,而前面代码的dict.items()生成的是一个由键值对对象(这是一个tuple对象)组成的可迭代对象。

到此这篇关于python排序函数用法的文章就介绍到这了,更多相关python排序函数用法内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!