python魔方方法__add__、__mul__、__sub__、__truediv__

python魔方方法__add__、__mul__、__sub__、__truediv__

一、简介

在自定义类的时候，我们可以重写继承来自object类的一些魔方方法，来达到自定义的目的。
本文主要介绍自定义数学运算的一些魔方方法。

二、详解

语法：object.__add__(self, other)
参数：self指代被操作的本类的实例化对象，other指代被操作的本类的另一个实例化对象

三、代码

# Vector类
import math
class Vector(object):

    def __init__(self, x=0, y=0):
        self.x = x
        self.y = y
    
    def __repr__(self):
        return 'Vector ({} {})'.format(self.x, self.y)
    
    def __abs__(self):
        return math.sqrt(self.x**2 + self.y**2)
    
    def __add__(self, other):  # 自定义+
        return (self.x + other.x, self.y + other.y)
    
    def __mul__(self, other):  # 自定义*
        return self.x * other.x + self.y * other.y
    
    def __sub__(self, other):  # 自定义-
        return (self.x - other.x, self.y - other.y)
    
    def __truediv__(self, other):  # 自定义 /
        return (self.x / other.x, self.y / other.y)
    
    def __bool__(self):
        return bool(abs(self))
    
A = Vector(1, 2)
B = Vector(2, 1)
print(A + B, A * B, A - B, A / B, abs(A))

(3, 3) 4 (-1, 1) (0.5, 2.0) 2.23606797749979

在这个例子中我们自定义了一个Vector类，并且改写了向量类的加减乘除的方法。

四、Reference

https://docs.python.org/zh-cn/3.9/reference/datamodel.html?highlight=add#object.add