[英]Count consecutive number of ones in a pandas series
如果我有以下熊猫系列:
import pandas as pd
s = pd.Series([0,1,1,0,1,0,0,1,1,0,1,1,1])
我想要一个连续计数的列表,即在这种情况下所需的输出是
[2,1,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, 1, 2, 3]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.