*非常*嵌套的元组理解

时间:2013-07-19 21:30:02

标签: python python-2.7 dictionary-comprehension

我有这样的数据(是的,这些元组保证正好有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 )。

1 个答案:

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

如果不同级别的索引具有含义,我会使用比abc更好的名称。