Séries de Vrai ou Faux dans les pandas Série

Question

J'essaie de trouver comment montrer des stries de True ou de False dans une série de pandas.

données:

p = pd.Series([True,False,True,True,True,True,False,False,True]) 

0  True 
1 False 
2  True 
3  True 
4  True 
5  True 
6 False 
7 False 
8  True 
dtype: bool 

J'ai essayé p.diff() mais pas sûr de savoir comment compter les valeurs False cela engendre pour montrer ma sortie désirée est la suivante :.

0  0 
1  0 
2  0 
3  1 
4  2 
5  3 
6  0 
7  1 
8  0 

Answer 1

Vous pouvez utiliser cumcount des groupes consécutifs créés par comparer si p n'est pas égal avec shift ed p et cumsum:

print (p.ne(p.shift())) 
0  True 
1  True 
2  True 
3 False 
4 False 
5 False 
6  True 
7 False 
8  True 
dtype: bool 

print (p.ne(p.shift()).cumsum()) 
0 1 
1 2 
2 3 
3 3 
4 3 
5 3 
6 4 
7 4 
8 5 
dtype: int32 

print (p.groupby(p.ne(p.shift()).cumsum()).cumcount()) 
0 0 
1 0 
2 0 
3 1 
4 2 
5 3 
6 0 
7 1 
8 0 
dtype: int64 

Merci MaxU pour une autre solution:

print (p.groupby(p.diff().cumsum()).cumcount()) 
0 0 
1 0 
2 0 
3 1 
4 2 
5 3 
6 0 
7 1 
8 0 
dtype: int64 

Answer 2

Une autre solution alternative est de créer la somme cumulée de pSeries et soustraire la somme cumulée la plus récente où p est 0. Puis inverser p et faire la même chose. Dernière plusieurs Series ensemble:

c = p.cumsum() 
a = c.sub(c.mask(p).ffill(), fill_value=0).sub(1).abs() 
c = (~p).cumsum() 
d = c.sub(c.mask(~(p)).ffill(), fill_value=0).sub(1).abs() 

print (a) 
0 0.0 
1 1.0 
2 0.0 
3 1.0 
4 2.0 
5 3.0 
6 1.0 
7 1.0 
8 0.0 
dtype: float64 

print (d) 
0 1.0 
1 0.0 
2 1.0 
3 1.0 
4 1.0 
5 1.0 
6 0.0 
7 1.0 
8 1.0 
dtype: float64 

print (a.mul(d).astype(int)) 
0 0 
1 0 
2 0 
3 1 
4 2 
5 3 
6 0 
7 1 
8 0 
dtype: int32