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

最新推荐文章于 2024-10-03 09:02:12 发布

我爱学AI

最新推荐文章于 2024-10-03 09:02:12 发布

阅读量956

点赞数 22

分类专栏： python脚本文章标签： python 开发语言数据分析

本文链接：https://blog.csdn.net/python1234_/article/details/137645271

版权

python脚本专栏收录该内容

30 篇文章 0 订阅

订阅专栏

简介

在Python中，魔术方法可以帮助模拟内置函数在Python类中的行为。这些方法的名称前后都有双下划线（__），因此也被称为双下划线（dunder）方法。

这些魔术方法还可以帮助在Python中实现运算符重载。简单列举一些常见的例子，比如使用乘法运算符*将两个整数相乘得到乘积。而将它与一个字符串和一个整数k结合使用时，则会将该字符串重复k次：

 >>> 3 * 4
12
>>> 'code' * 3
'codecodecode'

在这里，我们将通过创建一个简单的二维向量Vector2D类来探索Python中的魔术方法。

我们将从你可能熟悉的方法开始，并逐渐过渡到更有帮助的魔术方法。

接下来我们开始编写一些魔术方法吧！

1、`init`

考虑以下的Vector2D类：

class Vector2D:
    pass

当创建一个类并实例化一个对象时，可以像这样添加属性：obj_name.attribute_name = value。

不过，与其在创建的每个实例中手动添加属性，还不如在实例化对象时用一种方法来初始化这些属性。

为此，我们可以定义__init__方法。接下来让我们为Vector2D类定义__init__方法：

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

v = Vector2D(3, 5)

2、`repr`

当你尝试检查或打印实例化的对象时，会发现并没有得到任何有用的信息。

v = Vector2D(3, 5)
print(v)
Output >>> <__main__.Vector2D object at 0x7d2fcfaf0ac0>

这就是为什么应该添加一个表示字符串，即对象的字符串表示形式。为此，可以添加一个__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__方法通常用于调试。

3、`str`

__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__方法。

4、`eq`

接下来，让我们添加一个方法来检查任意两个Vector2D类的对象是否相等。如果两个向量对象具有完全相同的x和y坐标，则它们是相等的。

现在创建两个x和y坐标值均相等的Vector2D对象，并比较它们是否相等：

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 >>> True

5、`len`

Python的内置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 2

v = Vector2D(3, 5)
print(len(v))