我有这样的数据(是的,这些元组保证正好有5个元素):
ts = ([('a','b','c','d','e'), ('v','w','x','y','z'),
('f','g','h','i','j'), ('a','foo','bar',1,2),
('f','g','baz',1,3), ('f','g','baz',3,4)])
我正在尝试将其解析为嵌套字典结构,如下所示:
d = {
'a': {
'b': {
'c': [('d','e')]
},
'foo': {
'bar': [(1,2)]
}
},
'f': {
'g': {
'h': [('i', 'j')],
'baz': [(1,3), (3,4)]
}
},
'v': {
'w': {
'x': [('y', 'z')]
}
}
}
这是我到目前为止所拥有的;它似乎运作正常:
>>> d = {}
>>> for t in ts:
... if t[0] not in d:
... d[t[0]] = {t[1]: {t[2]: [(t[3], t[4])]}}
... elif t[1] not in d[t[0]]:
... d[t[0]][t[1]] = {t[2]: [(t[3], t[4])]}
... elif t[2] not in d[t[0]][t[1]]:
... d[t[0]][t[1]][t[2]] = [(t[3], t[4])]
... else:
... d[t[0]][t[1]][t[2]].append((t[3],t[4]))
...
>>> d
{'a': {'b': {'c': [('d', 'e')]}, 'foo': {'bar': [(1, 2)]}}, 'f': {'g': {'h': [('i', 'j')], 'baz': [(1, 3), (3, 4)]}}, 'v': {'w': {'x': [('y', 'z')]}}}
当我尝试这种理解时,当然,一些价值被覆盖了:
>>> {t[0]: {t[1]: {t[2]: [(t[3],t[4])]}} for t in ts}
{'a': {'foo': {'bar': [(1, 2)]}}, 'f': {'g': {'baz': [(3, 4)]}}, 'v': {'w': {'x': [('y', 'z')]}}}
你真的不想看到这个结果:
>>> {t[0]: {t[1]: {t[2]: [(t[3],t[4])] for t in ts} for t in ts} for t in ts}
如何正确编写此词典理解?
编辑:对不起,我忘了提及 - 我需要在一天结束时成为常规字典(它最终会通过PyObjC转换为NSDictionary
)。
答案 0 :(得分:5)
我根据需要设置字典自动构建自己的嵌套结构:
from collections import defaultdict
dct = defaultdict(lambda: defaultdict(lambda: defaultdict(list)))
然后将2元组添加到右侧列表中:
for a, b, c, d, e in ts:
dct[a][b][c].append((d, e))
如果不同级别的索引具有含义,我会使用比a
,b
,c
更好的名称。