熊貓數據框組內的計算

Question

我已經熊貓數據框如下所示。我想要做的是，partition (or groupby) by BlockID, LineID, WordID，然後在每個組內使用current WordStartX - previous (WordStartX + WordWidth)來導出另一列，例如WordDistance，以指示該詞與前一個詞之間的距離。

這篇文章Row operations within a group of a pandas dataframe是非常有用的，但在我的情況下涉及多列（WordStartX和WordWidth）。

*BlockID LineID WordID WordStartX WordWidth  WordDistance 
0  0  0  0   275  150     0 
1  0  0  1   431   96 431-(275+150)=6   
2  0  0  2   642   90 642-(431+96)=115 
3  0  0  3   746  104 746-(642+90)=14 
4  1  0  0   273   69   ... 
5  1  0  1   352  151   ... 
6  1  0  2   510   92 
7  1  0  3   647   90 
8  1  0  4   752  105** 

Answer 1

的diff()和shift()函數通常用於計算指的是一個或下一個行有所幫助：

df['WordDistance'] = (df.groupby(['BlockID', 'LineID']) 
     .apply(lambda g: g['WordStartX'].diff() - g['WordWidth'].shift()).fillna(0).values)