Python——操作符重载（重要）

目录#

什么是操作符重载
常见操作符重载方法
- 算术操作符
- 比较操作符
- 赋值操作符
- 一元操作符
操作符重载的最佳实践
示例代码
总结
参考资料

什么是操作符重载#

操作符重载是指在自定义类中重新定义操作符（如 +, -, *, / 等）的行为。在Python中，每个操作符都对应一个特殊的方法，当我们在自定义类中实现这些特殊方法时，就可以自定义操作符的行为。例如，当我们使用 + 操作符对两个自定义对象进行相加时，Python会自动调用对象的 __add__ 方法。

常见操作符重载方法#

算术操作符#

操作符	特殊方法	描述
`+`	`__add__(self, other)`	定义对象相加的行为
`-`	`__sub__(self, other)`	定义对象相减的行为
`*`	`__mul__(self, other)`	定义对象相乘的行为
`/`	`__truediv__(self, other)`	定义对象真除法的行为
`//`	`__floordiv__(self, other)`	定义对象整除的行为
`%`	`__mod__(self, other)`	定义对象取模的行为
`**`	`__pow__(self, other)`	定义对象幂运算的行为

比较操作符#

操作符	特殊方法	描述
`==`	`__eq__(self, other)`	定义对象相等的比较行为
`!=`	`__ne__(self, other)`	定义对象不相等的比较行为
`<`	`__lt__(self, other)`	定义对象小于的比较行为
`<=`	`__le__(self, other)`	定义对象小于等于的比较行为
`>`	`__gt__(self, other)`	定义对象大于的比较行为
`>=`	`__ge__(self, other)`	定义对象大于等于的比较行为

赋值操作符#

操作符	特殊方法	描述
`+=`	`__iadd__(self, other)`	定义对象加法赋值的行为
`-=`	`__isub__(self, other)`	定义对象减法赋值的行为
`*=`	`__imul__(self, other)`	定义对象乘法赋值的行为
`/=`	`__itruediv__(self, other)`	定义对象真除法赋值的行为
`//=`	`__ifloordiv__(self, other)`	定义对象整除赋值的行为
`%=`	`__imod__(self, other)`	定义对象取模赋值的行为
`**=`	`__ipow__(self, other)`	定义对象幂运算赋值的行为

一元操作符#

操作符	特殊方法	描述
`+`	`__pos__(self)`	定义对象正号的行为
`-`	`__neg__(self)`	定义对象负号的行为
`~`	`__invert__(self)`	定义对象按位取反的行为

操作符重载的最佳实践#

保持一致性：操作符重载的行为应该与内置类型的行为保持一致，这样可以让代码更加直观和易于理解。例如，如果重载了 + 操作符，那么它应该实现类似于加法的功能。
避免过度重载：不要为了追求功能而过度重载操作符，否则会让代码变得难以理解和维护。只在必要的情况下进行操作符重载。
处理不同类型的操作数：在操作符重载方法中，应该考虑处理不同类型的操作数。例如，当使用 + 操作符时，应该考虑另一个操作数可能是不同类型的情况，并进行相应的处理。

示例代码#

下面是一个简单的自定义类 Vector，用于表示二维向量，并对一些操作符进行了重载：

class Vector:
    def __init__(self, x, y):
        self.x = x
        self.y = y
 
    # 重载加法操作符
    def __add__(self, other):
        if isinstance(other, Vector):
            return Vector(self.x + other.x, self.y + other.y)
        else:
            raise TypeError("Unsupported operand type for +: 'Vector' and '{}'".format(type(other).__name__))
 
    # 重载减法操作符
    def __sub__(self, other):
        if isinstance(other, Vector):
            return Vector(self.x - other.x, self.y - other.y)
        else:
            raise TypeError("Unsupported operand type for -: 'Vector' and '{}'".format(type(other).__name__))
 
    # 重载乘法操作符（向量与标量相乘）
    def __mul__(self, scalar):
        if isinstance(scalar, (int, float)):
            return Vector(self.x * scalar, self.y * scalar)
        else:
            raise TypeError("Unsupported operand type for *: 'Vector' and '{}'".format(type(scalar).__name__))
 
    # 重载相等比较操作符
    def __eq__(self, other):
        if isinstance(other, Vector):
            return self.x == other.x and self.y == other.y
        return False
 
    def __str__(self):
        return "Vector({}, {})".format(self.x, self.y)
 
# 使用示例
v1 = Vector(1, 2)
v2 = Vector(3, 4)
 
# 加法操作
v_sum = v1 + v2
print("v1 + v2 =", v_sum)
 
# 减法操作
v_diff = v1 - v2
print("v1 - v2 =", v_diff)
 
# 乘法操作
v_mul = v1 * 2
print("v1 * 2 =", v_mul)
 
# 相等比较
print("v1 == v2:", v1 == v2)

总结#

操作符重载是Python中一项非常有用的功能，它可以让我们为自定义类的对象重新定义内置操作符的行为，从而提高代码的可读性和可维护性。在使用操作符重载时，需要遵循一些最佳实践，如保持一致性、避免过度重载等。通过合理使用操作符重载，我们可以让自定义对象的使用更加自然和直观。

参考资料#

Python官方文档：https://docs.python.org/3/reference/datamodel.html#special-method-names
《Python核心编程》
《流畅的Python》