python - dask:如何分组，聚合而不丢失用于分组的列

Question

如何在对以下数据进行分组时获得 SQL 样式的分组输出:

   item   frequency
    A      5
    A      9
    B      2
    B      4
    C      6

df.groupby(by = ["item"]).sum()

结果:

  item   frequency
    A      14
    B      6
    C      6

在 pandas 中，它是通过设置 as_index=False 来实现的。但是 dask 没有 support this argument in groupby .它目前省略 item 列并返回具有 frequency 列的系列。

Answer 1

也许之后调用 .reset_index？