Python的數據幀等級系列

Question

我所擁有的是超過2列的數據幀像這樣

columnA columnB 
    1  a 
    1  b 
    1  c 
    2  d 

我只是想排名columnA這樣

columnA columnB rankA 
    1  a  1 
    1  b  2 
    1  c  3 
    2  d  1 

那麼，應該怎麼辦？

Answer 1

使用groupby + cumcount：

df['rankA'] = df.groupby('columnA').cumcount() + 1 
print (df) 
    columnA columnB rankA 
0  1  a  1 
1  1  b  2 
2  1  c  3 
3  2  d  1