文章目录
- 自定义类比较
- 自定义类排序
自定义类比较
让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)
输出结果
