如何从JSON文件提取数据并绘制多条训练曲线？

1. 准备工具安装 #

打开电脑的命令提示符（Windows）或终端（Mac/Linux），输入以下命令安装必备工具库：

pip install pandas matplotlib

2. 编写可视化脚本 #

新建一个名为plot_training_curves.py的文件，复制以下代码进去，替换其中的JSON文件路径为你自己的文件路径：

import pandas as pd
import matplotlib.pyplot as plt

# 读取JSON数据，把这里的路径改成你的JSON文件实际路径
data = pd.read_json("你的训练数据.json")

# 创建画布，设置大小
plt.figure(figsize=(10, 6))

# 绘制四条曲线，若你的JSON键名和示例不同，替换成实际键名即可
plt.plot(data['x'], data['train_loss'], label='训练损失', color='#ff4444', linestyle='-')
plt.plot(data['x'], data['val_loss'], label='验证损失', color='#ffaa44', linestyle='--')
plt.plot(data['x'], data['train_acc'], label='训练准确率', color='#4488ff', linestyle='-')
plt.plot(data['x'], data['val_acc'], label='验证准确率', color='#44dd44', linestyle='--')

# 设置图表基础样式
plt.xlabel('训练步数')
plt.ylabel('数值')
plt.title('训练过程曲线')
plt.legend()  # 显示曲线图例
plt.grid(alpha=0.3)  # 显示网格，增强可读性

# 保存图表到本地，同时弹出窗口显示
plt.savefig('训练曲线结果.png')
plt.show()

3. 适配你的JSON数据结构 #

如果你的JSON是[{"x":1, "train_loss":0.8,...}, {"x":2,...}]这种列表嵌套字典的格式，把读取数据的代码改成：

data = pd.read_json("你的训练数据.json", orient='records')

如果JSON里的键名和示例不一致（比如训练损失叫train_loss_val），直接把代码中对应data['train_loss']改成data['train_loss_val']即可。

4. 运行脚本生成图表 #

回到命令提示符/终端，进入脚本所在的文件夹，输入：

python plot_training_curves.py

运行后会在同一文件夹生成名为训练曲线结果.png的图片，同时弹出窗口显示绘制好的多曲线图表。

内容的提问来源于stack exchange，提问作者Peter Mastnak