python - 在 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)