導出 Pandas DF 到嵌套的 JSON（多嵌套）

Question

我想將 Pandas df 導出到嵌套的 JSON 以便在 Mongodb 中攝取。

以下是數據示例：

data = {
    'product_id': ['a001','a001','a001'],
    'product': ['aluminium','aluminium','aluminium'],
    'production_id': ['b001','b002','b002'],
    'production_name': ['metallurgical','recycle','recycle'],
    'geo_name': ['US','EU','RoW'],
    'value': [100, 200 ,200]
}
df = pd.DataFrame(data=data)

product_id	產品	生產ID	生產名稱	地理名稱	價值
a001	鋁	b001	冶金	我們	100
a001	鋁	b002	回收	歐盟	200
a001	鋁	b002	回收	排	200

這就是最終 JSON 的樣子：

{
    "name_id": "a001",
    "name": "aluminium",
    "activities": [
        {
            "product_id": "b001"
            "product_name": "metallurgical",
            "regions": [
                {
                    "geo_name": "US",
                    "value": 100
                }
            ]
        },
        {
            "product_id": "b002"
            "product_name": "recycle",
            "regions": [
                {
                    "geo_name": "EU",
                    "value": 200
                },
                {
                    "geo_name": "RoW",
                    "value": 200
                }
            ]
        }
    ]
}

有一些問題與我的問題很接近，但它們要么是老版本，要么是舊版本的 Pandas，解決方案會中斷，或者不能完全按照我希望 json 分組和嵌套的方式工作（這例如，單級如何從 pandas DataFrame 創建嵌套的 JSON？ ）。

一些幫助將不勝感激。

Answer 1

我找到了最簡單的解決方案，可以用於無限數量的嵌套（本例中為 2 個）：

json_extract = df\
    .groupby(['product_id','product', 'production_id','production_name'])\
    .apply(lambda x: x[['geo_name','value']].to_dict('records'))\
    .reset_index(name='geos')\
    .groupby(['product_id','product'])\
    .apply(lambda x: x[['production_id','production_name', 'geos']].to_dict('records'))\
    .reset_index(name='production')\
    .to_json(orient='records')

[
    {
        "product_id": "a001",
        "product": "aluminium",
        "production": [
            {
                "production_id": "b001",
                "production_name": "metallurgical",
                "geos": [
                    {
                        "geo_name": "US",
                        "value": 100
                    }
                ]
            },
            {
                "production_id": "b002",
                "production_name": "recycle",
                "geos": [
                    {
                        "geo_name": "EU",
                        "value": 200
                    },
                    {
                        "geo_name": "RoW",
                        "value": 200
                    }
                ]
            }
        ]
    }
]