Reputation: 1694
I am trying to retrieve a single level from a nested dictionary. So for instance, given the dictionary below, I would expect to see the following results for the first and second level.
nested_dict = {
'foo':'bar',
'baz':{
'foo':'baz'
}}
# level 0
{'foo':'bar'}
# level 1
{'foo':'baz'}
I can retrieve the first level using a dict
comprehension:
{k:v for (k,v) in nested_dict.items() if type(v) is not dict}
>>> {'foo':'bar'}
Or retrieve a specified level using a recursion:
def get_level(nested_dict, level):
if level == 0:
return {k:v for (k,v) in nested_dict.items() if type(v) is not dict}
else:
this_level = {}
for (k,v) in nested_dict.items():
if type(v) is dict:
this_level.update(v)
return get_level(this_level, level - 1)
get_level(nested_dict, 1)
>>> {'foo':'baz'}
I am now wondering if there is a more Pythonic/clean/out of the box way to retrieve the levels of nested dictionaries (as I already did above), if necessary with help of a package.
Upvotes: 4
Views: 1257
Reputation: 195643
As stated in the comments, I'm assuming you don't have duplicates:
nested_dict = {
'foo':'bar',
'baz':{
'foo':'baz',
'fee': {
'foo': 'fee'
}
},
'baz2':{
'foo2':'baz2'
}
}
def get_level(dct, level):
if level == 0:
yield from ((k, v) for k, v in dct.items() if not isinstance(v, dict))
else:
yield from ((kk, vv) for v in dct.values() if isinstance(v, dict) for kk, vv in get_level(v, level-1))
print(dict(get_level(nested_dict, 1)))
Prints:
{'foo': 'baz', 'foo2': 'baz2'}
print(dict(get_level(nested_dict, 2)))
Prints:
{'foo': 'fee'}
Upvotes: 1