criando uma lista padrão em python
Estou tentando criar uma lista equivalente para o muito útilcollections.defaultdict
. O design a seguir funciona bem:
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)
Veja como você o usa:
>>> dl = defaultlist(lambda:'A')
>>> dl[2]='B'
>>> dl[4]='C'
>>> dl
['A', 'A', 'B', 'A', 'C']
O que devo adicionar à lista padrão para dar suporte ao seguinte comportamento?
>>> dl = defaultlist(dict)
>>> dl[2]['a'] = 1
>>> dl
[{}, {}, {'a':1}]