运算符的正确重载

阿拉辉

于 2020-07-31 15:53:39 发布

阅读量207

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/weixin_38492159/article/details/107671748

版权

python 专栏收录该内容

68 篇文章 1 订阅

订阅专栏

正确重载运算符

运算符重载基础
重载加法运算符+

:
这次主要讨论一下几个问题
• Python如何处理中缀运算符中不同类型的操作数
• 使用鸭子类型或显式类型检查处理不同类型的操作数
• 中缀运算符如何表明自己无法处理操作数
• 众多比较运算符（如==、>、<=，等等）的特殊行为
• 增量赋值运算符（如+=）的默认处理方式和重载方式

运算符重载基础

Python施加了一些限制，做好了灵活性、可用性和安全性方面的平衡：
• 不能重载内置类型的运算符
• 不能新建运算符，只能重载现有的
• 某些运算符不能重载——is、and、or和not（不过位运算符&、|和~可以） # 一元运算符

-（neg）
一元取负算术运算符。如果x是-2，那么-x == 2。

+（pos）
一元取正算术运算符。通常，x == +x，但也有一些例外。如果好奇，请阅读“x和+x何时不相等”附注栏。

~（invert）
对整数按位取反，定义为~x == -(x+1)。如果x是2，那么~x == -3。

还把内置的abs(…)函数列为一元运算符。它对应的特殊方法是__abs__

支持一元运算符很简单，只需实现相应的特殊方法。这些特殊方法只有一个参数，self。
遵守运算符的一个基本规则：始终返回一个新对象。也就是说，不能修改self，要创建并返回合适类型的新实例。

对-和+来说，结果可能是与self同属一类的实例。多数时候，+最好返回self的副本。abs(…)的结果应该是一个标量。

#示例1：把一元运算符-和+添加到之前的示例中
def __abs__(self): 
    return math.sqrt(sum(x * x for x in self)) 

def __neg__(self): 
    return Vector(-x for x in self)  #➊ 

def __pos__(self): 
    return Vector(self)  #➋

➊ 为了计算-v，构建一个新Vector实例，把self的每个分量都取反。
➋ 为了计算+v，构建一个新Vector实例，传入self的各个分量。

重载加法运算符+

序列应该支持+运算符（用于拼接），以及*运算符（用于重复复制）。然而，我们将使用向量数学运算实现+和*运算符。
两个欧几里得向量加在一起得到的是一个新向量，它的各个分量是两个向量中相应的分量之和。

#示例1
>>> v1 = Vector([3, 4, 5]) 
    >>> v2 = Vector([6, 7, 8]) 
    >>> v1 + v2 
    Vector([9.0, 11.0, 13.0]) 
    >>> v1 + v2 == Vector([3+6, 4+7, 5+8]) 
    True

#示例2：Vector.__add__方法，第1版
    # 在Vector类中定义 
 
    def __add__(self, other): 
        pairs = itertools.zip_longest(self, other, fillvalue=0.0)  # ➊ 
        return Vector(a + b for a, b in pairs)  # ➋

➊ pairs是个生成器，它会生成(a, b)形式的元组，其中a来自self，b来自other。如果self和other的长度不同，使用fillvalue填充较短的那个可迭代对象。
➋ 构建一个新Vector实例，使用生成器表达式计算pairs中各个元素的和。

注意__add__返回一个新Vector实例，而没有影响self或other。

实现一元运算符和中缀运算符的特殊方法一定不能修改操作数。使用这些运算符的表达式期待结果是新对象。只有增量赋值表达式可能会修改第一个操作数（self）

还可以把Vector加到元组或任何生成数字的可迭代对象上

#示例3
>>> 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])

示例3中的两个加法都能如我们所期待的那样计算，这是因为__add__使用了zip_longest(…)，它能处理任何可迭代对象，而且构建新Vector实例的生成器表达式仅仅是把zip_longest(…)生成的值对相加（a + b），因此可以使用任何生成数字元素的可迭代对象。
然而，如果对调操作数，混合类型的加法就会失败。

#示例4：果左操作数是Vector之外的对象，第一版Vector.__add__方法无法处理
>>> v1 = Vector([3, 4, 5]) 
>>> (10, 20, 30) + v1 
Traceback (most recent call last): 
  File "<stdin>", line 1, in <module> 
TypeError: can only concatenate tuple (not "Vector") to tuple 
>>> from vector2d_v3 import Vector2d 
>>> v2d = Vector2d(1, 2) 
>>> v2d + v1 
Traceback (most recent call last): 
  File "<stdin>", line 1, in <module> 
TypeError: unsupported operand type(s) for +: 'Vector2d' and 'Vector'