如何迭代組中的每條記錄

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