Вложенные defaultdict на основе ключа внутри словаря

Question

Я пытаюсь создать defaultdict в defaultdict на основе значения ключа. Мое мышление здесь может быть совершенно неправильным, но вот код для основного defaultdict;

def record(): 
    return { 
     'count': 0, 
     'key1': Counter(),  
    } 

1_record = defaultdict(record) 

Но что, если я хочу добавить ключ как defaultdict как это;

def record(): 
    return { 
     'count': 0, 
     'key1': Counter(), 
     'key2': { 
      'count': 0, 
      'nested_key1': Counter() 

    } 
} 

В приведенном выше примере, как я могу сделать 'key2' defaultdict? Возможно ли это, или я неправильно подхожу к проблеме?

Answer 1

Вы, безусловно, можете иметь «рекурсивный» defaultdict:

>>> from collections import defaultdict 
>>> def record(): 
... return { 
...  'key': defaultdict(record) 
... } 
... 
>>> d = defaultdict(record) 
>>> 
>>> d['foo'] 
{'key': defaultdict(<function record at 0x10b396f50>, {})} 
>>> d['foo']['key']['bar'] 
{'key': defaultdict(<function record at 0x10b396f50>, {})} 
>>> d 
defaultdict(<function record at 0x10b396f50>, {'foo': {'key': defaultdict(<function record at 0x10b396f50>, {'bar': {'key': defaultdict(<function record at 0x10b396f50>, {})}})}}) 

Однако замена из названия клавиш на разных уровнях, вероятно, потребуется некоторые специальный корпус и сделает код немного более грязным. ..