Python: defaultdict của defaultdict?

Có cách nào để có một defaultdict(defaultdict(int))đoạn mã sau hoạt động không?

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

dcần phải được xây dựng đặc biệt, tùy thuộc x.avà x.bcác yếu tố.

Tôi có thể sử dụng:

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

nhưng sau đó tôi sẽ không thể sử dụng:

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

Có như thế này:

defaultdict(lambda: defaultdict(int))

Đối số của một defaultdict(trong trường hợp này là lambda: defaultdict(int)) sẽ được gọi khi bạn cố gắng truy cập khóa không tồn tại. Giá trị trả về của nó sẽ được đặt là giá trị mới của khóa này, có nghĩa là trong trường hợp của chúng tôi, giá trị của d[Key_doesnt_exist]sẽ là defaultdict(int).

Nếu bạn cố gắng truy cập một khóa từ defaultdict cuối cùng này, tức là d[Key_doesnt_exist][Key_doesnt_exist]nó sẽ trả về 0, đó là giá trị trả về của đối số của defaultdict cuối cùng tức là int().

Tham số cho hàm tạo defaultdict là hàm sẽ được gọi để xây dựng các phần tử mới. Vì vậy, hãy sử dụng lambda!

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

Kể từ Python 2.7, có một giải pháp thậm chí tốt hơn bằng cách sử dụng 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})

Một số tính năng thưởng

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

Để biết thêm thông tin, hãy xem PyMOTW - Bộ sưu tập - Kiểu dữ liệu chứa và Tài liệu Python - bộ sưu tập

Tôi thấy nó thanh lịch hơn một chút để sử dụng partial:

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

Tất nhiên, điều này giống như lambda.

Để tham khảo, có thể thực hiện defaultdictphương pháp nhà máy lồng nhau thông qua:

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()

Độ sâu xác định số lượng từ điển lồng nhau trước khi loại được xác định trong default_factoryđược sử dụng. Ví dụ:

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

Các câu trả lời trước đã đề cập đến cách tạo một cấp độ hai hoặc cấp độ n defaultdict. Trong một số trường hợp bạn muốn một cái vô hạn:

def ddict():
    return defaultdict(ddict)

Sử dụng:

>>> 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})})

Những người khác đã trả lời chính xác câu hỏi của bạn về cách làm cho những điều sau đây hoạt động:

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

Một cách khác là sử dụng bộ dữ liệu cho các phím:

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

Điều hay ho của phương pháp này là nó đơn giản và có thể dễ dàng mở rộng. Nếu bạn cần ánh xạ ba cấp độ sâu, chỉ cần sử dụng bộ ba mục cho khóa.