Python/Pandas-基于列值向DataFrame添加计数器

使用Python，我有一个熊猫数据框架

A B 
1 red
2 blue
3 green
4 red
5 green
6 orange
7 red

我想根据B中的值创建一个带有递增计数器的列。也就是说，要解析为

A B      C
1 red    1
2 blue   1
3 green  1
4 red    2
5 green  2
6 orange 1
7 red    3

---

按每种颜色分组，然后使用

cumcount

方法

df['C'] = df.groupby('B').cumcount() + 1

   A       B  C
0  1     red  1
1  2    blue  1
2  3   green  1
3  4     red  2
4  5   green  2
5  6  orange  1
6  7     red  3