时间戳

首页 > 解决方案 > 将多列上的 groupby 结果转换为字典列表

python - 将多列上的 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 但这似乎不起作用。请建议如何实现这一点。

标签： pythonpandas

解决方案

你很亲密。关键是将 groupby 的结果转换回具有date索引和B1列的 DataFrame B2（它是一个系列）。这可以使用unstack. 然后你可以通过这种方式使用 DataFrame 得到你想要的to_dict：

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}]

推荐阅读