python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python sort 自定义函数排序

Python sort 自定义函数排序问题

作者:世界尽头与你

这篇文章主要介绍了Python sort 自定义函数排序问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

Python sort 自定义函数排序

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

sort() 方法语法:

list.sort( key=None, 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)

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

总结

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

您可能感兴趣的文章:
阅读全文