自定义Python排序函数比较方式

全栈程序员-站长 • 2025年6月25日下午5:22 • 未分类 • 阅读 3

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全家桶1年46，售后保障稳定

当你想按自己的方式对数组元素进行排序时，我们需要自定义比较函数实现我们想实现的排序方式。
例1
以降序对数组进行排序

>>> def comp(x,y):
...     return y-x
... 
>>> a = [1,8,4,5,2,7]
>>> a.sort(comp)
>>> a
[8, 7, 5, 4, 2, 1]

Jetbrains全家桶1年46，售后保障稳定

例2

class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e

intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)

def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))

for l in intervals:
	printInterval(l)
# [[1,3],[15,18],[8,10],[2,6]]

def comp(it1, it2):
	return it1.start-it2.start

intervals.sort(comp)

for l in intervals:
	printInterval(l)
# [[1,3],[2,6],[8,10],[15,18]]

以上方法皆是python2的实现。python3下实现有所不同（以上方法python3下跑不通），因为python3 sorted取消了对cmp的支持，即python3没有cmp函数。

例2的python3实现

class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e

intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)

def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))

for l in intervals:
	printInterval(l)
def comp(it1):
	return it1.start

intervals.sort(key=comp)
# sorted(intervals,key=comp)

for l in intervals:
	printInterval(l)

在python3中，待比较元素x通过custom_key函数转化为Python能比较的值custom_key(x)，进而再基于返回值进行排序。

python3 比较第二种方式
如果想和python2中一样定义比较函数，可以通过functools库中的cmp_to_key()函数将比较函数(comparison function)转化为关键字函数（key function)。
functools.cmp_to_key(func)函数将比较函数转化为关键字函数，与接受key function 的工具一同使用（如 sorted(), min(), max())。

from functools import cmp_to_key as ctk
class Interval(object):
	def __init__(self, s=0, e=0):
		self.start = s
		self.end = e

intervals = []
tmp = [[1,3],[15,18],[8,10],[2,6]]
for l in tmp:
	t = Interval(l[0],l[1])
	intervals.append(t)

def printInterval(inter):
	print("[{}, {}]".format(inter.start,inter.end))

for l in intervals:
	printInterval(l)
"""
Please note here.
"""
def comp(it1,it2):
	return it1.start-it2.start

intervals.sort(key=ctk(comp))
# sorted(intervals,key=comp)

for l in intervals:
	printInterval(l)

最后得到的结果和上面的代码是一样的。

发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/226750.html原文链接：https://javaforall.net