python - 在 Python 中使用 if 条件加速逐行循环

Question

我有一个 600 万行的数据集，列是:symbol、timeStamp、open price 和 close price。我运行以下循环，虽然非常简单，但需要很长时间(如果 open price 是 nan，则从上一行获取 close price ):

for i in range(0,len(price2)):
    print(i)
    if np.isnan(price3.iloc[i,2]):
        price3.iloc[i,2]=price3.iloc[i-1,3]

我怎样才能加快这个循环？据我所知，我可以更改为 apply()，但我怎样才能在其中包含 if 条件？

Answer 1

您可以使用 pandas.Series.fillna 而不是 for 循环与收盘价的转移系列。

price3['open price'].fillna(price3['close price'].shift(1), inplace=True)

这是矢量化的，因此应该比您的 for 循环快得多。

请注意，我假设 price2 和 price3 具有相同的长度，您也可以在循环中迭代 price3。