具有两个分类变量的数据帧上的 Groupby 和 count() [重复]

Question

我有一个数据集，我需要 groupby() 并找到每个唯一组合的计数。

    body-style        make
0  convertible  alfa-romeo
1  convertible  alfa-romeo
2    hatchback  alfa-romeo
3        sedan        audi
4        sedan        audi

我的需要是产生如下所示的输出：

             make   body-style   count
0     alfa-romero  convertible       2
1     alfa-romero    hatchback       1
2            audi    sedan           2

尝试了以下代码：

body = pd.DataFrame({'make':['alfa-romeo','alfa-romeo','alfa-romeo','audi','audi'], 'body-style':['convertible','convertible','hatchback','sedan','sedan']})

body.groupby(by=['make','body-style'], as_index=False).count()

此聚合引发“列表索引超出范围”错误。但是，当我从 groupby 子句中删除任一列时，它可以给我按剩余列分组的计数。

如果我删除 as_index=False，则不会出现错误，但生成的对象将包含两列 - make 和 body-style作为索引的一部分，不会有任何计数数据。

我可以在 datframe 中添加另一列，用 1 填充它，然后在 groupby 上使用 sum() 而不是 count()。但想知道是否有更清洁的方法来做到这一点。

Answer 1

只有当您至少有一列未用于分组时，您才能使用GroupBy.count()。

如果您按 DF 中的所有列分组 - 请改用 .size()：

In [119]: body.groupby(['make','body-style']).size().reset_index(name='count')
Out[119]:
         make   body-style  count
0  alfa-romeo  convertible      2
1  alfa-romeo    hatchback      1
2        audi        sedan      2