小编典典

是否有内置的dict.get()的递归版本?

python

我有一个嵌套的字典对象,并且希望能够检索具有任意深度的键的值。我可以通过子类化做到这一点dict

>>> class MyDict(dict):
...     def recursive_get(self, *args, **kwargs):
...         default = kwargs.get('default')
...         cursor = self
...         for a in args:
...             if cursor is default: break
...             cursor = cursor.get(a, default)
...         return cursor
... 
>>> d = MyDict(foo={'bar': 'baz'})
>>> d
{'foo': {'bar': 'baz'}}
>>> d.get('foo')
{'bar': 'baz'}
>>> d.recursive_get('foo')
{'bar': 'baz'}
>>> d.recursive_get('foo', 'bar')
'baz'
>>> d.recursive_get('bogus key', default='nonexistent key')
'nonexistent key'

但是,我不需要子类化dict即可获得此行为。是否有一些内置方法具有相同或相似的行为?如果不是,是否有任何提供此行为的标准或外部模块?

我目前正在使用Python 2.7,尽管我也很好奇也听说过3.x解决方案。


阅读 196

收藏
2020-12-20

共1个答案

小编典典

一个非常常见的模式是使用空dict作为默认值:

d.get('foo', {}).get('bar')

如果您有多个键,则可以使用reduce(请注意,在Python 3中reduce必须导入:)from functools import reduce以多次应用该操作

reduce(lambda c, k: c.get(k, {}), ['foo', 'bar'], d)

当然,您应该考虑将其包装到一个函数(或方法)中:

def recursive_get(d, *keys):
    return reduce(lambda c, k: c.get(k, {}), keys, d)
2020-12-20