python3 类排序类比较

深藏阁楼爱情的钟 2022-01-10 15:09 316阅读 0赞

### 文章目录 ###

*   *  自定义类比较
     *  自定义类排序

## 自定义类比较 ##

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

官方文档是这样说的  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dqaDI2MjIwNzUxMjc_size_16_color_FFFFFF_t_70]  
[来自此文档][Link 1]

示例代码:

这份代码由于是重载了比较符号, 因此可以进行比较操作, 而执行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])

输出结果  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dqaDI2MjIwNzUxMjc_size_16_color_FFFFFF_t_70 1]

## 自定义类排序 ##

这个思路是改变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)

输出结果  
![在这里插入图片描述][20190620111017795.png]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dqaDI2MjIwNzUxMjc_size_16_color_FFFFFF_t_70]: /images/20220110/a3450f1438eb4f91ae6f2780c58018b1.png
[Link 1]: https://python3-cookbook.readthedocs.io/zh_CN/latest/c08/p24_making_classes_support_comparison_operations.html
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dqaDI2MjIwNzUxMjc_size_16_color_FFFFFF_t_70 1]: /images/20220110/c1d5da04e749451f971f7adf91e10735.png
[20190620111017795.png]: /images/20220110/9efff5c569ec43979624a3ea9f301fb3.png