将关系1:N数据的pandas数据帧转换为嵌套JSON

import pandas as pd
import json

# 创建主表数据帧
main_df = pd.DataFrame({'主表键': [1, 2, 3, 4],
                       '主表数据': ['数据1', '数据2', '数据3', '数据4']})

# 创建从表数据帧
sub_df = pd.DataFrame({'主表键': [1, 1, 2, 3, 3],
                      '从表数据': ['数据A', '数据B', '数据C', '数据D', '数据E']})

merged_df = pd.merge(main_df, sub_df, on='主表键', how='left')

result = (merged_df.groupby('主表键')
                  .apply(lambda x: x[['从表数据']].to_dict('records'))
                  .reset_index()
                  .rename(columns={0: '从表数据'})
                  .to_json(orient='records'))