所有組組合的值計數

Question

我有一個包含多個組列和一個值列的數據框。

   a  b  val
0  A  C  1
1  A  D  1
2  A  D  1
3  A  D  2
4  B  E  0

對於任何一組，例如對於a==A ， b==C我可以在序列切片上進行value_counts 。 如何獲取類似於數據框格式的組列的所有可能組合的值計數：

  a  b val counts
0 A  C 1   1
1 A  D 1   2
2 A  D 2   1
2 B  E 0   1

Answer 1

那是你要的嗎？

In [47]: df.groupby(['a','b','val']).size().reset_index()
Out[47]:
   a  b  val  0
0  A  C    1  1
1  A  D    1  2
2  A  D    2  1
3  B  E    0  1

或這個？

In [43]: df['counts'] = df.groupby(['a','b'])['val'].transform('size')

In [44]: df
Out[44]:
   a  b  val  counts
0  A  C    1       1
1  A  D    1       3
2  A  D    1       3
3  A  D    2       3
4  B  E    0       1