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

本文将揭开Python中魔术背后的方法。

微信搜索关注《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从入门到精通（第3版）》

《Python从入门到精通（第3版）》从初学者角度出发，通过通俗易懂的语言、丰富多彩的实例，详细介绍了使用Python进行程序开发应该掌握的各方面技术。全书共分27章，包括初识Python、Python语言基础、运算符与表达式、流程控制语句、列表和元组、字典和集合、字符串、Python中使用正则表达式、函数、面向对象程序设计、模块、文件及目录操作、操作数据库、使用进程和线程、网络编程、异常处理及程序调试、Pygame游戏编程、推箱子游戏、网络爬虫开发、火车票分析助手、数据可视化、京东电商销售数据分析与预测、Web编程、Flask框架、e起去旅行网站、Python自动化办公、AI图像识别工具等内容。书中所有知识都结合具体实例进行介绍，涉及的程序代码都给出了详细的注释，读者可轻松领会Python程序开发的精髓，快速提升开发技能。

《Python从入门到精通（第3版）》https://item.jd.com/14055900.html

精彩回顾

《文本抓取利器，Python和Beautiful Soup爬虫助你事半功倍》

《基于Python和Surprise库，新手也能动手搭建推荐系统》

《仅需10行代码，用Python创建语音模拟器》

《PyCharm编辑器结合Black插件，轻松实现Python代码格式化》

《金融量化交易初探，使用Python实现遗传算法》

《Python GUI初学者教程，轻松构建用户界面》

微信搜索关注《Python学研大本营》，加入读者群，分享更多精彩

访问【IT今日热榜】，发现每日技术热点