如何以特定格式将pandas数据框导出到json

Question

我的数据框是

  'col1' , 'col2'
    A    ,   89
    A    ,   232
    C    ,   545
    D    ,   998

并希望按以下方式导出：

{
  'A' : [ 89, 232 ],
  'C' : [545],
  'D' : [998]   
}

但是，所有to_json都不适合这种格式（orient ='records'，...）。 有没有办法像这样输出？

Answer 1

使用groupby转换为list，然后转换为to_json：

json = df.groupby('col1')['col2'].apply(list).to_json()
print (json)
{"A":[89,232],"C":[545],"D":[998]}

<强>详细：

print (df.groupby('col1')['col2'].apply(list))
col1
A    [89, 232]
C        [545]
D        [998]
Name: col2, dtype: object