将相似的指数值分组，并用新的指数熊猫对每组进行重新索引

Question

我有带有值的df和类似的索引

>>> df
idx  a  b
0  15  15
0  15  15
0  15  15
1  20  20
1  20  20
1  20  20
1  20  20
1  20  20

而我想在下面为每组分配单独的类似索引的结果。我怎样才能在熊猫中得到这个

按分组

>>
idx  a  b
0  15  15
1  15  15
2  15  15
0  20  20
1  20  20
2  20  20
3  20  20
4  20  20

Answer 1

使用累计

>>> df['idx'] = df.groupby('idx').cumcount()
>>> df
idx   a   b
0    0  15  15
1    1  15  15
2    2  15  15
3    0  20  20
4    1  20  20
5    2  20  20
6    3  20  20
7    4  20  20