Python3.x Python 3 给定未知数量的类别动态地将字典嵌套到列表中

给出你的主要类别作为一个列表:

categories = ['name', 'type']

您可以首先将输入数据转换为列表的嵌套字典，以便通过键更容易、更高效地访问子对象，而不是所需的输出格式-字典的嵌套列表:

tree = {}
for record in tsv_data:
    node = tree
    for category in categories[:-1]:
        node = node.setdefault(record.pop(category), {})
    node.setdefault(record.pop(categories[-1]), []).append(record)

tree将变成:

{'bob': {'blue': [{'id': 1, 'size': 2}, {'id': 2, 'size': 3}, {'id': 3, 'size': 4}], 'red': [{'id': 4, 'size': 2}]}, 'sarah': {'blue': [{'id': 5, 'size': 2}, {'id': 6, 'size': 3}], 'green': [{'id': 7, 'size': 2}]}, 'jack': {'blue': [{'id': 8, 'size': 5}]}}

然后，您可以使用递归函数将嵌套的字典转换为所需的输出 struct :

def transform(node):
    if isinstance(node, dict):
        return [
            {'name': name, 'children': transform(child)}
            for name, child in node.items()
        ]
    return node

这样transform(tree)人就会回来:

[{'name': 'bob', 'children': [{'name': 'blue', 'children': [{'id': 1, 'size': 2}, {'id': 2, 'size': 3}, {'id': 3, 'size': 4}]}, {'name': 'red', 'children': [{'id': 4, 'size': 2}]}]}, {'name': 'sarah', 'children': [{'name': 'blue', 'children': [{'id': 5, 'size': 2}, {'id': 6, 'size': 3}]}, {'name': 'green', 'children': [{'id': 7, 'size': 2}]}]}, {'name': 'jack', 'children': [{'name': 'blue', 'children': [{'id': 8, 'size': 5}]}]}]

演示:https://replit.com/@blhsing/NotableCourageousTranslations