Create and lookup 2D dictionary with multiple keys per value

Keith picture Keith · Nov 10, 2012 · Viewed 13.9k times · Source

I think I want to make a 2D dictionary with multiple keys per value.

I know how to make a 2D dictionary using defaultdict:

from collections import defaultdict
2d_dict = defaultdict(dict)

2d_dict['canned_food']['spam'] = 'delicious'

And I know that using regular dictionaries you can make with multiple keys like:

dictionary={('food','canned_food'):spam}

But I want to do something like lookup by tuple-of-keys:

2d_dict[('canned_food','food')]['spam'] = 'delicious'

In the first dimension of dictionary I need ~25 keys per value. Is there a way to do this with defaultdict?

Even if there is a way to do it with dicts is this a reasonable way to make a simple multidimensional lookup table?

Answer

Zero Piraeus picture Zero Piraeus · Nov 10, 2012

Apart from 2d_dict being an invalid variable name (it starts with a digit), your existing solution already works:

>>> from collections import defaultdict
>>> d2_dict = defaultdict(dict)
>>> d2_dict[('canned_food', 'food')]['spam'] = 'delicious'
>>> d2_dict
defaultdict(<type 'dict'>, {('canned_food', 'food'): {'spam': 'delicious'}})

In fact, you don't even need the parentheses - Python will still recognise your key as a tuple:

>>> d2_dict['fresh_food', 'food']['eggs'] = 'delicious'
>>> d2_dict
defaultdict(<type 'dict'>, {('canned_food', 'food'): {'spam': 'delicious'},
('fresh_food', 'food'): {'eggs': 'delicious'}})

... and, yes, it's a perfectly reasonable way to build a 2D+1D lookup table.

If you want to build a 3D lookup table using nested dicts instead of tuple keys, this works:

>>> d3_dict = defaultdict(lambda: defaultdict(dict))
>>> d3_dict['dried_food']['food']['jerky'] = 'chewy'
>>> d3_dict
defaultdict(<function <lambda> at 0x7f20af38a2a8>, 
{'dried_food': defaultdict(<type 'dict'>, {'food': {'jerky': 'chewy'}})})