在python中创建默认列表

Question

我正在尝试创建一个非常有用的等效列表collections.defaultdict。以下设计效果很好：

class defaultlist(list):
    def __init__(self, fx):
        self._fx = fx
    def __setitem__(self, index, value):
        while len(self) <= index:
            self.append(self._fx())
        list.__setitem__(self, index, value)

使用方法如下：

>>> dl = defaultlist(lambda:'A')
>>> dl[2]='B'
>>> dl[4]='C'
>>> dl
['A', 'A', 'B', 'A', 'C']

我应该添加什么到默认列表中以支持以下行为？

>>> dl = defaultlist(dict)
>>> dl[2]['a'] = 1
>>> dl
[{}, {}, {'a':1}]

Answer 1

在给出的示例中，您首先尝试检索列表中不存在的值，就像这样做一样dl[2]['a']，Python首先检索列表中的第三个（索引2）元素，然后继续在该列表中获取名为“
a”的元素对象-因此，您还必须实现对__getitem__方法的自动扩展行为，如下所示：

class defaultlist(list):
    def __init__(self, fx):
        self._fx = fx
    def _fill(self, index):
        while len(self) <= index:
            self.append(self._fx())
    def __setitem__(self, index, value):
        self._fill(index)
        list.__setitem__(self, index, value)
    def __getitem__(self, index):
        self._fill(index)
        return list.__getitem__(self, index)