大家好,我们接上文,继续讲解Python开发者必会的魔术方法。
__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)
__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)
__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
__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
__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)
__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
综合上下两篇文章的内容,我们介绍了11个最常用的魔术方法。通过本文,可以学习如何向类中添加魔术方法来模拟内置函数的行为,创建一个自己的Python类,并根据所需的功能添加魔术方法。