Hinzufügen einer Zählung zu früheren Zelle Wert in Pandas

Question

in Pandas Ich bin auf der Suche nach einem Wert in einer Spalte "B" in Abhängigkeit von den booleschen Werten aus einer anderen Spalte "A" hinzuzufügen. Also, wenn 'A' wahr ist, dann fange an zu zählen (d. H. Addiere eine Eins jeder neuen Zeile), solange 'A' falsch ist. Wenn 'A' ist True zurückgesetzt und starten Sie erneut zu zählen. Ich habe das mit einer "for" -Schleife gemacht, aber das ist sehr zeitaufwendig. Ich frage mich, ob es keine zeiteffiziente Lösung mehr gibt?

das Ergebnis sollte wie folgt aussehen:

Date  A  B 
01.2010 False 0 
02.2010 True 1 
03.2010 False 2 
04.2010 False 3 
05.2010 True 1 
06.2010 False 2 

Answer 1

Sie cumsum mit groupby verwenden können und cumcount:

print df 
    Date  A 
0 1.201 False 
1 1.201 True 
2 1.201 False 
3 2.201 True 
4 3.201 False 
5 4.201 False 
6 5.201 True 
7 6.201 False 
roll = df.A.cumsum() 
print roll 
0 0 
1 1 
2 1 
3 2 
4 2 
5 2 
6 3 
7 3 
Name: A, dtype: int32 

df['B'] = df.groupby(roll).cumcount() + 1 
#if in first values are False, output is 0 
df.loc[roll == 0 , 'B'] = 0 
print df 
    Date  A B 
0 1.201 False 0 
1 1.201 True 1 
2 1.201 False 2 
3 2.201 True 1 
4 3.201 False 2 
5 4.201 False 3 
6 5.201 True 1 
7 6.201 False 2 

Answer 2

Dank, habe ich die Lösung aus einem anderen Beitrag wie folgt aus:

rolling_count = 0 
def set_counter(val): 

    if val == False: 

    global rolling_count 
    rolling_count +=1 

    else: 

    val == True 
    rolling_count = 1 

    return rolling_count 

df['B'] = df['A'].map(set_counter)