Pandas 和 JSON 格式

Question

我想導出 pandas dataframe 到 JSON 但具有特定結構。 這是我希望在 output JSON 中擁有的結構：

{
    "Var_1":"A1",
    "Var_2": [
        {"Var_3": "Value_3",
        "Var_4":  "Value_4",
        },
        {"Var_3": "Value_5",
        "Var_4":  "Value_6",
        }
    ]
    "Var_1":"A2",
    "Var_2": [
        {"Var_3": "Value_7",
        "Var_4":  "Value_8",
    ]
}

這是我的 dataframe: pd.DataFrame({"Var_1": ['A1', 'A1', 'A2'], "Var_3": ["Value_3", "Value_5","Value7"], "Var_4": ["Value_4", "Value_6", "Value_8"]})

Var_2 可以用 Var_3 和 Var_4 計算。

我嘗試使用https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.to_json.html進行記錄、拆分等……但無法獲得正確的結構。

Answer 1

IIUC，嘗試：

output = (df.groupby("Var_1")
            .apply(lambda x: x.drop("Var_1",axis=1).to_dict("records"))
            .rename("Var_2")
            .reset_index()
            .to_json(orient="records")
          )

>>> output

[
  {
    "Var_1": "A1",
    "Var_2": [
      {
        "Var_3": "Value_3",
        "Var_4": "Value_4"
      },
      {
        "Var_3": "Value_5",
        "Var_4": "Value_6"
      }
    ]
  },
  {
    "Var_1": "A2",
    "Var_2": [
      {
        "Var_3": "Value7",
        "Var_4": "Value_8"
      }
    ]
  }
]