将pandas数据帧转换为自定义嵌套json

import pandas as pd
import json

data = {'Name': ['John', 'Emma', 'Mark'],
        'Age': [28, 32, 30],
        'Country': ['USA', 'UK', 'Canada']}
df = pd.DataFrame(data)

# 转换为字典
df_dict = df.to_dict()

# 自定义嵌套JSON结构
nested_json = {
    "data": {
        "records": []
    }
}

# 生成嵌套JSON
for i in range(len(df)):
    record = {
        "name": df_dict['Name'][i],
        "age": df_dict['Age'][i],
        "country": df_dict['Country'][i]
    }
    nested_json["data"]["records"].append(record)

# 将嵌套JSON转换为字符串
json_str = json.dumps(nested_json)

print(json_str)

# 保存为文件
with open('nested_json.json', 'w') as f:
    f.write(json_str)