访问内部字典时__getitem__开销大

Question

我有以下地图类，它使用字典作为其基本数据结构：

class Map:
''' A map object to handle the game world
'''
    def __init__(self, initial_grid=None, rooms=None, default=None):
        self.grid = initial_grid
        if not self.grid:
            self.grid = {}
        self.default = default

它有一个访问方法，尝试从它的网格中获取信息，但如果它不能返回默认参数

    def __getitem__(self, key):
        return self.grid.get(key, self.default)

现在在我的代码的另一部分，我让这个函数运行了很长时间，给我以下 cProfile 结果：

ncalls  tottime  percall  cumtime  percall filename:lineno(function)
2510000    0.739    0.000    0.926    0.000 map.py:57(__getitem_ _)
2510001    0.187    0.000    0.187    0.000 {method 'get' of 'dict' objects}

如果我理解正确，函数中的大部分时间 （~80%） 都没有花在检索值上，那么在这段时间里到底发生了什么？有什么方法可以改善吗？

Answer 1

涉及很多字典：self.grid是一个__getitem(…)，它从自字典中获取网格，然后self.grid.get是另一个，最后self.default从自我获得default条目。

若要减少get的数量，可能需要向类中添加以下内容：

__slots__ = ('grid', 'default') # add whatever field you have

这应该会大大加快您的代码速度，因为插槽访问被简化为简单的数组访问。