Python: defaultdict ของ defaultdict หรือไม่

มีวิธีที่จะมีdefaultdict(defaultdict(int))เพื่อให้รหัสต่อไปนี้ทำงานได้หรือไม่

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

dจะต้องมีการสร้างโฆษณาขึ้นอยู่กับองค์ประกอบx.aและx.b

ฉันสามารถใช้:

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

แต่ฉันจะไม่สามารถใช้:

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

ใช่เช่นนี้:

defaultdict(lambda: defaultdict(int))

อาร์กิวเมนต์ของdefaultdict(ในกรณีนี้lambda: defaultdict(int)) จะถูกเรียกเมื่อคุณพยายามเข้าถึงคีย์ที่ไม่มีอยู่ ค่าตอบแทนของมันจะถูกกำหนดให้เป็นค่าใหม่ของคีย์นี้ซึ่งหมายความว่าในกรณีของเราค่าของจะd[Key_doesnt_exist]defaultdict(int)

ถ้าคุณพยายามที่จะเข้าถึงคีย์จาก defaultdict สุดท้ายนี้คือd[Key_doesnt_exist][Key_doesnt_exist]มันจะกลับ 0 ซึ่งเป็นค่าตอบแทนของการโต้แย้งของสุดท้าย defaultdict int()คือ

พารามิเตอร์ของตัวสร้าง defaultdict เป็นฟังก์ชันที่จะถูกเรียกสำหรับการสร้างองค์ประกอบใหม่ ดังนั้นลองใช้แลมบ์ดากัน!

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

ตั้งแต่ Python 2.7 มีวิธีแก้ปัญหาที่ดียิ่งขึ้นเมื่อใช้ 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})

คุณสมบัติโบนัสบางอย่าง

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

สำหรับข้อมูลเพิ่มเติมโปรดดูPyMOTW - ชุดรวม - ประเภทข้อมูลคอนเทนเนอร์และเอกสาร Python - ชุด

ฉันคิดว่ามันใช้งานได้ดีกว่าเล็กน้อยpartial:

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

แน่นอนว่านี่เป็นแลมบ์ดา

สำหรับการอ้างอิงเป็นไปได้ที่จะใช้วิธีการซ้อนกันของdefaultdictโรงงานทั่วไปผ่าน

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

ความลึกกำหนดจำนวนพจนานุกรมซ้อนกันก่อนประเภทที่กำหนดไว้ในdefault_factoryถูกนำมาใช้ ตัวอย่างเช่น:

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

คำตอบก่อนหน้านี้ได้รับการแก้ไขวิธีที่จะทำให้สองระดับหรือ defaultdictn-ระดับ ในบางกรณีคุณต้องการอนันต์:

def ddict():
    return defaultdict(ddict)

การใช้งาน:

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

คนอื่น ๆ ตอบคำถามของคุณอย่างถูกต้องถึงวิธีการทำงานดังต่อไปนี้:

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

อีกทางเลือกหนึ่งคือใช้สิ่งอันดับสำหรับคีย์:

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

สิ่งที่ดีเกี่ยวกับวิธีนี้คือมันง่ายและสามารถขยายได้อย่างง่ายดาย หากคุณต้องการการแมปสามระดับที่ลึกเพียงใช้ tuple สามรายการสำหรับคีย์