在嵌套的Python字典中搜索并记录“路径”

Question

[借助此answer，我试图提供一个函数，该函数在嵌套的python dict中的键之后进行搜索，并还记录每个匹配项的“路径”。我的函数（见下文）似乎起作用，但是无法将结果保存在列表中（见代码输出）。我可以肯定，困难在于yield命令，但我还无法弄清楚。

o={
  'dict1': {
    'dict11': {
      'entry11_1':1,
      'entry11_2':2,
    },
    'dict12': {
      'entry12_1':12,
      'entry12_2':22,
    },
  },
  'dict2': {
    'dict21': {
      'entry21_1':21,
    }
  },
}


curr_pos=[]
def gen_dict_extract(key, var):
  global curr_pos
  if hasattr(var,'iteritems'):
    for k, v in var.iteritems():
      #print curr_pos
      if k == key:
        yield v,curr_pos
      if isinstance(v, dict):
        curr_pos.append(k)
        for result in gen_dict_extract(key, v):
          yield result
      elif isinstance(v, list):
        for d in v:
          for result in gen_dict_extract(key, d):
            yield result
    if len(curr_pos)>0:
      curr_pos.pop()


result_list=[]
for ind,i in enumerate(gen_dict_extract('entry12_1',o)):
  result_list.append(i)
  print result_list[-1]
print result_list[-1]

输出：

(12, ['dict1', 'dict12'])
(12, [])

Answer 1

在gen_dict_extract中，使用全局列表curr_pos，并在找到键（yield v,curr_pos）时直接产生。但是列表是可变类型，您稍后可以对其进行修改（curr_pos.pop()）

您存储在result_list中的内容只是对全局对象的引用，因此它包含循环内的期望值，但在循环结束时将其清空。您应该只在屈服时间返回一个浅表副本：yield v,curr_pos[:]

您将获得预期的效果：

(12, ['dict1', 'dict12'])
(12, ['dict1', 'dict12'])

---

顺便说一句，要避免使用全局列表，可以将列表作为可选参数传递：

def gen_dict_extract(key, var, curr_pos = None):
    if curr_pos is None:
        curr_pos = []
    ...
        for result in gen_dict_extract(key, v, curr_pos):
    ...
          for result in gen_dict_extract(key, d, curr_pos):
    ...

这将确保您在每次新调用时都使用一个新列表，同时在递归时正确传递它

Answer 2

问题是我是一个tupple对象。您需要复制我以避免覆盖。

import copy
result_list = []
for in ind in enumerate(gen_dict_extract('entry12_1',o)):
    result_list.append(copy.deepcopy(i))
print result_list

Answer 3

为了完整起见，以下是带有Serge建议的版本。另外，我进行了一些其他更改，以便该函数能够处理任何嵌套列表和字典组合。

def gen_dict_extract(key, var,curr_pos=None):
  """
  key: key to search for
  var: nested dict to search in 
  """
  #print curr_pos
  if curr_pos is None:
    curr_pos=[]
  if hasattr(var,'iteritems'):
    for k, v in var.iteritems():
      curr_pos.append(k)
      if k == key:
        yield v,curr_pos[:]
      if isinstance(v, dict):
        for result in gen_dict_extract(key, v,curr_pos):
          yield result
      elif isinstance(v, list):
        curr_pos.append(0)
        for ind,d in enumerate(v):
          curr_pos.pop()
          curr_pos.append(ind)
          for result in gen_dict_extract(key, d,curr_pos):
            yield result
        curr_pos.pop()
      curr_pos.pop()
  elif isinstance(var, list):
    curr_pos.append(0)
    for ind,d in enumerate(var):
      curr_pos.pop()
      curr_pos.append(ind)
      for result in gen_dict_extract(key, d,curr_pos):
        yield result
    curr_pos.pop()

Answer 4

函数返回路径和值以及元组列表。

def dict_key_lookup(_dict, key, path=[]):
    results = []
    if isinstance(_dict, dict):
        if key in _dict:
            results.append((path+[key], _dict[key]))
        else:
            for k, v in _dict.items():
                results.extend(dict_key_lookup(v, key, path= path+[k]))
    elif isinstance(_dict, list):
        for index, item in enumerate(_dict):
            results.extend(dict_key_lookup(item, key, path= path+[index]))
    return results

希望这会有所帮助。