python - pandas 数据框中前几行的累积计数

Question

我有一个像这样的数据框:

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

想要ID的累计计数在前几天(包括那天)，标志是 True ，像这样:

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

我尝试了 bool 掩码和 cumsum() ，但没有让它发挥作用。有建议吗？

Answer 1

这不是您正在寻找的 groupby 和 cumsum

df.groupby('ID').flag.cumsum().astype(int)
Out[362]: 
0    1
1    1
2    2
3    2
4    3
Name: flag, dtype: int32