Defaultdict bersarang dari defaultdict

Apakah ada cara untuk membuat defaultdict juga menjadi default untuk defaultdict? (Yaitu default default rekursif tingkat tak terbatas?)

Saya ingin dapat melakukan:

x = defaultdict(...stuff...)
x[0][1][0]
{}

Jadi, saya bisa melakukannya x = defaultdict(defaultdict), tetapi itu hanya level kedua:

x[0]
{}
x[0][0]
KeyError: 0

Ada resep yang bisa melakukan ini. Tetapi dapatkah itu dilakukan hanya dengan menggunakan argumen defaultdict yang normal?

Catatan ini menanyakan bagaimana cara melakukan defaultdict rekursif tingkat infinite, jadi berbeda dengan Python: defaultdict of defaultdict? , yang merupakan cara melakukan defaultdict dua tingkat.

Saya mungkin hanya akan berakhir menggunakan pola tandan , tetapi ketika saya menyadari saya tidak tahu bagaimana melakukan ini, itu membuat saya tertarik.

Untuk jumlah level yang berubah-ubah:

def rec_dd():
    return defaultdict(rec_dd)

>>> x = rec_dd()
>>> x['a']['b']['c']['d']
defaultdict(<function rec_dd at 0x7f0dcef81500>, {})
>>> print json.dumps(x)
{"a": {"b": {"c": {"d": {}}}}}

Tentu saja Anda juga bisa melakukan ini dengan lambda, tetapi saya menemukan lambda kurang bisa dibaca. Bagaimanapun akan terlihat seperti ini:

rec_dd = lambda: defaultdict(rec_dd)

Jawaban lain di sini memberi tahu Anda cara membuat defaultdictyang berisi "tak terhingga banyaknya" defaultdict, tetapi mereka gagal menjawab apa yang saya pikir mungkin adalah kebutuhan awal Anda yang hanya memiliki standar default dua kedalaman.

Anda mungkin telah mencari:

defaultdict(lambda: defaultdict(dict))

Alasan mengapa Anda lebih suka konstruksi ini adalah:

• Ini lebih eksplisit daripada solusi rekursif, dan karenanya lebih mudah dipahami oleh pembaca.
• Ini memungkinkan "daun" defaultdictmenjadi sesuatu selain dari kamus, misalnya ,: defaultdict(lambda: defaultdict(list))ataudefaultdict(lambda: defaultdict(set))

Ada trik bagus untuk melakukan itu:

tree = lambda: defaultdict(tree)

Kemudian Anda bisa membuatnya xdengan x = tree().

Mirip dengan solusi BrenBarn, tetapi tidak mengandung nama variabel treedua kali, sehingga berfungsi bahkan setelah perubahan ke kamus variabel:

tree = (lambda f: f(f))(lambda a: (lambda: defaultdict(a(a))))

Kemudian Anda dapat membuat masing-masing xdengan x = tree().

Untuk defversi, kita bisa menggunakan lingkup penutupan fungsi untuk melindungi struktur data dari cacat di mana instans yang ada berhenti bekerja jika treenamanya kembali. Ini terlihat seperti ini:

from collections import defaultdict

def tree():
    def the_tree():
        return defaultdict(the_tree)
    return the_tree()

Saya juga akan mengusulkan implementasi lebih banyak gaya OOP, yang mendukung bersarang tanpa batas serta diformat dengan benar repr.

class NestedDefaultDict(defaultdict):
    def __init__(self, *args, **kwargs):
        super(NestedDefaultDict, self).__init__(NestedDefaultDict, *args, **kwargs)

    def __repr__(self):
        return repr(dict(self))

Pemakaian:

my_dict = NestedDefaultDict()
my_dict['a']['b'] = 1
my_dict['a']['c']['d'] = 2
my_dict['b']

print(my_dict)  # {'a': {'b': 1, 'c': {'d': 2}}, 'b': {}}

di sini adalah fungsi rekursif untuk mengubah dict default rekursif ke dict normal

def defdict_to_dict(defdict, finaldict):
    # pass in an empty dict for finaldict
    for k, v in defdict.items():
        if isinstance(v, defaultdict):
            # new level created and that is the new value
            finaldict[k] = defdict_to_dict(v, {})
        else:
            finaldict[k] = v
    return finaldict

defdict_to_dict(my_rec_default_dict, {})

Saya mendasarkan jawaban Andrew ini di sini. Jika Anda mencari untuk memuat data dari json atau dikt yang ada ke dalam nerd defaultdict lihat contoh ini:

def nested_defaultdict(existing=None, **kwargs):
    if existing is None:
        existing = {}
    if not isinstance(existing, dict):
        return existing
    existing = {key: nested_defaultdict(val) for key, val in existing.items()}
    return defaultdict(nested_defaultdict, existing, **kwargs)

https://gist.github.com/nucklehead/2d29628bb49115f3c30e78c071207775