重载加法运算符
对于一个向量,我们期望它支持加法运算:
>>> v1 = Vector([3, 4, 5])
>>> v2 = Vector([6, 7, 8])
>>> v1 + v2
Vector([9.0, 11.0, 13.0])
为了达到上面加法的效果,可以在Vector
类中添加__add__
方法:
# 在Vector类中定义
def __add__(self, other):
pairs = itertools.zip_longest(self, other, fillvalue=0.0)
return Vector(a + b for a, b in pairs)
需要注意的是,这里的加法操作没有对self
或者other
重新赋值,而是返回了一个新的Vector
对象。这也符合加法操作的期望。
这里的__add__
方法使用了zip_longest
方法,不仅可以支持Vector类之间的相加操作,还可以支持Vector类和其他可迭代对象之间的相加。但是在处理可迭代对象和Vector类之间的相加时,可能会出现类型错误的现象(注意加数的先后关系):
>>> v1 = Vector([3, 4, 5])
>>> v1 + (10, 20, 30)
Vector([13.0, 24.0, 35.0])
>>> from vector2d_v3 import Vector2d
>>> v2d = Vector2d(1, 2)
>>> v1 + v2d
Vector([4.0, 6.0, 5.0])
>>> v1 = Vector([3, 4, 5])
>>> (10, 20, 30) + v1
Traceback (most recent call last):
File "", line 1, in
TypeError: can only concatenate tuple (not "Vector") to tuple
这里需要简单说一下,在执行a+b
这个操作的时候,方法查找的顺序:
- 如果a实现了
__add__
方法,则调用a.__add__(b)
,并返回结果; - 如果a没有实现
__add__
方法,则检查b是否实现了__radd__
方法,若实现了,则调用b.__radd__(a)
,并返回结果; - 若b中没有实现
__radd__
方法,则跑出TypeError
,说明操作类型不支持;
下面为Vector
类添加__radd__
方法:
# 在Vector类中定义
def __add__(self, other):
try:
pairs = itertools.zip_longest(self, other, fillvalue=0.0)
return Vector(a + b for a, b in pairs)
except TypeError:
return NotImplemented
def __radd__(self, other):
return self + other
重载乘法运算符
类似重载加法运算符,重载乘法运算符需要重写__mul__
和__rmul__
:
# 在Vector类中定义
def __mul__(self, scalar):
return Vector(n * scalar for n in self)
def __rmul__(self, scalar):
return self * scalar
同样的,为了提供明确可读的报错信息:
def __mul__(self, scalar):
if isinstance(scalar, numbers.Real):
return Vector(n * scalar for n in self)
else:
return NotImplemented
def __rmul__(self, scalar):
return self * scalar
重载等号运算符
等号操作需要进行两步判断:
- 是否属于同一个类的实例
- 实例属性是否完全相等
def __eq__(self, other):
if isinstance(other, Vector):
return (len(self) == len(other) and all(a == b for a, b in zip(self, other)))
else:
return NotImplemented