如何在python中的json.load期间编辑/重命名键？

Question

我有一个json文件(~3Gb),我需要加载到mongodb.相当多的json键包含一个.(点),导致负载进入mongodb失败.我想加载json文件,并在进程中编辑键名,比如用空格替换点.使用以下python代码

import json

def RemoveDotKey(dataPart):
    for key in dataPart.iterkeys():
        new_key = key.replace(".","")
        if new_key != key:
            newDataPart = deepcopy(dataPart)
            newDataPart[new_key] = newDataPart[key]
            del newDataPart[key]
            return newDataPart
    return dataPart

new_json = json.loads(data, object_hook=RemoveDotKey) 

名为RemoveDotKey的object_hook应遍历所有键,键包含一个点,创建一个副本,用空格替换点,然后返回副本.创建了dataPart的副本,因为我不确定是否可以迭代dataPart的密钥并同时插入/删除键值对.

这里似乎有一个错误,所有带有点的json键都没有被编辑.我不太清楚json.load是如何工作的.也是python的新手(使用它不到一周)

Answer 1

你几乎拥有它:

import json

def remove_dot_key(obj):
    for key in obj.keys():
        new_key = key.replace(".","")
        if new_key != key:
            obj[new_key] = obj[key]
            del obj[key]
    return obj

new_json = json.loads(data, object_hook=remove_dot_key) 

你在循环中返回一个字典,所以你只修改一个键.而且您不需要复制值,只需重命名键即可.