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

全栈程序员站长

发布于 2022-11-15 17:51:17

6690

发布于 2022-11-15 17:51:17

当你想按自己的方式对数组元素进行排序时，我们需要自定义比较函数实现我们想实现的排序方式。例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.cn/226750.html原文链接：https://javaforall.cn

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2022年10月30日，如有侵权请联系 cloudcommunity@tencent.com 删除

https