pandas data fram中前几行的累计计数

Question

我有这样一个数据帧：

 date   ID   flag
 Apr1   1    True
 Apr2   2    True
 May1   1    True
 May1   1    False
 May2   1    True

需要前几天（包括那天）的ID的累计计数，其中标志是True，如下所示：

^{pr2}$

我尝试了布尔蒙版和cumsum()，但没有让它起作用。建议？在

Answer 1

这似乎是你需要的：

df['count'] = df.groupby(by=['ID'])['flag'].cumsum().astype(int)

输出：

^{pr2}$