python - 计算 Pandas 系列中连续的个数

Question

如果我有以下 Pandas 系列:

import pandas as pd
s = pd.Series([0,1,1,0,1,0,0,1,1,0,1,1,1])

我想要一个连续计数的列表，即在这种情况下所需的输出是

[2,1,2,3]

如何获得上述输出？

Answer 1

让我们尝试使用 cumsum创建新 key

s[s==1].groupby(s.eq(0).cumsum()).sum()
1    2
2    1
4    2
5    3
dtype: int64

#s[s == 1].groupby(s.eq(0).cumsum()).sum().tolist()
#[2, 1, 2, 3]