摘要:在 Python 中,魔法方法帮助你模拟 Python 类中内置函数的行为。这些方法有前后双下划线(__),因此也被称为魔法方法。这些魔法方法也帮助你实现 Python 中的运算符重载。你很可能见过这样的例子。比如使用乘法运算符 * 作用于两个整数会得到它们的
在 Python 中,魔法方法帮助你模拟 Python 类中内置函数的行为。这些方法有前后双下划线(__),因此也被称为魔法方法 。这些魔法方法也帮助你实现 Python 中的运算符重载。你很可能见过这样的例子。比如使用乘法运算符 * 作用于两个整数会得到它们的积。而作用于一个字符串和一个整数 k 时,会得到该字符串重复 k 次:
>>> 3 * 412>>> 'code' * 3'codecodecode'
在本文中,将通过创建一个简单的二维向量 Vector2D 类来探索 Python 中的魔法方法。
我们将从熟悉的方法开始,逐步介绍更多有用的魔法方法。!
class Vector2D: pass一旦你创建了一个类并实例化一个对象,你可以像这样添加属性: obj_name.attribute_name = value 。但是,你不需要手动为每个你创建的实例添加属性,你需要一种方法在实例化对象时初始化这些属性。
要做到这一点,可以定义 __init__ 方法。让我们为我们的 Vector2D 类定义 __init__ 方法:
class Vector2D: def __init__(self, x, y): self.x = x self.y = yv = Vector2D(3, 5)当尝试检查或打印出你实例化的对象时,你会发现你得不到任何有用的信息。
v = Vector2D(3, 5)print(v)Output >>>这就是为什么你应该添加一个表示字符串,即对象的字符串表示。为此,添加一个 __repr__ 方法,如下所示:
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})"v = Vector2D(3, 5)print(v)Output >>> Vector2D(x=3, y=5)__repr__ 应包含创建类实例所需的所有属性和信息。通常使用 __repr__ 方法用于调试目的。
__str__ 也可以用来添加对象的字符串表示。通常,__str__ 方法用于向类的最终用户提供信息。
在类中添加一个 __str__ 方法:
class Vector2D: def __init__(self, x, y): self.x = x self.y = y def __str__(self): return f"Vector2D(x={self.x}, y={self.y})"v = Vector2D(3, 5)print(v)Output >>> Vector2D(x=3, y=5)如果没有 __str__ 的实现,它会回退到 __repr__。所以对于你创建的每一个类,你至少应该添加一个 __repr__ 方法。
接下来,让我们为 Vector2D 类添加一个方法来检查任意两个对象的相等性。如果两个向量对象的 x 和 y 坐标相同,则它们相等。
现在创建两个 Vector2D 对象,它们的 x 和 y 值都相等,并比较它们的相等性:
v1 = Vector2D(3, 5)v2 = Vector2D(3, 5)print(v1 == v2)结果是 False。因为默认的比较方式是检查对象在内存中的 ID 是否相等。
Output >>> False让我们添加 __eq__ 方法来检查是否相等:
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 __eq__(self, other): return self.x == other.x and self.y == other.y现在的相等检查应该可以按预期工作了:
v1 = Vector2D(3, 5)v2 = Vector2D(3, 5)print(v1 == v2)Output >>> TruePython 的内置 len 函数可以帮助你计算内置可迭代对象的长度。假设对于一个向量,长度应该返回该向量包含的元素数量。
那么,让我们为 Vector2D 类添加一个 __len__ 方法:
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 __len__(self): return 2v = Vector2D(3, 5)print(len(v))所有 Vector2D 类的对象长度都是 2:
现在想想我们通常对向量执行哪些常见操作。让添加魔法方法来添加和减去任意两个向量。
如果你直接尝试添加两个向量对象,你会遇到错误。所以你应该添加一个 __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 + v2print(result)Output >>> Vector2D(x=4, y=7)接下来,让我们添加一个 __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 - v2print(result)Output >>> Vector2D(x=2, y=3)我们也可以定义一个`__mul__`方法来定义对象之间的乘法。
Let's implement let's handle
标量乘法:向量与标量的乘法,以及内积:两个向量的点积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)# Scalar multiplicationresult1 = v1 * 2print(result1) # Dot productresult2 = v1 * v2print(result2)Output >>>Vector2D(x=6, y=10)13__getitem__ 魔法方法允许您使用熟悉的方括号 语法索引对象并访问属性或属性切片。
对于 v 类的 Vector2D 对象:
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 >>>35通过实现 __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)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来源:自由坦荡的湖泊AI一点号