脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python sort 自定义函数排序

Python sort 自定义函数排序问题

 更新时间：2023年09月12日 09:17:39   作者：世界尽头与你  
这篇文章主要介绍了Python sort 自定义函数排序问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

Python sort 自定义函数排序

sort() 函数用于对原列表进行排序，如果指定参数，则使用比较函数指定的比较函数。

sort() 方法语法：

list.sort( key=None, reverse=False)

参数

  • key – 主要是用来进行比较的元素，只有一个参数，具体的函数的参数就是取自于可迭代对象中，指定可迭代对象中的一个元素来进行排序。
  • reverse – 排序规则， reverse = True 降序， reverse = False 升序（默认）。

以下实例演示了通过指定列表中的元素排序来输出列表：指定第二个元进行素排序

def take_second(elem):
    return elem[1]
random = [(2, 2), (3, 4), (5, 6), (1, 5), (6, 3)]
random.sort(key=take_second)
print(random)  # [(2, 2), (6, 3), (3, 4), (1, 5), (5, 6)]

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

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

例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]

例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)

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

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

最新评论