使用映射字典更改字典的键

Question

我想通过传递映射字典和替换嵌套键的函数来替换字典的键名称。问题是我在嵌套字典中有多个名为“id”的键，我想用特定名称重命名这些“id”。

初始词典：

initial_dict = {'id': 1, 'netAmount': 10.2, 'modifiedOn': '2017-01-01',
                    'statusId': 3, 'approvalStateId': 3, 'approvalState': {'id': 3,'name':'Approved'}}

映射字典：

 mapping_dict = {'id': 'pr_id', 'netAmount': 'net_amount', 'modifiedOn': 'modified_date',
                'statusId': 'status_id', 'approvalStateId': 'approval_id','approvalState':{'id':'approv_id'}}

字典所需的输出：

    output_dict = {'pr_id': 1, 'net_amount': 10.2, 'modified_date': '2017-01-01',
                    'status_id': 3, 'approval_id': 3, 'approvalState': {'approv_id': 3, 'name': 'Approved'}}

我所做的是这个，但它只替换字典第一级的键，如果我尝试在映射字典中设置嵌套键，则会收到错误。

def map_fields(obj):
    new_obj = {}
    mapping_dict = {'id': 'pr_id', 'netAmount': 'net_amount', 'modifiedOn': 'modified_date',
                    'statusId': 'status_id', 'approvalStateId': 'approval_id','approvalState':{'id':'approv_id'}}
    for key in obj.keys():
        if key in mapping_dict:
            new_key = mapping_dict[key]
        else:
            new_key = key
        new_obj[new_key] = obj[key]
    return new_obj

你知道该怎么做吗？

谢谢

Answer 1

您需要一个递归函数才能逐步遍历嵌套字典。关键是，递归时，需要同时传递子字典和子映射字典。

请注意，映射字典的结构是针对此问题的，并且不允许您更改嵌套字典的键 - 您需要重新构建存储映射的方式才能实现这一点。

以下应该执行您想要的操作（添加打印语句以帮助遵循运行时的逻辑）：

def map_fields(init_dict, map_dict, res_dict=None):

    res_dict = res_dict or {}
    for k, v in init_dict.items():
        print("Key: ", k)
        if isinstance(v, dict):
            print("value is a dict - recursing")
            v = map_fields(v, map_dict[k])
        elif k in map_dict.keys():
            print("Remapping:", k, str(map_dict[k]))
            k = str(map_dict[k])
        res_dict[k] = v
    return res_dict

print(map_fields(initial_dict, mapping_dict))