Python: defaultdict of defaultdict?

Apakah ada cara untuk memiliki defaultdict(defaultdict(int))agar kode berikut berfungsi?

for x in stuff:
    d[x.a][x.b] += x.c_int

dperlu dibangun ad-hoc, tergantung pada x.adan x.belemen.

Saya bisa menggunakan:

for x in stuff:
    d[x.a,x.b] += x.c_int

tapi kemudian saya tidak bisa menggunakan:

d.keys()
d[x.a].keys()

Ya seperti ini:

defaultdict(lambda: defaultdict(int))

Argumen a defaultdict(dalam hal ini adalah lambda: defaultdict(int)) akan dipanggil ketika Anda mencoba mengakses kunci yang tidak ada. Nilai pengembaliannya akan ditetapkan sebagai nilai baru dari kunci ini, yang berarti dalam kasus kami nilai d[Key_doesnt_exist]akan defaultdict(int).

Jika Anda mencoba mengakses kunci dari default default terakhir ini yaitu d[Key_doesnt_exist][Key_doesnt_exist]akan mengembalikan 0, yang merupakan nilai pengembalian argumen dari default default yaitu int().

Parameter ke konstruktor defaultdict adalah fungsi yang akan dipanggil untuk membangun elemen baru. Jadi mari kita gunakan lambda!

>>> from collections import defaultdict
>>> d = defaultdict(lambda : defaultdict(int))
>>> print d[0]
defaultdict(<type 'int'>, {})
>>> print d[0]["x"]
0

Sejak Python 2.7, ada solusi yang lebih baik lagi menggunakan Counter :

>>> from collections import Counter
>>> c = Counter()
>>> c["goodbye"]+=1
>>> c["and thank you"]=42
>>> c["for the fish"]-=5
>>> c
Counter({'and thank you': 42, 'goodbye': 1, 'for the fish': -5})

Beberapa fitur bonus

>>> c.most_common()[:2]
[('and thank you', 42), ('goodbye', 1)]

Untuk informasi lebih lanjut lihat PyMOTW - Koleksi - tipe data wadah dan Dokumentasi Python - koleksi

Saya merasa sedikit lebih elegan untuk digunakan partial:

import functools
dd_int = functools.partial(defaultdict, int)
defaultdict(dd_int)

Tentu saja, ini sama dengan lambda.

Untuk referensi, dimungkinkan untuk menerapkan defaultdictmetode pabrik bersarang generik melalui:

from collections import defaultdict
from functools import partial
from itertools import repeat


def nested_defaultdict(default_factory, depth=1):
    result = partial(defaultdict, default_factory)
    for _ in repeat(None, depth - 1):
        result = partial(defaultdict, result)
    return result()

Kedalaman menentukan jumlah kamus bersarang sebelum jenis yang didefinisikan default_factorydigunakan. Sebagai contoh:

my_dict = nested_defaultdict(list, 3)
my_dict['a']['b']['c'].append('e')

Jawaban sebelumnya telah membahas cara membuat dua level atau n-level defaultdict. Dalam beberapa kasus Anda ingin yang tak terbatas:

def ddict():
    return defaultdict(ddict)

Pemakaian:

>>> d = ddict()
>>> d[1]['a'][True] = 0.5
>>> d[1]['b'] = 3
>>> import pprint; pprint.pprint(d)
defaultdict(<function ddict at 0x7fcac68bf048>,
            {1: defaultdict(<function ddict at 0x7fcac68bf048>,
                            {'a': defaultdict(<function ddict at 0x7fcac68bf048>,
                                              {True: 0.5}),
                             'b': 3})})

Orang lain telah menjawab dengan benar pertanyaan Anda tentang cara membuat yang berikut ini berfungsi:

for x in stuff:
    d[x.a][x.b] += x.c_int

Alternatifnya adalah menggunakan tuple untuk kunci:

d = defaultdict(int)
for x in stuff:
    d[x.a,x.b] += x.c_int
    # ^^^^^^^ tuple key

Hal yang menyenangkan tentang pendekatan ini adalah sederhana dan dapat dengan mudah diperluas. Jika Anda membutuhkan pemetaan tiga level, gunakan tuple tiga item untuk kunci tersebut.