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

Question

使用 Python，我有一个 panda 数据框

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 方法。

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