在嵌套字典中移动字典

Question

我有以下嵌套字典和列表列表。我想创建一个名为 'food' 的新父 (L0) 类别，并将 fruit 和 vegs 字典中的所有值向下移动（以便 'L0': 'fruit' 变为 'L1': 'fruit'、'L1': 'banana'变成'L2': 'banana'等）。

D = [{
        "L0": "fruit",
        "L1_list": [
            {
                "L1": "banana"
            },
            {
                "L1": "apple", 
                "L2_list": [
                    {
                        "L2": "Green apple"
                    }, 
                    {
                        "L2": "Red apple"
                    }
                ]
            }, 
            {
                "L1": "kiwi"
            }
        ]
    },
    {
        "L0": "vegs", 
        "L1_list": [
            {
                "L1": "potato"
            }, 
            {
                "L1": "carrot"
            }
        ]
    }]

例外的输出应该是这样的：

Expected_output = [
    {
        "L0": "food",
        "L1_list": [
            {
                "L1": "fruit",
                "L2_list": [
                    {
                        "L2": "banana"
                    },
                    {
                        "L2": "apple",
                        "L3_list": [
                            {
                                "L3": "Green apple"
                            },
                            {
                                "L3": "Redapple"
                            }
                        ]
                    },
                    {
                        "L2": "kiwi"
                    }
                ]
            },
            {
                "L1": "vegs",
                "L2_list": [
                    {
                        "L2": "potato"
                    },
                    {
                        "L2": "carrot"
                    }
                ]
            }
        ]
    }
]

现在，由于我的字典的大小和深度不同，我需要一个程序化解决方案。所以我想我会创建一个递归函数来迭代 util 它到达树的末尾。当函数到达特定分支的末尾时，它会将 1 添加到键（L0 --> L1、L1_list --> L2_list）。尽管该过程确实将所有内容都降低了一级，但我不知道如何重建初始结构。特别是，我不能让孩子们回到他们各自的名单中。

Final_list = []
def digger(list_to_dig):
    import re
    for x in list_to_dig:
        for k,v in x.items():
            if isinstance(v, list):
                print("keep digging")
                digger(v)
            elif isinstance(x, dict):
                new_D = {}
                new_k = "L" + str(int(re.sub("L", "", k)) + 1)
                new_D[new_k] = v
                temp = re.sub("L", "", k)
                new_child_list = "L" + str(int(re.sub("_list", "", temp)) + 2) + "_list"
                new_D[new_child_list] = ""
                Final_list.append(new_D)
            else:
                print("no dictionary avail")
                pass
    print("_________")
    print(Final_list)
    print("_________")

    test = digger(D)

关于我应该如何解决这个问题有什么建议吗？非常感谢

按照@running.t 的建议，我尝试使用dict.pop 方法。但是，因为它发生在迭代中，它会弹出旧密钥，创建并插入新密钥，但在下一次迭代中，将采用刚刚创建的新密钥，弹出它，然后创建并插入一个新的新密钥，等等on（虽然它也没有进入无限循环）。

这里有一个简化的例子来说明问题：

第 1 步创建新的顶级字典

new_top_level = {"L0": "Food"}
new_dict = {}
for k, v in new_top_level.items():
    lst_k = "L" + str(int(re.sub("L", "", ka)) + 1) + "_list"
    new_dict[k] = v
    new_dict[lst_k] = []

第 2 步将旧树添加到新列表中

old_d = {'L0': 'Fruit', 'L1_list': [{'L1': 'Green apple'}, {'L1': 'Red apple'}]}
new_dict[lst_k].append(old_d)

第 3 步将老树的所有键加 1

def digger(list_to_update):
    import re
    pattern1 = r"L.$"
    pattern2 = r"L._list"
    for x in list_to_update:
        for k1, v1 in x.items():
            if re.match(pattern1, k1):
                new_k1 = "L" + str(int(re.sub("L", "", k1)) + 1)
                x[new_k1] = x.pop(k1)
            elif re.match(pattern2, k1):
                temp = re.sub("L", "", k1)
                new_k1 = "L" + str(int(re.sub("_list", "", temp)) + 1) + "_list"
                x[new_k1] = x.pop(k1)
                digger(v1)

test = digger(new_dict[lst_k])

Answer 1

晚了一年，我知道，但让我们快速分析一下散文中的问题。你有一本字典。字典可以有两种类型的键：L* 和 L*_list。在这两种情况下，* 都是整数。 L* 将始终有一个字符串值。 L*_list 将始终具有一个字典列表值。您的目标是递归递增键名中的整数。

显然，这样的东西很适合递归。您递归到 L*_list 值的每个元素。当您获得没有 L*_list 键的字典列表时，递归结束。在这种情况下，您只需增加 L* 键并返回。到目前为止，我们完全同意，因为我所说的一切都已成为问题。

要回答实际问题，我们只需要进行一项更改：递归函数需要就地修改嵌套对象，或者返回一个新的替换对象。构造一个全新的数据结构比修改现有的字典更简单，因为它使迭代更容易（您也注意到了）。

顶层有一些特殊情况，因为您想将所有内容都推入一个新的food 类别。这不是问题，因为递归解决方案将返回新 L1_list 键的值。

这是一个简单的示例实现：

def increment_keys(d):
    def process_key(key, value):
        key = f'L{int(key[1:]) + 1}'
        return key, value

    def process_list(key, value):
        key = f'L{int(key[1:-5]) + 1}_list'
        value = [increment_keys(d) for d in value]
        return key, value

    def process(key, value):
        if key.endswith('_list'):
            return process_list(key, value)
        return process_key(key, value)

    return dict(process(key, value) for key, value in d.items())

expected_output = [{'L0': 'food', **increment_keys({'L0_list': D})}]

您可以使用三元运算符将嵌套的process 函数吸收到生成器中，该生成器将返回increment_keys 的值。我认为它不会提高可读性，但它会为您节省大约四行代码：

return dict(process_list(k, v) if k.endswith('_list') else process_key(k, v) for k, v in d)

现在，如果您绝对必须就地执行此操作，最好的方法是在迭代之前冻结每个字典的键。如果您遍历冻结的键，pop 和 __setitem__ 不会给您带来任何问题。

由于您永远不会在给定级别的原始键和递增键之间出现重复，因此您不必特别注意丢失先前值（例如，如果您在同一个字典中有 L1 和 L2 ，并首先递增L1。

这是一个就地递归示例：

def increment_keys(obj):
    def process(d):
        for key in list(d.keys()):
            value = d.pop(key)
            if key.endswith('_list'):
                key = f'L{int(key[1:-5]) + 1}_list'
                increment_keys(value)
            else:
                key = f'L{int(key[1:]) + 1}'
            d[key] = value

    for d in obj:
        process(d)

increment_keys(D)
expected_output = [{'L0': 'food', 'L1_list': D}]

根据 Python 约定，我没有从就地函数返回任何内容。

Answer 2

您不应该创建新列表并将所有内容都放入其中。实际上这就是你正在做的事情：

Final_list.append(new_D)

您应该做的是递归迭代您拥有的所有字典和列表（与您当前执行的方式相同），并且如果 object 是一个字典，则相应地重命名该字典中的所有键。

Here you can find how to rename keys i dict。我认为best answer there 建议使用以下内容：

new_k = "L"+str(int(re.sub("L","",k))+1) 
x[new_key] = x.pop(k)

最后，在完成所有D 的挖掘之后，您应该将修改后的D 放入新的Expected_output 列表中。