создание списка defaultlist в python
Я пытаюсь создать список, эквивалентный очень полезному 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']
что я должен добавить в defaultlist, чтобы поддержать следующее поведение?
>>> dl = defaultlist(dict)
>>> dl[2]['a'] = 1
>>> dl
[{}, {}, {'a':1}]
2 ответов
В приведенном примере вы сначала пытаетесь получить несуществующее значение в списке, как вы это делаете 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)
доступен пакет python:
$ pip install defaultlist
добавленные показатели по умолчанию не заполняются.
>>> from defaultlist import defaultlist
>>> l = defaultlist()
>>> l
[]
>>> l[2] = "C"
>>> l
[None, None, 'C']
>>> l[4]
>>> l
[None, None, 'C', None, None]
срезы и отрицательные показатели поддерживаются аналогично
>>> l[1:4]
[None, 'C', None]
>>> l[-3]
'C'
простые функции фабрики можно создать через лямбда.
>>> l = defaultlist(lambda: 'empty')
>>> l[2] = "C"
>>> l[4]
'empty'
>>> l
['empty', 'empty', 'C', 'empty', 'empty']
также возможно снабдить предварительные функции фабрики:
>>> def inc():
... inc.counter += 1
... return inc.counter
>>> inc.counter = -1
>>> l = defaultlist(inc)
>>> l[2] = "C"
>>> l
[0, 1, 'C']
>>> l[4]
4
>>> l
[0, 1, 'C', 3, 4]
посмотреть документация для получения дополнительной информации.