发布时间:2019-08-05 15:33:48编辑:auto阅读(1902)
让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)
输出结果
上一篇: Python 繁体中文与简体中文相互转换
下一篇: 【Python3】f-string——s
47458
45752
36758
34275
28931
25562
24411
19582
19066
17605
5435°
6017°
5531°
5609°
6533°
5345°
5345°
5853°
5825°
7138°