从一个开始枚举每组中的行

Question

我有一个数据框（按日期排序，为简单起见，示例中不包含日期列），如下所示：

df = pd.DataFrame(['A', 'B' , 'B', 'A', 'C', 'B'], columns=['letters'])

  letters
0       A
1       B
2       B
3       A
4       C
5       B

我想创建一个新列来计算letters 列中每个值的出现次数，并随着letters 列中出现的值增加1。我想要到达的数据框是这样的：

  letters  occurance_counter
0       A                  1
1       B                  1
2       B                  2
3       A                  2
4       C                  1
5       B                  3

任何帮助都会很棒，在此先感谢！

Answer 1

您可以groupby 并使用方法cumcount：

df['occurance_counter'] = df.groupby('letters')['letters'].cumcount().add(1)

结果：

  letters  occurance_counter
0       A                  1
1       B                  1
2       B                  2
3       A                  2
4       C                  1
5       B                  3