当值更改和时间超过条件时，Pandas 数据帧中的增量计数

Question

每次Person的值发生变化时，将count重置为0。每次时间从初始增量超过2时增加count。

Person Time DesiredResult
B      0    0
B      2    0
B      2    0
B      4    1
A      0    0
S      0    0
S      1    0
S      2    0
S      4    1
S      8    2

生成数据框的代码：

df = pd.DataFrame({'Person': ['Bob','Bob','Bob','Bob','Alvin','Steve','Steve','Steve','Steve','Steve']
                     ,'Time': [0,2,2,4,0,0,1,2,4,8]
                     ,'DesiredResult': [0,0,0,1,0,0,0,0,1,2]})

Image附在下面的链接中

Answer 1

我假设你没有DesiredResult列。

要创建它，请定义以下函数：

def fn(grp):
    t0 = grp.iloc[0].Time
    return (grp.Time > t0 + 2).cumsum()

然后运行：

df['DesiredResult'] = df.groupby('Person').apply(fn).reset_index(level=0, drop=True)