Đặt giá trị dict lồng nhau và tạo các phím trung gian

Question

Tôi cảm thấy như tôi đã thấy cách để thực hiện việc này gần đây. Nói rằng tôi đã có một dict trống và tôi muốn đặt một giá trị trong một dict lồng nhau bên trong dict trống, nhưng rõ ràng là dict lồng nhau chưa được tạo ra. Có cách nào để tạo khóa trung gian không? Đây là những gì tôi muốn làm:

mydict = {} 
mydict['foo']['bar']['foobar'] = 25 

Nếu bạn thực thi mã này, bạn sẽ nhận được ngoại lệ KeyError cho 'foo'. Có chức năng tạo khóa trung gian không?

Cảm ơn.

Answer 1

from collections import defaultdict 
recursivedict = lambda: defaultdict(recursivedict) 
mydict = recursivedict() 

Khi bạn truy cập mydict['foo'], nó đặt mydict['foo'] để recursivedict khác. Nó sẽ thực sự xây dựng một recursivedict cho mydict['foo']['bar']['foobar'] là tốt, nhưng sau đó nó sẽ được ném ra bằng cách gán cho 25.

Answer 2

Không chắc lý do tại sao bạn muốn, nhưng:

>>> from collections import defaultdict as dd 
>>> mydict = dd(lambda: dd(lambda: {})) 
>>> mydict['foo']['bar']['foobar'] = 25 
>>> mydict 
defaultdict(<function <lambda> at 0x021B8978>, {'foo': defaultdict(<function <lambda> at 0x021B8618>, {'bar': {'foobar': 25}})}) 

Answer 3

Một lựa chọn thay thế - tùy thuộc vào mục đích sử dụng của bạn, là sử dụng các bộ như phím thay vì các từ điển lồng nhau:

mydict = {} 
mydict['foo', 'bar', 'foobar'] = 25 

này sẽ hoạt động hoàn toàn tốt trừ khi bạn muốn lấy cành cây tại bất kỳ điểm nào (bạn không thể nhận được mydict ['foo'] trong trường hợp này).

Nếu bạn biết có bao nhiêu lớp làm tổ mà bạn muốn, bạn cũng có thể sử dụng functools.partial thay vì lambda.

from functools import partial 
from collections import defaultdict 

tripledict = partial(defaultdict, partial(defaultdict, dict)) 
mydict = tripledict() 
mydict['foo']['bar']['foobar'] = 25 

nào một số người thấy dễ đọc hơn, và nhanh hơn để tạo ra các trường hợp so với giải pháp lambda dựa trên tương đương:

python -m timeit -s "from functools import partial" -s "from collections import defaultdict" -s "tripledefaultdict = partial(defaultdict, partial(defaultdict, dict))" "tripledefaultdict()" 
1000000 loops, best of 3: 0.281 usec per loop 

python -m timeit -s "from collections import defaultdict" -s "recursivedict = lambda: defaultdict(recursivedict)" "recursivedict()" 
1000000 loops, best of 3: 0.446 usec per loop 

Mặc dù, như mọi khi, không có điểm tối ưu hóa cho đến khi bạn biết có là một nút cổ chai, vì vậy hãy chọn những gì hữu ích nhất và có thể đọc được trước những gì là nhanh nhất.