Python运算符重载，代码秒变高大上！

最新推荐文章于 2024-07-31 18:10:45 发布

图灵学者

最新推荐文章于 2024-07-31 18:10:45 发布

阅读量1.4k

点赞数 21

分类专栏： python精华文章标签： python 开发语言

本文链接：https://blog.csdn.net/xyh2004/article/details/139830303

版权

4.1 __eq__, __lt__等方法

1、运算符重载基础介绍 🧮

1.1 什么是运算符重载

运算符重载是面向对象编程中的一项重要特性，它允许程序员为自定义的数据类型（如类）重新定义标准运算符的行为。这意味着，对于加号+、减号-等运算符，我们可以在不同类型的对象间赋予其特定的操作意义，而不局限于基本数据类型的操作。例如，在自定义的复数类中，我们可以重载+运算符，使其能够实现复数的加法操作。

1.2 为何使用运算符重载

• 增强代码可读性：运算符重载使得代码更接近自然语言，减少冗长的函数调用，使得代码更加简洁易懂。
• 提升代码灵活性：通过为运算符赋予新的含义，可以轻松地扩展已有类的功能，适应更多的应用场景。
• 促进代码复用：标准运算符的普遍认知减少了新用户的学习成本，同时也便于在不同场景下复用已有的运算逻辑。

1.3 Python中的特殊方法魔法

Python中，运算符重载主要通过定义特殊的魔法方法（也称为dunder方法，即双下划线包围的方法）来实现。这些方法通常以__开始并以__结束，比如__add__用于重载加法运算符+。当对两个对象使用相应运算符时，Python会自动调用这些特殊方法来完成操作。以下是一些基本的特殊方法及其用途：

• __add__(self, other): 定义加法行为，如a + b。
• __sub__(self, other): 定义减法行为，如a - b。
• __mul__(self, other): 定义乘法行为，如a * b。
• __truediv__(self, other): 定义真除法行为，如a / b。
• __str__(self): 定义对象的字符串表示形式，用于打印或转换为字符串。

示例：重载加法运算符

假设我们要创建一个简单的向量类，重载加法运算符以便于向量之间的相加。

class Vector:
    def __init__(self, x=0, y=0):
        self.x = x
        self.y = y
    
    def __add__(self, other):
        """重载加法运算符，实现向量加法"""
        return Vector(self.x + other.x, self.y + other.y)
    
    def __str__(self):
        return f"Vector({self.x}, {self.y})"

# 使用示例
v1 = Vector(1, 2)
v2 = Vector(3, 4)
result = v1 + v2
print(result)  # 输出: Vector(4, 6)

在这个例子中，我们定义了一个简单的Vector类，重载了__add__方法来实现向量的加法运算。当使用+运算符时，Python内部调用了__add__方法，实现了向量对象的相加，并返回了一个新的向量对象作为结果。

通过这种方式，运算符重载不仅提高了代码的可读性和表达力，还为自定义类型提供了与内置类型相似的操作体验。

2、实战：重载加法运算符 + 🧩

2.1 自定义类与`add()`

在Python中，自定义类可以通过定义特殊方法__add__来重载加法运算符+。这个方法接收另一个对象作为参数，并返回运算后的结果。让我们通过一个简单例子来理解这一过程：

class Point:
    def __init__(self, x=0, y=0):
        self.x = x
        self.y = y
    
    def __add__(self, other):
        if isinstance(other, Point):
            return Point(self.x + other.x, self.y + other.y)
        else:
            raise TypeError("Operand must be an instance of Point")

# 使用示例
p1 = Point(3, 4)
p2 = Point(1, 2)
sum_point = p1 + p2
print(sum_point.x, sum_point.y)  # 输出: 4 6

这里，Point类通过实现__add__方法，使得两个点对象可以通过加法运算符相加，得到一个新的点，其坐标为两原点坐标的和。

2.2 应用案例：复数加法

Python的内置complex类型已经重载了加法运算符，但为了演示，我们可以创建一个简化版的复数类来展示这一机制：

class MyComplex:
    def __init__(self, real=0, imag=0):
        self.real = real
        self.imag = imag
    
    def __add__(self, other):
        if isinstance(other, MyComplex):
            return MyComplex(self.real + other.real, self.imag + other.imag)
        else:
            raise TypeError("Operand must be an instance of MyComplex")

# 使用示例
c1 = MyComplex(3, 4)
c2 = MyComplex(1, -2)
sum_complex = c1 + c2
print(sum_complex.real, sum_complex.imag)  # 输出: 4 2

通过定义MyComplex类的__add__方法，我们可以像操作内置复数那样，使用加法运算符直接相加两个自定义复数对象。

2.3 深入理解`add`方法

__add__方法是Python中用于定义加法行为的核心特殊方法。当对两个对象使用加号+时，Python会检查右侧对象是否有__radd__方法，如果没有，则调用左侧对象的__add__方法。这保证了即使在操作符两边的对象类型不同时，也能尝试找到合适的方法来完成操作。值得注意的是，正确实现isinstance检查对于确保运算符的兼容性和避免类型错误至关重要。此外，实现__add__时，还应考虑运算的逆操作，即可能需要定义__radd__方法来处理非对称操作情况。