熊猫 for 循环太慢

Question

我现在正忙于分析 CSV 数据日志文件，我现在正在努力解决如何加快一些 Pandas 计算。 以下 for 循环正在运行，但速度不是很快：

for i in range(len(df)):

    if i==0:
        df.loc[i,"delta_t"] = 3
        df.loc[i,"E_sun"] = 0
    else:
        df.loc[i,"delta_t"] = (df.loc[i,"date_time"]- df.loc[i-1,"date_time"]).total_seconds()
        df.loc[i,"E_sun"] = df.loc[i-1,"E_sun"] + df.loc[i,"delta_t"] * df.loc[i,"E_flow_sun"]

有没有一种快速的方法来进行这个计算？ 问题是我正在引用不同行的信息。 如果所有数据都在同一行上，那么事情就变得很容易了，例如：

df["column1"] = df["column2"] + df["column3"] 

Answer 1

您正在查看diff和cumsum ：

df['delta_t'] = df['date_time'].diff().total_seconds().fillna(3)

df['E_sun'] = (df['delta_t'] * df['E_flow_sun']).cumsum()