Python / Pandas-根据列值向DataFrame添加计数器

Question

使用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

Answer 1

对每种颜色进行cumcount ，然后使用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