python3 类排序 类比较

自定义类比较

让python3 的类支持比较操作可以利用total_ordering修饰器重载 ==, <=, > 运算符

官方文档是这样说的

来自此文档

示例代码:

这份代码由于是重载了比较符号, 因此可以进行比较操作, 而执行sorted函数时也可以根据自定义的比较操作排序了.

from functools import total_ordering

@total_ordering
class Movies:
    '''
    电影信息, 包含标题和评分数据, 期望自定义根据评分排序
    '''
    def __init__(self, title='default', score=-0.0): # 初始化操作, 设置默认值
        self.title = title
        self.score = score

    def __str__(self): # 自定义打印类的内容
        return "{} : {}".fORMat(self.title, self.score)

    # 使用total_ordering修饰器重载 ==, <=, > 运算符
    def __eq__(self, other):
        return self.score == other.score

    def __le__(self, other):
        return self.score < other.score

    def __gt__(self, other):
        return self.score > other.score

movie = [0]*3

movie[0] = Movies("电影1", 8.1)

# movie_2 = Movies()
movie[1] = Movies('电影2', 9.2)

movie[2] = Movies("电影3", 3.4)

for each in movie:
    print(each)
print("-----\n")

sorted_movie = sorted(movie)

for each in sorted_movie:
    print(each)

print(movie[1] > movie[2])

输出结果

自定义类排序

这个思路是改变sorted函数的key, 从而实现对类的自定义排序

使用functools库或者自己直接构造lambda自定义排序的元素.

感觉自定义lambda函数要方便很多, 代码少, 要切换升序降序给元素添个负号就好了.

import functools

class Movies:
    '''
    电影信息, 包含标题和评分数据, 期望自定义根据评分排序
    '''
    def __init__(self, title='default', score=-0.0): # 初始化操作, 设置默认值
        self.title = title
        self.score = score

    def __str__(self): # 自定义打印类的内容
        return "{} : {}".format(self.title, self.score)

def cmp(self, other): # 自定义比较函数
    if self.score < other.score:
        return 1
    elif self.score == other.score:
        return 0
    else:
        return -1

movie = [0]*3

movie[0] = Movies("电影1", 8.1)

# movie_2 = Movies()
movie[1] = Movies('电影2', 9.2)

movie[2] = Movies("电影3", 3.4)

# sorted_movie = sorted(movie, key=functools.cmp_to_key(cmp))
sorted_movie = sorted(movie, key=lambda mov: -mov.score)

for each in sorted_movie:
    print(each)

输出结果