使用匹配键递归替换字典值

时间:2017-07-26 19:05:32

标签: python dictionary recursion

我尝试使用字典并查找与key匹配的所有密钥,并将其值替换为replace_value。理论上,词典可以无限深入,因此必须以递归方式完成。

我当前的解决方案正确地替换了值,但引发了一个异常,说'#34;调用Python对象时超出了最大递归深度" (更不用说在没有返回值的情况下使用递归很少的事实)。

def replace_item(obj, key, replace_value):
    """
    Replaces the dictionary value of key with replace_value in the obj dictionary.
    """
    if key in obj:
        obj[key] = replace_value

    for k, v in obj.items():
        if isinstance(v, dict):
            item = replace_item(v, key, replace_value)

            if item is not None:
                item = replace_value

     return obj

它将执行的操作示例如下:

原创词典

person_dict = {
    "name": "Alex",
    "sex": "M",
    "title": "Engineer",
    "misc": {
        "mailbox": "3A",
        "work_type": "remote"
    }
}

然后我打电话给replace_item(person_dict, "work_type", "office"),我最好转而更改更新的字典(person_dict = replace_item(person_dict, "work_type", "office"))。

替换值词典

person_dict = {
    "name": "Alex",
    "sex": "M",
    "title": "Engineer"
    "misc": {
        "mailbox": "3A",
        "work_type": "office"
    }
}

如何修复递归?提前谢谢。

3 个答案:

答案 0 :(得分:2)

你有一些奇怪的行为,你期待return,但你没有。{0}。此外,您的描述暗示它应该替换嵌套键,但是当您在顶级字典中没有键但在较低级别的字典时,您会错过您的代码。我相信下面的代码完成了你所描述的内容:

def replace_item(obj, key, replace_value):
    for k, v in obj.items():
        if isinstance(v, dict):
            obj[k] = replace_item(v, key, replace_value)
    if key in obj:
        obj[key] = replace_value
    return obj

编辑:正如@dashiell建议的那样,在递归搜索/替换之后移动顶级重新分配可以避免key中存在replace_value的无限递归陷阱。 / p>

答案 1 :(得分:2)

这是一个功能风格的作品:

def replace(obj, key, val):
    return {k: replace(val if k == key else v, key, val) 
        for k,v in obj.items()} if isinstance(obj, dict) else obj

它在Python中效率不高(因为重新创建了所有值/子区域),但演示了如何在没有副作用的情况下解决问题,并且不会改变对象。

答案 2 :(得分:1)

def replace_item(obj, key, replace_value):
    """
    Replaces the dictionary value of key with replace_value in the obj dictionary.
    """
    if key in obj:
        obj[key] = replace_value

    for k, v in obj.items():
        if isinstance(v, dict):
            replace_item(v, key, replace_value)

我认为这已经足够了......不需要额外的变量, 在大多数系统中,默认递归深度大约为1000,您可以更改它https://docs.python.org/3/library/sys.html#sys.setrecursionlimit