python数据结构：字典列表到一个字典

Question

我有一个数据结构。 它看起来如下：

data = [[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2A', 'name': 'child', 'steps': 1},
 {'id': '3A', 'name': 'grandChild-A', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2A', 'name': 'child', 'steps': 1},
 {'id': '3B', 'name': 'grandChild-B', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3A', 'name': 'grandChild-C', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3B', 'name': 'grandChild-D', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3C', 'name': 'grandChild-E', 'steps': 2},
 {'id': '4A', 'name': 'final', 'steps': 3}
 ],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2', 'name': 'child', 'steps': 1},
 ]
]

我的预期输出如何

预期产出

output = {
    "1" : {
        "2A": {
            "3A": "grandChild-A",
            "3B": "grandChild-B"

        },
        "2B": {
            "3A": "grandChild-C",
            "3B": "grandChild-D",
            "3C": {
                "4A": "final"
            }

        },
        "2":"child"

    }
}

我怎样才能做到这一点？ 我想使用枚举器，但我总是把所有的东西都放在 1 里面。提前致谢

更新：

我尝试了以下代码：

parent = data[0][0]["id"]
dict_new = {}
dict_new[parent] = {}

for e in data:
    for idx, item in enumerate(e):
        display(item)
        if idx>0:
            dict_new[parent][e[idx]["id"]] = e[idx]["name"]

Answer 1

你可以试试：

d = {}
root = d
for L in data:
    d = root
    for M in L[:-1]:
        d = d.setdefault(M["id"], {})
    d[L[-1]["id"]] = L[-1]['name']

这个想法是按照每个列表构建一棵树（因此d.setdefault(M["id"], {}) 。叶子的处理方式不同，因为它必须是 'name' 的值。

from pprint import pprint
pprint(root)

输出：

{'1': {'2': 'child',
       '2A': {'3A': 'grandChild-A', '3B': 'grandChild-B'},
       '2B': {'3A': 'grandChild-C',
              '3B': 'grandChild-D',
              '3C': {'4A': 'final'}}}}

---

上述解决方案不适用于以下输入：

data = [[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1}]],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3A', 'name': 'grandChild-C', 'steps': 2}]]

迭代第二个列表将尝试将新元素3A -> grandChild-C到d['1']['2B'] dict。 但是d['1']['2B']不是字典而是这里的'child'字符串，因为第一个列表。

当我们迭代元素时，我们检查键是否已经被映射，否则创建一个新的字典（这是setdefault作业）。 我们还可以检查键是否映射到str ，如果是这种情况，请用新的字典替换字符串：

...
for M in L[:-1]:
    if M["id"] not in d or isinstance(d[M["id"]], str):
        d[M["id"]] = {}
    d = d[M["id"]]
...

输出：

{'1': {'2B': {'3A': 'grandChild-C'}}}

Answer 2

我修复了您的数据：（缺少逗号）

data = [[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2A', 'name': 'child', 'steps': 1},
 {'id': '3A', 'name': 'grandChild-A', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2A', 'name': 'child', 'steps': 1},
 {'id': '3B', 'name': 'grandChild-B', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3A', 'name': 'grandChild-C', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3B', 'name': 'grandChild-D', 'steps': 2}],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2B', 'name': 'child', 'steps': 1},
 {'id': '3C', 'name': 'grandChild-E', 'steps': 2},
 {'id': '4A', 'name': 'final', 'steps': 3}
 ],

[{'id': '1', 'name': 'parent', 'steps': 0},
 {'id': '2', 'name': 'child', 'steps': 1},
 ]
]

我想出了这个代码：

output = {}
#print(data)

for lis in data:
    o = output
    ln = len(lis) - 1
    for idx,d in enumerate(lis):
        id = d['id']
        if idx == ln:
            o[id] = d['name']
        else:
            if id not in o:
                o[id] = {}
            o = o[id]

print('Result:')
print(output)