给出一个数据框,记录一些这样的书的使用：
- 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
如何才能做到这一点？
谢谢！
解决方法
你想要以下内容：
- 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