python list 自定义排序

您所在的位置：网站首页 › python自定义数组 › python list 自定义排序

python list 自定义排序

#python list 自定义排序| 来源: 网络整理| 查看: 265

我们都知道python中有两种排序的方法，原地排序的x.sort()，和不改变原列表有返回值的sorted(x)自定义排序函数

先不关心其实原地排序还是有返回值的排序，研究其内部的排序原理，其内部是根据什么排序的呢，默认的排序方法是内建的cmp函数对于cmp(x,y):

# 比较器：按目录的深度倒序显示 def tcmp(a,b): if len(a.split('\\')) > len(b.split('\\')) : return -1 elif len(a.split('\\')) < len(b.split('\\')) : return 1 else: return 0

如果想要x排在y前面那么返回一个负数，如果想x排在y后面那么返回一个正数

根据这个原理我们可以自定义一些排序函数compare_personal，并将这个函数传入sort或sorted方法中。在python2中，用关键字cmp直接传入即可，

a = [1,2,4,3,5] def compare_personal: pass a.sort(cmp=compare_personal) sorted(a,cmp=compare_personal)

而在python3中，cmp关键字被移除了，这样的写法就无法运行了，需要借助functools包来完成

a = [1,2,4,3,5] def compare_personal(x,y): pass a.sort(key= functools.cmp_to_key(compare_personal)) c = sorted(b, key=functools.cmp_to_key(compare_personal))

好了，知道如何使用以后就可以构建自定义的排序函数，比如们要对一个数组进行排序，排序的依据是元素mode7后的大小，按照升序排列，我们可以电泳仪排序函数如下

import functools a = [3,8,14,6,7] def compare_personal(x,y): return x%7-y%7 a.sort(key=functools.cmp_to_key(compare_personal)) b = sorted(a,key=functools.cmp_to_key(compare_personal)) print(a) print(b) >>>[14, 7, 8, 3, 6] >>>[14, 7, 8, 3, 6]

按照某一元素进行排序

加入我们有下面的数据，列表中都是元组，我们想根据元组中的某些元素进行排序。元组中的三个元素分别表示姓名，班级和年龄

students = [("zhangsan","A",10),("lisi","C",9),("lisi1","A",9),("lisi2","B",9),("wangwu","B",13)]

我们只需要把要排序的元素通过lamada 表达式返回即可，

students.sort(key=lambda x:(x[2],x[1])) print(students) >>[('lisi1', 'A', 9), ('lisi2', 'B', 9), ('lisi', 'C', 9), ('zhangsan', 'A', 10), ('wangwu', 'B', 13)]

返回一个很好理解，就是根据那一个元素进行排序，上面的代码中返回了两个元素，也就是数据中x[2]相同的数据会根据x[1]再排个序如果是对象的话也是一个道理，lamada返回对象中要排序的属性名称即可

class student: def __init__(self,name,age,weight): self.name = name self.age = age self.weight = weight std1 = student("std1",10,55) std2 = student("std2",8,30) std3 = student("std3",11,65) std4 = student("std4",9,50) students = [std1,std2,std3,std4] students.sort(key= lambda x:x.age) for std in students: print(std.name) >>> std2 std4 std1 std3

利用Operator 模块指定排序元素上面我们说了两种情况，一个是排序元组要通过下标来指定要被排序的元素，第二种是排序对象，要通过指定属性名来指定排序依据的属性。Operator模块的attrgetter和itemgetteritemgetter可用来指定下标attrgetter可用来指定属性名

import operator students = [("zhangsan","A",10),("lisi","C",9),("lisi1","A",9),("lisi2","B",9),("wangwu","B",13)] students.sort(key=operator.itemgetter(2)) print(students) >>> [('zhangsan', 'A', 10), ('lisi1', 'A', 9), ('lisi2', 'B', 9), ('wangwu', 'B', 13), ('lisi', 'C', 9)]

import operator class student: def __init__(self,name,age,weight): self.name = name self.age = age self.weight = weight std1 = student("std1",10,55) std2 = student("std2",8,30) std3 = student("std3",11,65) std4 = student("std4",9,80) students = [std1,std2,std3,std4] students.sort(key= operator.attrgetter("weight")) for std in students: print(std.name) >>> std2 std1 std3 std4

import functools def tcmp(a,b): if a > b : return -1 elif a < b : return 1 else: return 0 nums = [1,2,3,4] sorted_nums = sorted(nums, key = functools.cmp_to_key(tcmp))

Note： functools.cmp_to_key() 将 cmp函数转化为 key。 cmp函数的返回值必须为 [1,-1,0]eg:

import functools def cmpele(a, b): stra = str(a) strb = str(b) stra_len = len(stra) strb_len = len(strb) max_len = max(stra_len, strb_len) i = 0 while i < max_len: if i < stra_len: sa = int(stra[i]) elif i == stra_len: sa = max(int(stra[0]), int(stra[i-1])) if i < strb_len: sb = int(strb[i]) elif i == strb_len: sb = max(int(strb[0]), int(strb[i-1])) if sa > sb: return -1 elif sa < sb: return 1 i += 1 return 0 class Solution: def largestNumber(self, nums): """ :type nums: List[int] :rtype: str """ nums = sorted(nums,key=functools.cmp_to_key(cmpele)) res = "" for ele in nums: res += str(ele) while len(res) > 1 and res[0] == '0': res = res[1:] return res nums = [3,30,34,5,9] print(Solution().largestNumber(nums))

【本文地址】

python list 自定义排序

python list 自定义排序

今日新闻

推荐新闻