当有一组比较复杂的对象需要进行排序时,我们的第一想法就是尽量利用已有的函数快速完成自己的排序需求。接触过c++的码友应该知道,在c++里若想利用已有的sort函数来完成复杂的排序,对运算符进行重载即可(好久没接触过c++里的概念,应该是这个叫法)。
笔者刚刚需要在python里面进行相关的操作,也试了用这样的方法,但是会提示报错
def comp(x, y):
pass
X = sorted(X, cmp=comp)
笔者的写法类似于以上写法,就是定义了一个comp函数,如果
x
<
y
x<y
x<y,则返回负数-1;如果
x
>
y
x>y
x>y,则返回正数1,否则返回0.然后在sorted函数中直接将comp函数传给cmp参数。但是会报错,查阅博客找到原因是因为python2中仍然有cmp的参数,按上述方法进行操作没有问题;但是python3中去掉了cmp参数,因此不能按以上方法进行操作。
但是稍微再利用一下functools包里面的cmp_to_key()方法就行了,从刚刚那篇博客里贴一个简单的例子,供笔者自己以后查阅
from functools import cmp_to_key
nums = [4, 3, 2, 1]
sorted(nums,key=cmp_to_key(lambda a, b: a - b))