计算 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创建新密钥

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]

Answer 2

以下应该有效：

[len(i) for i in ''.join([str(i) for i in list(s)]).split('0') if '1' in i]

输出：

[2, 1, 2, 3]