(27)Python操作符重载

以双下划线__开头的类函数在Python中称为特殊函数。 这是因为，它们是有特殊含义。 上面定义的__init__()函数是其中之一。 每次创建该类的新对象时都会调用它。 Python中有很多特殊功能。

使用特殊功能，可以使类与内置函数兼容。

>>> p1 = Point(2,3)
>>> print(p1)
<__main__.Point object at 0x00000000031F8CC0>
Python

但是如果打印不好或不够美观。可以在类中定义__str__()方法，可以控制它如何打印。 所以，把它添加到类中，如下代码 –

class Point:
    def __init__(self, x = 0, y = 0):
        self.x = x
        self.y = y

    def __str__(self):
        return "({0},{1})".format(self.x,self.y)
Python

现在再试一次调用print()函数。

>>> p1 = Point(2,3)
>>> print(p1)
(2,3)
Shell

当使用内置函数str()或format()时，调用同样的方法。

>>> str(p1)
'(2,3)'

>>> format(p1)
'(2,3)'
Python

所以，当执行str(p1)或format(p1)，Python在内部执行p1.__str__()。

要重载+号，需要在类中实现__add__()函数。可以在这个函数里面做任何喜欢的事情。 但是返回Point对象的坐标之和是最合理的。

class Point:
    def __init__(self, x = 0, y = 0):
        self.x = x
        self.y = y

    def __str__(self):
        return "({0},{1})".format(self.x,self.y)

    def __add__(self,other):
        x = self.x + other.x
        y = self.y + other.y
        return Point(x,y)
Python

现在让我们再试一次运行上面的代码 –

>>> p1 = Point(2,3)
>>> p2 = Point(-1,2)
>>> print(p1 + p2)
(1,5)
Shell

实际发生的是，当执行p1 + p2语句时，Python将调用p1.__add__(p2)，之后是Point.__add__(p1，p2)。 同样，也可以重载其他运算符。需要实现的特殊功能列在下面。

Python中的运算符重载特殊函数 –

Python不会限制操作符重载算术运算符。也可以重载比较运算符。

假设想在Point类中实现小于符号<比较运算。

比较这些来自原点的数值，并为此返回结果。 可以实现如下。

class Point:
    def __init__(self, x = 0, y = 0):
        self.x = x
        self.y = y

    def __str__(self):
        return "({0},{1})".format(self.x,self.y)

    def __lt__(self,other):
        self_mag = (self.x ** 2) + (self.y ** 2)
        other_mag = (other.x ** 2) + (other.y ** 2)
        return self_mag < other_mag
Pythton

在Python shell中尝试这些示例运行。

>>> Point(1,1) < Point(-2,-3)
True

>>> Point(1,1) < Point(0.5,-0.2)
False

>>> Point(1,1) < Point(1,1)
False
Pythton

类似地，可以实现的特殊功能，以重载其他比较运算符，如下表所示。