导语
小编一直都觉自己公众号的排版很鸡肋,从这篇文章开始将使用了新的排版风格,还特意地做了一个卡通二维码(见文末),希望大家会喜欢(不要脸地假装有很多粉丝)。其实关于排版,小编要真心感谢一下景禹大佬的指导。好了,今天想跟大家谈谈如何重载运算符+,认真看完这篇文章,你将收获:
-
了解中缀运算符特殊方法的分派机制
-
了解向量类如何实现 __ add __ 方法
-
了解向量类如何实现 __ radd __ 方法
a+b背后如何调用特殊方法
大家都知道若a和b都是同类型序列,a+b可以实现序列接拼,若a和b都是int或者float等数值类型,a+b会实现数学上的加法,见示例1。
#示例1
a = (1,2)
b = (4,5)
print(a+b) #(1,2,4,5)
a = [1,2]
b = [4,5]
print(a+b) #[1,2,4,5]
a = b = 1
print(a+b) #2
a = b = 1.0
print(a+b) #2.0
但是,如果a和b是不同类型的序列,他们能否接拼成功呢?
要回答这个问题,我们先了解一下Python为中缀运算符特殊方法提供的特殊分派机制,其流程见下图。
对于表达式a+b,为了支持涉及不同类型的运算,Python解释器会执行以下几步操作。
- 如果a有__ add __ 方法,返回值不是NotImplemented,调用a.__ add __(b),然后返回结果
- 如果a没有__ add __ 方法,或者调用__ add __ 方法返回NotImplemented,检查b有没有__ radd __ 方法,如果有,调用b.__ radd __(a)方法后没有返回NotImplemented,返回结果。
- 如果b没有__ radd __ 方法,或者调用__ radd __方法返回NotImplemented,抛出TypeError,并在错误消息中指明操作类型不支持。
__ radd __ 是 __ add __的“反向”版本,Alex、Anna和Leo几位技术大佬喜欢称之为“右向”(right)特殊方法,因为他们都在右操作数上调用。
如果a和b是不同类型的序列,执行示例2的程序,会发生什么呢?
#示例2
a = (1,2)
b = [3,4]
print(a+b)
首先解释器会调用a.__ add __ (b)方法,因为a和b是不同类型的序列,所以返回NotImplemented。然后解释器检查b是否有b.__ radd __ (a) 方法,然后调用该方法,但是还是返回NotImplemented,最终抛出TypeError结果。
TypeError: can only concatenate tuple (not "list") to tuple
重载运算符+
关于重载运算符,有几点需要注意一下的:
- 不能重载内置类型的运算符,即list等这些内置类型的运算符不能重载。
- 不能新建运算符,只能重载现有的。
- 有些运算符不能重载,is、and、or和not(位运算&、|和~可以)。
现在我们尝试定义一个Vector类,见示例3。
#示例3
class Vector(object):
def __init__(self,components):
self.components = list(components)
如果我们不对Vector重载运算符+,两个Vector类相加抛出错误,见示例4
#示例4
v1 = Vector([1,2,3])
v2 = Vector([1,2,3])
print(v1+v2)
#TypeError: unsupported operand type(s) for +: 'Vector' and 'Vector'
好了,我们现在马上对Vector重载运算符+吧,见示例5,但是我们不是实现接拼,而是实现数学上的加法,因为对于一个向量,接拼功能意义不大。
#示例5
class Vector(object):
def __init__(self,components):
self.components = list(components)
def __iter__(self):
return iter(self.components)
def __str__(self):
return str(self.components)
def __add__(self,others):
print("__add__")
try:
pairs = itertools.zip_longest(self,others,fillvalue=0.0)
return Vector(a+b for a,b in pairs)
except TypeError:
return NotImplemented
v1 = Vector([1,2,3])
v2 = Vector([1,2,3])
v3 = [1,2,3]
print(v1+v2)
print(v1+v3)
输出结果:
__add__
[2,4,6]
__add__
[2,4,6]
从示例5中可以看到,解析器调用了__ add __方法实现了两个Vector或者Vector和具有数值元素的可迭代类型的相加。示例5能够实现V1+V3,那是否能实现V3+v1呢?见示例6。
#示例6
print(V3+V1)
#TypeError: can only concatenate list (not "Vector") to list
示例6抛出错误了,当解释器执行v3.__ add __ (V1)时候,因为调用的是列表的__ add __ 方法,不能与自定义的类型相加,所以返回结果NotImplemented,并尝试执行V1.__ radd __ (V3)方法,但是在自定义的Vector类并没有实现该方法,所以最后还是抛出了TypeError。好了,现在尝试实现Vector类的__ radd __ 方法。在示例5的基础上添加示例7的代码。
#示例7
def __radd__(self,other):
print("__radd__")
return self+other
再次运行示例6,得到如下输出结果。
__radd__
__add__
[2, 4, 6]
当解释器执行V3.__ add __ (V1)返回NotImplemented之后,就会调用自定义类中的__ radd __ 方法, __ radd __ 方法把计算结果委托给 __ add __ ,事实上,任何可交换的运算符都可以这么做。
好啦,以上就是就是小编今天分享的内容,希望对你有用。
公众号:CVpython
专注于分享Python和计算机视觉,快点扫码关注我吧!