如何使用 groupby 为 Dataframe 中的每个组创建重复索引？ [复制]

Question

当使用 groupby() 时，如何创建一个 DataFrame，其中包含一个包含每个组递增索引的新列。例如，如果我有

df=pd.DataFrame('a':[1,1,1,2,2,2])
df
   a  
0  1  
1  1  
2  1  
3  2  
4  2  
5  2  

如何获得一个 DataFrame，其中为列中的每个新组重置索引。 a 和 index 之间的关联并不重要...只需要让每个案例 a 接收一个从 1 开始的唯一索引。

   a    idx
0  1    1
1  1    2
2  1    3
3  2    1
4  2    2
5  2    3

Answer 1

cmets 中的答案：

df['idx'] = df.groupby('a').cumcount() + 1