基于python pandas中的现有计数器添加新计数器

Question

我有一个看起来像这样的系列

   col
0   1
1   2
2   3
3   4
4   1
5   2
6   3
7   1
8   2
9   3
10  1
11  2

并且我想生成另一个像这样的计数器

   col  col2
0   1     1
1   2     1
2   3     1
3   4     1
4   1     2
5   2     2
6   3     2
7   1     3
8   2     3
9   3     3
10  1     4
11  2     4

如何在python中做到这一点？

Answer 1

如果1始终是组的开头，则通过按Series.eq进行比较来创建掩码，然后将Series.eq累加起来：

Series.cumsum