在日常的数据处理工作中,经常需要对 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('完成转换。')