сгруппировать аналогичное значение индекса и переиндексировать каждую группу по новому индексу pandas

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

, в то время как я хочу назначить каждой группе отдельный индекс, как показано ниже. Как я могу получить это в pandas

группе по

>>
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

Использование cumcount

>>> 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