将多列上的groupby的结果转换为字典列表
我正在尝试将结果从 Pandas groupby 函数转换为一个字典数组，然后可以在前端的 recharts 中使用它。
代码运行：
data.groupby(['date','board_type'])['count'].sum()
groupby 的输出：
date board_type
2021-01-02 B1 11
B2 105
2021-01-03 B1 30
B2 73
2021-01-04 B1 188
转换后的预期输出：
[
{
date: 2021-01-02, B1: 11, B2: 105
},
{
date: 2021-01-03, B1: 30, B2: 73
},
{
date: 2021-01-04, B1: 188, B2: 0
}
]
我尝试使用 .to_dict 但这似乎不起作用。请建议如何实现这一点。
回答
你很近。关键是到GROUPBY回数据框用的结果转换date为指数B1和B2为列（这是一个系列）。这可以使用unstack. 然后你可以通过to_dict这种方式使用 DataFrame 得到你想要的东西：
data.groupby(['date', 'board_type'])['count'].sum().unstack(fill_value=0).reset_index().to_dict(orient='records')
输出：
[{'date': '2021-01-02', 'B1': 11, 'B2': 105},
{'date': '2021-01-03', 'B1': 30, 'B2': 73},
{'date': '2021-01-04', 'B1': 188, 'B2': 0}]