python collections.defaultdict 分配但不查找

Question

我有以下代码：

from collections import *
nested_dict = lambda: defaultdict(nested_dict)
data = nested_dict()

这使我能够在字典中将任何新的“路径”写成一个衬里：

data['A']['B']['C']=3

这就是我想要的。 但是我想在运行时得到一个异常（对于任何不存在的路径）：

var = data['A']['XXX']['C']

我觉得写作时需要 defaultdict，阅读时需要普通 dict...

或者，有没有一种简单的好方法来检查默认字典中是否存在“路径”而不修改其内容...

我尝试在查找之前将 defaultdict 转换回 dict，希望：

dict(data)['A']['XXX']['C']

会引发异常...但它不断创建缺少的键...

Answer 1

您无法在此处区分查找和写入，因为在 data['A']['B']['C'] = 3 分配中创建中间结构的是 查找。 Python 先执行索引操作data['A']，然后再执行['B']，然后再分配给'C' 键。 __getitem__、__setitem__ 和 __missing__ 钩子没有提供足够的上下文来区分访问，然后在第二个示例中导致 'C' 分配与仅“阅读”'XXX'。

你真的只有3个选项：

• 不要使用defaultdict。编写时，显式使用dict.setdefault() 创建新的嵌套字典；您可以根据需要链接这些调用：

  var = {}
  var.setdefault('A', {}).setdefault('B', {})['C'] = 3
  

  或者你可以wrap recursive behaviour in a few functions。

• 完成编写后，创建defaultdict 结构的递归副本，以将其替换为dict 结构：

  def dd_to_d(dd):
      r = {}
      stack = [(r, dd)]
      while stack:
          target, dd = stack.pop()
          for key, value in dd.items():
              if isinstance(value, defaultdict):
                  sub = {}
                  stack.append((sub, value))
                  value = sub
              target[key] = value
      return r
  
  var = dd_to_d(var)
  

• 将所有 default_factory 属性设置为 None 以禁用为缺少的键创建新值：

  def disable_dd(dd):
      stack = [dd]
      while stack:
          dd = stack.pop()
          dd.default_factory = None
          for key, value in dd.items():
              if isinstance(value, defaultdict):
                  stack.append(value)
  
  disable_dd(var)
  

Answer 2

一个明显的解决方案是只使用普通的 dicts 和一个可以“实现”中间键的函数：

def write_path(d, path, value):
    for key in path[:-1]:
        d = d.setdefault(key, {})
    d[path[-1]] = value

d = {}

write_path(d, ['a', 'b', 'c'], 3)
print(d)
print(d['a']['b']['c'])
print(d['a']['b']['d'])

输出

{'a': {'b': {'c': 3}}}
3
Traceback (most recent call last):
  File "writedefaultdict.py", line 11, in <module>
    print(d['a']['b']['d'])
KeyError: 'd'