python - 使 pandas 数据框按计数分组并具有列名

Question

在 python 3.6 和 pandas 0.19.0 中，有一个数据框“df”

id
abc
abc
def

我正在尝试创建一个带有 id_count 列的数据框，这与 SQL 相同:

SELECT id, COUNT(1) AS id_count FROM table group by id

理想返回:

id  id_count
abc 2
def 1

我已经尝试过:

print (df.groupby(['id'])['id'].count())

返回

id
abc 2
def 1

print (df.groupby(['id'])['id'].transform('count'))

返回

0    2
1    1

有什么想法吗？谢谢!

Answer 1

在带有 groupby 的聚合函数中使用这种基于字典的格式。

df.groupby('id', as_index=False)['id'].agg({'id_count':'count'})

    id  id_count
0  abc         2
1  def         1