python - 如何为 size() 列分配名称？

Question

我在 groupby 结果上使用 .size() 以计算每个组中有多少项目。

我希望在不手动编辑列名数组的情况下将结果保存到新的列名中，怎么做？

这是我尝试过的:

grpd = df.groupby(['A','B'])
grpd['size'] = grpd.size()
grpd

我得到的错误:

TypeError: 'DataFrameGroupBy' object does not support item assignment(on the second line)

Answer 1

DataFrameGroupBy 对象的 .size() 内置方法实际上返回一个具有组大小的 Series 对象，而不是 DataFrame。如果您想要一个 DataFrame，其列是组大小，由组索引，具有自定义名称，您可以使用 .to_frame() 方法并将所需的列名称用作其参数。

grpd = df.groupby(['A','B']).size().to_frame('size')

如果您希望组再次成为列，您可以在末尾添加 .reset_index()。