Pandas groupby - 计算相对点的距离

假设我有一些看起来像这样的东西

df = pd.DataFrame({'Event':['A','A','A','A', 'A' ,'B','B','B','B','B'],  'Number':[1,2,3,4,5,6,7,8,9,10],'Ref':[False,False,False,False,True,False,False,False,True,False]})

我想要做的是创建一个新列，该列是 Number 与 ref 中 True 的差异。因此，对于 A 组，True 是最后一个，因此该列将显示为 -4,-3,-2,-1,0。我一直在考虑做以下事情：

for col in df.groupby('Event'):
    temp = col[1]
    reference = temp[temp.Ref==True]
    dist1 = temp.apply(lambda x:x.Number-reference.Number,axis=1)

这似乎对每个组都正确计算，但我不确定如何将结果加入到 df 中。

在你的情况下

df['new']=(df.set_index('Event').Number-df.query('Ref').set_index('Event').Number).to_numpy()
df
  Event  Number    Ref  new
0     A       1  False   -4
1     A       2  False   -3
2     A       3  False   -2
3     A       4  False   -1
4     A       5   True    0
5     B       6  False   -3
6     B       7  False   -2
7     B       8  False   -1
8     B       9   True    0
9     B      10  False    1