Python sort 自定义函数排序问题
目录
Python 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)
最后得到的结果和上面的代码是一样的。
总结
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
您可能感兴趣的文章: