python中定义一个类时,可以使用其魔术方法来定制化自己的类,简单来说,魔术方法就是在定义一个类时,用双下划线包起来的方法,现在要说的是其中的一部分,就是运算符重载:
首先我们要弄清什么是运算符,运算符就是用在变量之间执行某种操作的符号,例如,+、-、*、/、==、<等等符号都属于运算符,在python中重载它们是非常简单的,例如我们现在定义一个Point类,其代表二维笛卡尔坐标系中的一个点
class Point:
def __init__(self, x, y):
self.x = x
self.y = y
def __add__(self, other):
return Point(self.x + other.x, self.y + other.y)
def __sub__(self, other):
return Point(self.x - other.x, self.y - other.y)
这样一个简单的Point类就建好了,并且实现了矢量加法与减法,我们还可以实现乘法和除法,但是两个矢量直接相乘或相除其实数学意义是不明确的,因此矢量乘法和除法应该是相对于一个常量而言的
def __mul__(self, other):
assert isinstance(other, (int, float))
return Point(self.x * other, self.y * other)
def __truediv__(self, other):
assert isinstance(other, (int, float))
return Point(self.x / other, self.y / other)
mul表示乘法,truediv表示浮点除法,也就是 / ,当然还有整数除法floordiv,也就是 //
到这里一个Point类的几何信息就基本设计完成了,但这还不够,现在希望可以实现比大小,一个点距离原点的长度比另一个更大就认为该Point大于另一个
from math import sqrt
class Point:
def __init__(self, x, y):
self.x = x
self.y = y
self.value = sqrt(x ** 2 + y ** 2) # 距离原点的欧氏距离
def __le__(self, other):
return self.value <= other.value
def __lt__(self, other):
return self.value < other.value
def __ge__(self, other):
return self.value >= other.value
def __gt__(self, other):
return self.value > other.value
def __eq__(self, other):
return self.value == other.value
def __ne__(self, other):
return self.value != other.value
其实,在这里我们可以不写ne(not equal),ne默认为eq取反,到此为止Point类已经设计完成了,以上也只是非常简单的演示了如何重载运算符,需要注意的是,add、sub、mul、div等等涉及到算术运算的魔术方法都有对应的一个r方法,比如radd、rsub,这是将一个实例对象放在该运算符的右边时对应的处理逻辑,比如想实现Point加一个常数
def __add__(self, other):
if isinstance(other, (int, float)):
return Point(self.x + other, self.y + other)
return Point(self.x + other.x, self.y + other.y)
p = Point(1, 1)
print(p + 1) # Point(2, 2)
print(1 + p) # TypeError: unsupported operand type(s) for +: 'int' and 'Point'
如果p在+的右侧,执行的是1(int类型)的__add__方法,很显然,int内置的add方法不会知道怎么与Point相加,因此我们需要定义p的__radd__方法,当int尝试调用add失败后,python会继续调用Point的radd方法,只有都失败的情况下才会出现上面的TypeError
def __radd__(self, other):
if isinstance(other, (int, float)):
return Point(self.x + other, self.y + other)
return Point(self.x + other.x, self.y + other.y)
这次,上面的两个print就不会报错了
本来想到此为止的,但最近在学C++,突然想到一个很有意思的东西,就是在python中使用cout,单独开一篇探讨一下