2017-01-27 12 views
3

Я пытаюсь выяснить, как показывать полосы True или False в серии pandas.Полосы истинного или ложного в пандах Серия

данные:

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 

Я попытался p.diff(), но не уверены, как сосчитать False значения этого сгенерированного, чтобы показать свою желаемого результата который выглядит следующим образом :.

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

ответ

3

Вы можете использовать cumcount из consecutives групп, созданных сравнить, если p не равен с shift ред p и 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 

Спасибо MaxU другого решения:

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 
+1

@MaxU - спасибо;) – jezrael

+0

выглядит хорошо спасибо :-) и спасибо @MaxU – ade1e

0

Другим альтернативным решением является совокупная сумма pSeries и вычтите самую последнюю суммарную сумму, где p - 0. Затем инвертируйте p и сделайте то же самое. Последний множественный Series вместе:

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