如何将Pandas DataFrame转为Django指定格式的JSON初始化数据

我来帮你解决这个问题——把Pandas DataFrame转换成Django fixture需要的特定JSON格式其实很直接，关键是要把DataFrame的索引映射为pk，列对应fields里的字段，同时避免多余的结构。

先明确目标格式 #

你需要的是这种结构的JSON数组，每个元素对应一条模型数据：

[
{ "model": "myapp.person", "pk": 1, "fields": { "first_name": "John", "last_name": "Lennon" } },
{ "model": "myapp.person", "pk": 2, "fields": { "first_name": "Paul", "last_name": "McCartney" } }
]

实现步骤 #

下面是具体的代码实现，一步步来：

1. 准备你的DataFrame
   确保DataFrame的索引就是你要作为pk的值，列名和Django模型的字段名完全匹配：

import pandas as pd

# 示例数据，索引对应pk，列对应模型字段
data = {
    "first_name": ["John", "Paul"],
    "last_name": ["Lennon", "McCartney"]
}
df = pd.DataFrame(data, index=[1, 2])

2. 编写转换函数
   遍历DataFrame的每一行，生成符合Django要求的字典结构：

def df_to_django_fixture(df, app_name, model_name):
    fixture = []
    # 遍历索引（pk）和行数据
    for pk, row in df.iterrows():
        fixture.append({
            "model": f"{app_name}.{model_name}",
            "pk": pk,
            "fields": row.to_dict()  # 直接把行转为fields字典
        })
    return fixture

3. 生成并导出JSON文件
   调用函数得到fixture数据，然后导出为JSON文件（放在Django app的fixtures目录下即可）：

# 生成fixture数据
fixture_data = df_to_django_fixture(df, "myapp", "person")

# 导出为JSON，indent=4让格式更易读，ensure_ascii=False支持中文
import json
with open("myapp/fixtures/persons.json", "w", encoding="utf-8") as f:
    json.dump(fixture_data, f, ensure_ascii=False, indent=4)

为什么这个方法能解决你的问题？ #

• 直接用iterrows()获取索引（作为pk）和行数据，不会生成多余的索引键或额外方括号
• row.to_dict()自动把列名和值转为fields的键值对，完全匹配你需要的格式
• 最终生成的是标准的JSON数组，符合Django fixture的要求

注意事项 #

• 确保DataFrame的索引类型和Django模型的pk类型一致（通常是整数，字符串类型也可以）
• 如果有日期时间类型的字段，要确保格式是ISO 8601（比如2024-05-20T12:30:00），pandas默认会处理，但如果有特殊格式可以手动转换
• 列名必须和Django模型的字段名完全一致（包括大小写）

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