python - Pandas 计算唯一行

Question

我有一个 pandas 数据框类似于:

ColA ColB
1    1
1    1
1    1
1    2
1    2
2    1
3    2

我想要一个与 Counter 具有相同功能的输出.我需要知道每行出现了多少次(所有列都相同。

在这种情况下，正确的输出是:

ColA ColB Count
1    1    3
1    2    2
2    1    1
3    2    1

我试过这样的东西:

df.groupby(['ColA','ColB']).ColA.count()

但这给了我一些丑陋的输出，我在格式化时遇到了问题

Answer 1

您可以使用 size与 reset_index :

print df.groupby(['ColA','ColB']).size().reset_index(name='Count')
   ColA  ColB  Count
0     1     1      3
1     1     2      2
2     2     1      1
3     3     2      1