计算熊猫系列中的连续个数

问题描述

如果我有以下熊猫系列：

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

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

[2,2,3]

如何获得以上输出？

解决方法

让我们尝试使用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]

,

以下方法应该起作用：

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

输出：

[2,3]