Python 两列布尔值之间的条件滚动计数器

提前感谢您帮助像我这样的python新手

我希望在“count”列中实现以下结果，而不使用愚蠢的slow for循环

我相信这是可能的矢量化。有什么建议吗

再次感谢

       A       B       count
    0  False   False     0
    1  True    False     1
    2  False   False     1 
    3  True    False     2
    4  False   True      1 # True value in the other column: reset the counter
    5  False   False     1
    6  True    False     1 # True value in the other column: reset the counter
    7  False   True      1 # True value in the other column: reset the counter
    8  False   True      2
    9  False   False     2
   10  False   True      3

您可以尝试：

df['count'] = (df.groupby(df.A.cumsum())['B'].cumsum() + df.groupby(df.B.cumsum())['A'].cumsum())

输出：

        A      B  count
0   False  False      0
1    True  False      1
2   False  False      1
3    True  False      2
4   False   True      1
5   False  False      1
6    True  False      1
7   False   True      1
8   False   True      2
9   False  False      2
10  False   True      3

---

绝对正确快速的回答。非常感谢你！你是怎么想的这么快？