python中基于组的数据的序列计数器

Question

我想在A列中对重复出现的序列进行分组。A列可以有任何数字，但不限于0,1,2。因此，基本上，需要为两个0之间的所有行分配一个B值，该值随每个此类组而增加。这是示例数据。 A和B是df上的字段，还有其他字段。我认为for循环应该可以做到。.但是没有成功。

A   B
0   1
1   1
2   1
0   2
1   2
2   2
0   3
1   3
2   3
1   3
2   3
0   4
1   4
2   4
2   4