让类支持比较操作

原创于 2020-10-27 18:34:53 发布 · 102 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

本文介绍如何在Python中通过定义Shape基类并使用functools.total_ordering装饰器来实现不同形状类间的比较运算符重载。具体示例包括矩形Rectangle和圆形Circle的面积比较。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

方法1：比较运算符重载，需要实现以下方法：小于，小于等于，大于，大于等于，等于，不等于。

from functools import total_ordering
from abc import ABCMeta,abstractmethod

@total_ordering  #装饰类
class Shape(object):   #父类
    @abstractmethod
    def area(self):
        pass
    # 小于
    def __lt__(self, other):
        print('in_lt_')
        if not isinstance(other, Shape):
            raise TypeError('other is not Shape')
        return self.area() < other.area()

    def __eq__(self, other):
        print('in_eq_')
        if not isinstance(other, Shape):
            raise TypeError('other is not Shape')
        return self.area() == other.area()

class Rectangle(Shape):           #继承
    def __init__(self, w, h):
        self.w = w
        self.h = h
    def area(self):
        return self.w * self.h

class Circle(Shape):
    def __init__(self, r):
        self.r = r
    def area(self):
        return self.r ** 2 * 3.14
r1 = Rectangle(5, 3)
r2 = Rectangle(4, 4)
c1 = Circle(3)

print(c1 <= r1)
print(r1 > c1)