根据 N 个数组项查找嵌套字典值

Question

我正在尝试解决这个问题：我有一个值数组，这些值可能 是字典中的键。然后，如果它们不存在，我想添加它们。

myarr = ['one', 'two', 'three', 'four']
mydict = {'one': {'two': {'three': {}}}}

for item in myarr:
   if item in mydict:
      (this is where my brain shuts off)
   else:
      (via some sort of magic)
      mydict[insert_magic_here] = {'four': {}}

我尝试使用 for i in range(len(myarr)): 自动递增，但没有成功。我也尝试过使用mydict = mydict[i] 来更深入地了解字典，但这让我陷入了困境。

感谢任何帮助！

Answer 1

这是一个有趣的练习；有点类似于将元素附加到linked list。通常的解决方案是使用current 变量来跟踪您当前在序列中的位置，并将current 更新为每次迭代的下一个“节点”。

myarr = ['one', 'two', 'three', 'four']
mydict = {'one': {'two': {'three': {}}}}

current = mydict

for item in myarr:
    # insert if not present
    if item not in current:
        current[item] = dict()
    # advance to next
    current = current[item]

结果：

>>> mydict
{'one': {'two': {'three': {'four': {}}}}}

原始解决方案的问题可能是您使用mydict 作为“当前”变量，这意味着您丢失了对主字典（第一个“节点”）的原始引用。