如何迭代组中的每条记录

Question

我想添加一个迭代组值的列。数据框根据需要进行了预先排序：

Index :|: GRP BY Col :|: New Col:|  
0     :|: 1          :|: 0      :|
1     :|: 1          :|: 1      :|   
2     :|: 1          :|: 2      :|   
3     :|: 2          :|: 0      :|   
4     :|: 2          :|: 1      :|    

Answer 1

我认为你需要cumcount：

df = pd.DataFrame({'GRP BY Col':[1,1,1,2,2]})
df['New Col'] = df.groupby('GRP BY Col').cumcount()
print (df)
   GRP BY Col  New Col
0           1        0
1           1        1
2           1        2
3           2        0
4           2        1