如何将Pandas DataFrame与额外变量组合生成指定格式JSON文件
解决方案
先明确你要的目标JSON格式大概是这样的:
{ "Error": 45, "Data": [ {"a": [1, 0.1]}, {"b": [2, 0.2]}, {"c": [3, 0.3]} ] }
直接用下面的代码就能实现:
import pandas as pd import json # 示例数据 df = pd.DataFrame([[1,2,3], [0.1, 0.2, 0.3]], columns=('a','b','c')) Error = 45 # 构造目标结构 result = { "Error": Error, "Data": [{col: df[col].tolist()} for col in df.columns] } # 输出为JSON字符串(用于查看结果) json_str = json.dumps(result, indent=2) print(json_str) # 如果要保存到JSON文件 with open('output.json', 'w') as f: json.dump(result, f, indent=2)
关键步骤说明
- 用列表推导式遍历DataFrame的每一列,把列名和对应列的列表数据组成单键字典,统一放到
Data数组中 - 直接将
Error变量加入结果字典 - 用
json模块的dumps(转成字符串)或dump(写入文件)完成输出,indent=2是让JSON格式更易读,属于可选参数
内容的提问来源于stack exchange,提问作者Niro Mal
相关产品推荐
相关产品推荐

