如何从DataFrame生成指定结构的JSON文件（含版本信息）

问题描述 #

现有如下DataFrame：

name        age
Paul        12
Andres      13
Caroline    14
Justin      15

需要生成结构如下的JSON文件：

{"NAME":["Paul","Andres","Caroline","Justin"], "NAME-AGE":{"Paul":12,"Andres":13,"Caroline":14,"Justin":15},"VERSION":1}

已知df.to_json()无法直接生成该精确结构，求最佳实现方法。

解决方案 #

直接手动构造目标格式的字典，再用json库转成JSON即可，步骤简单直接，完全贴合需求：

• 先导入需要的库：

import pandas as pd
import json

• 基于DataFrame构造目标字典：

# 假设你的DataFrame变量是df
target_dict = {
    "NAME": df["name"].tolist(),  # 提取name列转为列表
    "NAME-AGE": df.set_index("name")["age"].to_dict(),  # 以name为键、age为值生成字典
    "VERSION": 1  # 直接添加版本字段
}

• 转成JSON字符串或写入文件：

# 生成JSON字符串
json_result = json.dumps(target_dict)
# 如果要写入文件，执行以下代码
with open("result.json", "w") as f:
    json.dump(target_dict, f)

这种方式比硬套df.to_json()灵活得多，额外字段可以直接往字典里添加，完全能得到你需要的精确结构。

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