使用列表/元组元素作为键创建字典

Question

我需要生成这样的字典：

{
  'newEnv': {
     'newProj': {
        'newComp': {
           'instances': [],
           'n_thing': 'newThing'
        }
     }
  }
}

来自一个元组，像这样：('newEnv','newProj','newComp','newThing') 但前提是它不存在。所以，我尝试了这个：

myDict = {}
(env,proj,comp,thing) = ('newEnv','newProj','newComp','newThing')

if env not in myDict:
    myDict[env] = {}
if proj not in myDict[env]:
    myDict[env][proj] = {}
if comp not in myDict[env][proj]:
    myDict[env][proj][comp] = {'n_thing': thing, 'instances': []}

这非常有效，但不确定它的效率如何，或者我是否应该这样做。有什么建议）？？

Answer 1

您可以使用循环（只有前 3 个键，newThing 不是链中的键）：

myDict = {}
path = ('newEnv','newProj','newComp')
current = myDict
for key in path:
    current = current.setdefault(key, {})

current 最终成为最里面的字典，让您可以在其中设置 'n_thing' 和 'instances' 键。

您可以使用reduce() 将其折叠成单行：

myDict = {}
path = ('newEnv','newProj','newComp')
reduce(lambda d, k: d.setdefault(k, {}), path, myDict)

reduce 调用返回最里面的字典，因此您可以使用它来分配最终值：

myDict = {}
path = ('newEnv','newProj','newComp')
inner = reduce(lambda d, k: d.setdefault(k, {}), path, myDict)
inner.update({'n_thing': 'newThing', 'instances': []})

Answer 2

您可以使用 defaultdicts 的 defaultdict 做类似的事情，也许稍微简单一些（有关讨论，请参阅 defaultdict of defaultdict, nested）

tree = lambda: defaultdict(tree)

base = tree()
for x in mytuple[:-2]:
    base = base[x]
base[x] = mytuple[-1]

这与martijn 的非常相似，只是使用默认功能来创建subdicts，而不是直接使用setdefault。

这也可以让你直接输入

myDict[env][proj][comp].setdefault('instances', list()).append(queue)

如果那是您真正想要的。 （不幸的是，没有办法删除 setdefault；毕竟，我不提前知道你想要一个列表还是字典。你只有一个默认值）....