将相似的索引值分组，并通过新的索引熊猫为每个组重新索引

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

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