11个Python开发者必知必会的魔术方法（下）

6.__add__

现在，让我们思考一些在向量上执行的常见操作。添加对任意两个向量进行加减运算的魔术方法。

如果直接尝试将两个向量对象相加，将遇到错误。因此，应该添加一个__add__方法：

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

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

现在可以像这样将任意两个向量相加：

v1 = Vector2D(3, 5)
v2 = Vector2D(1, 2)
result = v1 + v2
print(result)
Output >>> Vector2D(x=4, y=7)

7、__sub__

接下来，添加一个__sub__方法来计算Vector2D类的任意两个对象之间的差值：

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

    def __sub__(self, other):
        return Vector2D(self.x - other.x, self.y - other.y)
v1 = Vector2D(3, 5)
v2 = Vector2D(1, 2)
result = v1 - v2
print(result)
Output >>> Vector2D(x=2, y=3)

8、__mul__

我们还可以定义一个__mul__方法来定义对象之间的乘法运算。

可以实现：

• 标量乘法：向量与标量的乘法。
• 内积：两个向量的点积。

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

    def __mul__(self, other):
        # Scalar multiplication
        if isinstance(other, (int, float)):
            return Vector2D(self.x * other, self.y * other)
        # Dot product
        elif isinstance(other, Vector2D):
            return self.x * other.x + self.y * other.y
        else:
            raise TypeError("Unsupported operand type for *")

现在我们举几个例子，来看看__mul__方法的实际应用。

v1 = Vector2D(3, 5)
v2 = Vector2D(1, 2)

# 标量乘法
result1 = v1 * 2
print(result1)  
# 点积
result2 = v1 * v2
print(result2)
Output >>>

Vector2D(x=6, y=10)
13

9、__getitem__

__getitem__魔术方法允许使用熟悉的方括号[]语法，对对象进行索引并访问属性或属性的片段。

对于Vector2D类的对象v：

• v[0]：x坐标
• v[1]：y坐标

如果尝试通过索引进行访问，将会遇到错误：

v = Vector2D(3, 5)
print(v[0],v[1])
---------------------------------------------------------------------------

TypeError                              Traceback (most recent call last)

 in ()
----> 1 print(v[0],v[1])

TypeError: 'Vector2D' object is not subscriptable

让我们来实现__getitem__方法：

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

    def __getitem__(self, key):
        if key == 0:
            return self.x
        elif key == 1:
            return self.y
        else:
            raise IndexError("Index out of range")

如下所示，现在可以使用索引访问元素了：

v = Vector2D(3, 5)
print(v[0])  
print(v[1])
Output >>>

3
5

10、__call__

通过使用__call__方法，可以像调用函数一样调用对象。

在Vector2D类中，我们可以使用一个__call__方法来按给定系数缩放向量：

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y
   
    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

    def __call__(self, scalar):
        return Vector2D(self.x * scalar, self.y * scalar)

因此，如果现在调用3，就会得到按3倍缩放的向量：

v = Vector2D(3, 5)
result = v(3)
print(result)
Output >>> Vector2D(x=9, y=15)

11、__getattr__

__getattr__方法用于获取对象特定属性的值。

对于这个示例，我们可以添加一个__getattr__魔术方法，当计算向量的幅值（L2-norm）时调用该方法：

class Vector2D:
    def __init__(self, x, y):
        self.x = x
        self.y = y

    def __repr__(self):
        return f"Vector2D(x={self.x}, y={self.y})"

    def __getattr__(self, name):
        if name == "magnitude":
            return (self.x ** 2 + self.y ** 2) ** 0.5
        else:
            raise AttributeError(f"'Vector2D' object has no attribute '{name}'")

现在来验证一下是否能按预期工作：

v = Vector2D(3, 4)
print(v.magnitude)
Output >>> 5.0

结论

这就是本教程的全部内容，介绍了一些最常用的魔术方法。

希望你学会了如何向类中添加魔术方法来模拟内置函数的行为，可以创建一个自己的Python类，并根据所需的功能添加魔术方法。

关于Python学习指南

学好 Python 不论是就业还是做副业赚钱都不错，但要学会 Python 还是要有一个学习规划。最后给大家分享一份全套的 Python 学习资料，给那些想学习 Python 的小伙伴们一点帮助！

包括：Python激活码+安装包、Python web开发，Python爬虫，Python数据分析，人工智能、自动化办公等学习教程。带你从零基础系统性的学好Python！

👉Python所有方向的学习路线👈

Python所有方向路线就是把Python常用的技术点做整理，形成各个领域的知识点汇总，它的用处就在于，你可以按照上面的知识点去找对应的学习资源，保证自己学得较为全面。（全套教程文末领取）

👉Python学习视频600合集👈

观看零基础学习视频，看视频学习是最快捷也是最有效果的方式，跟着视频中老师的思路，从基础到深入，还是很容易入门的。

温馨提示：篇幅有限，已打包文件夹，获取方式在：文末

👉Python70个实战练手案例&源码👈

光学理论是没用的，要学会跟着一起敲，要动手实操，才能将自己的所学运用到实际当中去，这时候可以搞点实战案例来学习。

👉Python大厂面试资料👈

我们学习Python必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有阿里大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

👉Python副业兼职路线&方法👈

学好 Python 不论是就业还是做副业赚钱都不错，但要学会兼职接单还是要有一个学习规划。

👉 这份完整版的Python全套学习资料已经上传，朋友们如果需要可以扫描下方CSDN官方认证二维码或者点击链接免费领取【保证100%免费】