有没有更快的方法来做这个循环？

Question

我想使用以下循环创建一个新列。表中只有“open”和“start”列。我想创建一个新列“startopen”，如果“start”等于1，那么“startopen”等于“open”。否则，“startopen”等于此新创建列上方行中的任何“startopen”。目前，我能够通过以下方式实现这一点：

for i in range(df.shape[0]):
    if df['start'].iloc[i] == 1:
        df.loc[df.index[i],'startopen'] = df.loc[df.index[i],'open']
    else:
        df.loc[df.index[i],'startopen'] = df.loc[df.index[i-1],'startopen']

这有效，但对于大型数据集来说非常慢。是否有任何内置函数可以更快地完成此操作？

Answer 1

我想创建一个新列“startopen”，其中如果“start”等于1，则“startopen”等于“open”

否则，“startopen”等于此新创建列上方行中的任何“startopen”。

IIUC，否则部分等于用最后等于1的< code>startopen向前填充非1的< code>startopen

df['startopen'] = pd.Series(np.where(df['start'].eq(1), df['open'], np.nan), index=df.index).ffill()