python - 统计列中连续True的个数，False时重新开始

Question

我在 pandas df 中使用以下列:

A
True
True
True
False
True
True

我想添加 B 列来计算 A 中连续出现“True”的次数。每次出现“False”时我都想重新开始。期望的输出:

A      B
True   1
True   2
True   3
False  0
True   1
True   2

Answer 1

使用 cumsum 识别 A 列中的值保持 True 的行 block ，然后将 A 列分组code> 在这些 block 上计算累积和以分配序号

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

---

       A  B
0   True  1
1   True  2
2   True  3
3  False  0
4   True  1
5   True  2