将“计数”列添加到 Pandas 中的 groupby 的结果中？

Question

我认为这是一个相当基本的问题，但我似乎无法找到解决方案。

我有一个类似于以下内容的熊猫数据框：

import pandas as pd

df = pd.DataFrame({'A' : ['x','x','y','z','z'],
                   'B' : ['p','p','q','r','r']})
df

它创建了一个这样的表：

    A   B
0   x   p
1   x   p
2   y   q
3   z   r
4   z   r

我正在尝试创建一个表来表示该数据框中不同值的数量。 所以我的目标是这样的：

    A   B   c
0   x   p   2
1   y   q   1
2   z   r   2

不过，我找不到正确的功能来实现这一点。 我试过了：

df.groupby(['A','B']).agg('count')

这将生成一个包含 3 行（如预期）但没有“计数”列的表。 我不知道如何添加该计数列。 有人能指出我正确的方向吗？

Answer 1

您可以使用size

df.groupby(['A','B']).size()
Out[590]: 
A  B
x  p    2
y  q    1
z  r    2
dtype: int64

对于您的解决方案，添加一列

df.groupby(['A','B']).B.agg('count')
Out[591]: 
A  B
x  p    2
y  q    1
z  r    2
Name: B, dtype: int64

更新 ：

df.groupby(['A','B']).B.agg('count').to_frame('c').reset_index()

#df.groupby(['A','B']).size().to_frame('c').reset_index()
Out[593]: 
   A  B  c
0  x  p  2
1  y  q  1
2  z  r  2

Answer 2

熊猫 >= 1.1： DataFrame.value_counts

这是df.groupby(['A', 'B']).size()的相同替代品。

df.value_counts(['A', 'B'])

A  B
z  r    2
x  p    2
y  q    1
dtype: int64

df.value_counts(['A', 'B']).reset_index(name='c')

   A  B  c
0  z  r  2
1  x  p  2
2  y  q  1