【Intermediate Python】十、__slots__ 魔法

在 Python 中,每个类都有实例属性。默认情况下 Python 用一个字典来保存一个对象的实例属性。这非常有用,因为它允许我们在运行时去设置任意的新属性。

然而,对于有着已知属性的小类来说,它可能是个瓶颈。这个字典浪费了很多内存。Python 不能在对象创建时直接分配一个固定量的内存来保存所有的属性。因此如果你创建许多对象(我指的是成千上万个),它会消耗掉很多内存。

不过还是有一个方法来规避这个问题。这个方法需要使用 __slots__ 来告诉 Python 不要使用字典,而是只给定一个固定集合的属性分配空间。

下面是一个是否使用 __slots__ 的例子:

# 不使用__slots__
class MyClass:
    def __init__(self, name, identifier):
        self.name = name
        self.identifier = identifier
        self.set_up()
		
	# ...
	
# 使用__slots__
class Myclass:
    __slots__ = ['name', 'identifier']
    def __init__(self, name, identifier):
        self.name = name
        self.identifier = identifier
        self.set_up()
		
    # ...

第二段代码会为你的内存减轻负担。通过这个技巧,有些人已经看到内存占用率几乎 40%——50% 的减少。

稍微多嘴说一句,你也许需要试一下 PyPy。它已经默认地做了所有这些优化。