從嵌套字典中檢索單級的 Pythonic 方法

Question

我正在嘗試從嵌套字典中檢索單個級別。 因此，例如，給定下面的字典，我希望看到第一級和第二級的以下結果。

nested_dict = {
  'foo':'bar',
  'baz':{
    'foo':'baz'
  }}

# level 0
{'foo':'bar'}

# level 1
{'foo':'baz'}

我可以使用dict理解檢索第一級：

{k:v for (k,v) in nested_dict.items() if type(v) is not dict}
>>> {'foo':'bar'}

或使用遞歸檢索指定級別：

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'}

我現在想知道是否有更 Pythonic/clean/out of the box 的方式來檢索嵌套字典的級別（正如我在上面已經做過的那樣），如果需要的話，在 package 的幫助下。

Answer 1

如評論中所述，我假設您沒有重復項：

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

印刷：

{'foo': 'baz', 'foo2': 'baz2'}

---

print(dict(get_level(nested_dict, 2)))

印刷：

{'foo': 'fee'}