Compter le nombre de 1 consécutifs dans une série de pandas

Question

Si j'ai la série de pandas suivante :

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

Je souhaite obtenir une liste des nombres de 1 consécutifs, c'est-à-dire, dans ce cas, la sortie requise est la suivante

[2,1,2,3]

Comment puis-je obtenir le résultat ci-dessus ?

Answer 1

Essayons d'utiliser cumsum créer la nouvelle clé

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

Ce qui suit devrait fonctionner :

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

Sortie :

[2, 1, 2, 3]