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

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

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 2

在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 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()

在嵌套的Python dict中搜索并记录“path”

3 个答案: