python - 如何在 Pandas 中对数据框进行分组并保留列

Question

给定一个记录使用此类书籍的数据框:

Name   Type   ID
Book1  ebook  1
Book2  paper  2
Book3  paper  3
Book1  ebook  1
Book2  paper  2

我需要计算所有书籍的数量，保留其他列并得到这个:

Name   Type   ID    Count
Book1  ebook  1     2
Book2  paper  2     2
Book3  paper  3     1

如何做到这一点？

谢谢!

Answer 1

您想要以下内容:

In [20]:
df.groupby(['Name','Type','ID']).count().reset_index()

Out[20]:
    Name   Type  ID  Count
0  Book1  ebook   1      2
1  Book2  paper   2      2
2  Book3  paper   3      1

在您的情况下，“名称”、“类型”和“ID”列的值匹配，因此我们可以 groupby关于这些，请调用 count然后 reset_index .

另一种方法是使用 transform 添加“Count”列，然后调用 drop_duplicates:

In [25]:
df['Count'] = df.groupby(['Name'])['ID'].transform('count')
df.drop_duplicates()

Out[25]:
    Name   Type  ID  Count
0  Book1  ebook   1      2
1  Book2  paper   2      2
2  Book3  paper   3      1