我在数据框中有一个布尔True/ False-列“ ”,例如:Mask
Mask
True
True
True
False
False
True
False
False
Run Code Online (Sandbox Code Playgroud)
现在我尝试添加一列,其中包含连续True/False行的计数,其中True是正和(计数为+1),False是负和(计数为-1),例如
Mask Count
True 3
True 3
True 3
False -2
False -2
True 1
False -2
False -2
Run Code Online (Sandbox Code Playgroud)
我尝试过groupby,sum但现在我脑子里有一个结。
尝试过类似的东西
mask.groupby((~mask).cumsum()).cumsum().astype(int)
Run Code Online (Sandbox Code Playgroud)
(是/mask的条件) 但这仅计算 True 并进行计数而不显示总和。TrueFalse
非常感谢任何建议!