python使用add进行重载加法

python使用add进行重载加法
python使用add进行重载加法

本文教程操作环境:windows7系统、Python 3.9.1,DELL G3电脑。

1、先定义一个类:

class Point:
    def __init__(self, x, y):
        self.x = x
        self.y = y
 
>>> a = Point(2, 4)
>>> b = Point(3, 5)
>>> a + b
Traceback (most recent call last):
  File "/usr/local/python3/lib/python3.6/site-packages/IPython/core/interactiveshell.py", line 2862, in run_code
    exec(code_obj, self.user_global_ns, self.user_ns)
  File "<ipython-input-7-f96fb8f649b6>", line 1, in <module>
    a + b
TypeError: unsupported operand type(s) for +: 'Point' and 'Point'

很显然 a 和 b 并不能相加,但是我们可以定义一个方法让它们实现相加。

class Point:
    def __init__(self, x, y):
        self.x = x
        self.y = y
    
    # 定义一个 add 方法
    def add(self, other):
        return Point(self.x + other.x, self.y + other.y)
 
>>> a = Point(2, 4)
>>> b = Point(3, 5)
>>> c = a.add(b)
>>> c.x
Out[6]: 5

2、通过一个 add 方法,我们实现了它们的相加功能。但是,我们还是习惯使用加号,事实上,我们只要改下函数名就可以使用 + 进行运算了。

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

很显然 + 就是调用类的 __add__ 方法,因为我们只要加入这个方法就能够实现加法操作。

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注