本文将揭开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库,新手也能动手搭建推荐系统》
微信搜索关注《Python学研大本营》,加入读者群,分享更多精彩
访问【IT今日热榜】,发现每日技术热点