将字典列表转换为嵌套字典列表

Question

我尝试并看到了多个针对上述问题的解决方案，但没有找到适合我情况的解决方案。 我有以下字典列表。

input_list = [
                {'k0':'v0','level':0,'row':0},
                {'k1':'v1','level':1,'row':1},
                {'k2':'v2','level':2,'row':2},
                {'k3':'v3','level':2,'row':3},
                {'k4':'v4','level':1,'row':4},
                {'k5':'v5','level':2,'row':5},
                {'k6':'v6','level':1,'row':6},
                {'k7':'v7','level':0,'row':7},
                {'k8':'v8','level':1,'row':8},
                {'k9':'v9','level':2,'row':9},
            ]

我需要将此列表按以下方式转换为嵌套字典的列表

[
  {'k0':'v0',
   'level':0,
   'child':[
          {'k1':'v1','level':1,'child':[{'k2':'v2','level':2}, 
                                        {'k3':'v3','level':2}]},
          {'k4':'v4','level':1,'child':[{'k5':'v5','level':2}]},
          {'k6':'v6','level':1}
         ]},

  {'k7':'v7',
   'level':0,
   'child':[{'k8':'v8','level':1,'child':[{'k9':'v9','level':2}]}]}
]

我尝试的解决方案如下：

levels = dict()
for n in input_list:
    levels.setdefault(n['level'], []).append(n)

这使我得到了按级别分离的字典，但是我不能超越这段代码以实现我想要的解决方案。如有帮助，将不胜感激。谢谢！

- supernova

---

请在您的问题中更新您尝试过的代码。 - quamrana

找不到解决方案 - 你尝试制作解决方案了吗？ - jonrsharpe

我建议您首先用文字写出决定如何处理输入列表中每个字典的规则。 - Code-Apprentice

好奇心驱使着：是什么原因让你需要那种超复杂的嵌套字典结构呢？ - Maurice Meyer

你需要在字符串周围加上引号。 - Kenny Ostrom

Answer 1

您可以使用递归和itertools.groupby一起使用：

from itertools import groupby as gb
data = [{'k0': 'v0', 'level': 0, 'row': 0}, {'k1': 'v1', 'level': 1, 'row': 1}, {'k2': 'v2', 'level': 2, 'row': 2}, {'k3': 'v3', 'level': 2, 'row': 3}, {'k4': 'v4', 'level': 1, 'row': 4}, {'k5': 'v5', 'level': 2, 'row': 5}, {'k6': 'v6', 'level': 1, 'row': 6}, {'k7': 'v7', 'level': 0, 'row': 7}, {'k8': 'v8', 'level': 1, 'row': 8}, {'k9': 'v9', 'level': 2, 'row': 9}]
_d = [{a:b for a, b in i.items() if a not in {'row'}} for i in data]

def get_results(d, l = 0):
  r, p = [(a, list(b)) for a, b in gb(d, key=lambda x:x['level'] == l)], []
  for a, b in r:
     if a:
        p.extend(b)
     else:      
        p[-1]['child'] = get_results(b, l+1)
  return p

---

import json
print(json.dumps(get_results(_d), indent=4))

输出：

[
  {
    "k0": "v0",
    "level": 0,
    "child": [
        {
            "k1": "v1",
            "level": 1,
            "child": [
                {
                    "k2": "v2",
                    "level": 2
                },
                {
                    "k3": "v3",
                    "level": 2
                }
            ]
        },
        {
            "k4": "v4",
            "level": 1,
            "child": [
                {
                    "k5": "v5",
                    "level": 2
                }
            ]
        },
        {
            "k6": "v6",
            "level": 1
        }
    ]
 },
 {
    "k7": "v7",
    "level": 0,
    "child": [
        {
            "k8": "v8",
            "level": 1,
            "child": [
                {
                    "k9": "v9",
                    "level": 2
                }
            ]
         }
      ]
    }
]

- Ajax1234

---

谢谢回答，尽管它未能捕获所有二级值。 - supernova

@KetanPujare 是的，原始代码将多个子元素压缩成一个字典，以防有一组子元素具有自己的子级。但是，请查看我的最近编辑，因为我添加了一个解决方法，可以匹配您想要的输出。 - Ajax1234

Answer 2

# This stores last dicts seen by their levels
dict_by_levels = [None] * len(input_list)

# To be returned; begins with the first "root"
output_list = [input_list[0]]

for prev_dict, cur_dict in zip(input_list, input_list[1:]):
    # Get previous and current level, and store them in levels dict
    prev_level, cur_level = prev_dict["level"], cur_dict["level"]
    dict_by_levels[prev_level] = prev_dict
    dict_by_levels[cur_level] = cur_dict

    # If we are currently in a deeper level, append to previous dict's child
    if cur_level > prev_level:
        prev_dict.setdefault("child", []).append(cur_dict) 

    # If we are currently in a shallower or equal level *and* the current level is not 0,
    # then get the last one-level-before dict seen so far, and append to it
    elif cur_level <= prev_level and cur_level != 0:
        dict_by_levels[cur_level-1].setdefault("child", []).append(cur_dict)

    # This else is equivalent to "if cur_level == 0" i.e. a "root" dict
    else:
        output_list.append(cur_dict)

我为每个循环迭代跟踪两个级别，即前一个字典和当前字典。我还创建了一个名为dict_by_levels的字典列表，按级别存储; 索引0、1、2..将包含该级别上看到的最后一个字典（如果有的话）。这在我们进入更浅或相等的级别时很有帮助（例如从2到2）。

如果当前级别高于先前级别（例如1> 0），则我们只需将其附加到当前字典的子项中。

如果当前级别小于等于先前级别，则意味着我们应该将其附加到上次看到的级别为current level-1的字典子项中（此处使用dict_by_levels有帮助）。当然，如果当前级别是0，则不执行此操作-我们有一个新的根。

而且，这种情况在else子句中通过将其附加到要返回的输出列表来处理。

- Mustafa Aydın