使用 Python修改JSON 文件中对应键值

文章目录

    • 前言
    • 代码分析


前言

  在日常的数据处理工作中,经常需要对 JSON 文件进行读取和修改。在 Python 中,处理 JSON 文件非常方便。本文将通过一个简单的示例程序来演示如何读取和修改 JSON 文件。

代码分析

  首先,需要导入 json 和 os 模块,这两个模块提供了处理 JSON 文件和文件路径的功能。

import json
import os

接下来,指定 JSON 文件的路径和文件名,并定义一个保存修改后文件的路径。

path = 'D:/python auto'  # 修改为包含 via_export_coco.json 文件的目录路径
file_name = 'via_export_coco.json'  # 文件名
path2 = 'via_export_coco_int.json'  # 保存路径
file_path = os.path.join(path, file_name)  # 完整文件路径

然后,使用 os.path.isfile() 函数检查文件是否存在。如果文件存在,将读取文件内容,并进行修改。

num_flag = 0
if os.path.isfile(file_path):  # 检查文件是否存在
    num_flag = 1
    with open(file_path, 'r', encoding='utf-8') as load_f:
        load_dict = json.load(load_f)

    # 修改的代码开始
    for annotation in load_dict['annotations']:
        annotation['image_id'] = int(annotation['image_id'])
    # 修改的代码结束

在上述代码中,通过遍历 JSON 文件中的 ‘annotations’ 键对应的列表,并将其中的 ‘image_id’ 值转换为整数类型。

最后,将修改后的数据保存到新文件中。

    with open(os.path.join(path, path2), 'w', encoding='utf-8') as dump_f:
        json.dump(load_dict, dump_f, indent=4, ensure_ascii=False)

最后,根据 num_flag 的值输出相应的提示信息。

if num_flag == 0:
    print('Sorry!该文件不存在,请重新确认文件路径是否正确。')
else:
    print('完成转换。')

通过以上的步骤,可以轻松地读取和修改 JSON 文件,实现自动化处理。

完整的程序代码如下:

import json
import os

path = 'D:/python auto'  # 修改为包含 via_export_coco.json 文件的目录路径
file_name = 'via_export_coco.json'  # 文件名
path2 = 'via_export_coco_int.json'  # 保存路径
file_path = os.path.join(path, file_name)  # 完整文件路径

num_flag = 0
if os.path.isfile(file_path):  # 检查文件是否存在
    num_flag = 1
    with open(file_path, 'r', encoding='utf-8') as load_f:
        load_dict = json.load(load_f)

    # 修改的代码开始
    for annotation in load_dict['annotations']:
        annotation['image_id'] = int(annotation['image_id'])
    # 修改的代码结束

    with open(os.path.join(path, path2), 'w', encoding='utf-8') as dump_f:
        json.dump(load_dict, dump_f, indent=4, ensure_ascii=False)

if num_flag == 0:
    print('Sorry!该文件不存在,请重新确认文件路径是否正确。')
else:
    print('完成转换。')

你可能感兴趣的:(python小问题,python,json,java)