Kann DataFrame.apply auf vorherige Zeilen verweisen?

Question

Ich weiß, dass ähnliche Versionen dieser Frage gestellt wurden, und ich habe Antworten von nein gesehen, oder schaue auf rolling_apply, aber bisher habe ich noch nicht gesehen, wie ich eine schnellere Version dieses Codes implementieren kann, obwohl es scheint sollte möglich sein.

In einem Datenframe möchte ich eine Spalte erstellen, deren Werte entweder gleich einer anderen Spalte oder gleich ihrem vorherigen Wert (frühere Zeile) abhängig von einer zweiten Spalte sind. Ich kann das in einer Schleife machen, aber es ist langsam.

import pandas as pd 

df = pd.DataFrame({'a':[0,1,2,0,1,2,0,1,2],'t':[0,1,2,3,4,5,6,7,8]}) 

startTime = df.loc[0,'t'] 
df.loc[:,'startTime'] = startTime 

for idx in range(df.shape[0]): 
    if df.loc[idx,'a'] == 0: 
     startTime = df.loc[idx,'t'] 
    df.loc[idx,'startTime'] = startTime 

df 

Answer 1

Würde das funktionieren?

df['startTime'] = df.loc[df['a']==0, 't'].reindex(df.index).ffill() 

df 
Out: 
    a t startTime 
0 0 0  0.0 
1 1 1  0.0 
2 2 2  0.0 
3 0 3  3.0 
4 1 4  3.0 
5 2 5  3.0 
6 0 6  6.0 
7 1 7  6.0 
8 2 8  6.0