python:查找具有值的所有键

Question

我有一个如下的字典:

mydict = {'a' : 'apple',
     'b' : 'bobb',
    'c' : {
         'd' : 'dog'
       },
      'e' : 'dog'
     }

假设我想找到值为' dog'的所有键.在嵌套的情况下,键应该用点分隔.

所以输出应该是一个列表 ['e', 'c.d']

现在,如果我在python 3中编写下面的代码,它只输出' e'.

print(list(mydict.keys())[list(mydict.values()).index('dog')])

如何获取嵌套密钥？

Answer 1

您可以使用如下的递归函数:

def find_key(mydict, pre=tuple()):
    for key, value in mydict.items():
        if isinstance(value, dict):
            yield from find_key(value, pre=pre+(key,))
        elif value == 'dog':
            if pre:
                yield '.'.join(pre + (key,))
            else:
                yield key

测试:

In [23]: list(find_key(mydict))
Out[23]: ['e', 'c.d']
In [26]: mydict = {'a' : 'apple',
     'b' : 'bobb',
    'c' : {
         'd' : 'dog'
       },
      'e' : 'dog',
     'k':{'f':{'c':{'x':'dog'}}}}

In [27]: 

In [27]: list(find_key(mydict))
Out[27]: ['k.f.c.x', 'e', 'c.d']