首页 > 解决方案 > python中的i循环慢

问题描述

我在优化 for i 循环时遇到了麻烦,该循环使用基于前一个结果的下一个结果的时序数据。

import pandas as pd
import numpy as np
df= pd.Series(range(1,1000000))
df=df.to_frame(name="CusUnits")
df['UpperBoundary'] = 10
df['LowerBoundary'] = -10
from progress.bar import Bar
import time
from tqdm import tqdm
from time import sleep
length=len(df)
df.loc[0, 'Hedged'] = df.loc[0, 'CusUnits']

for i in tqdm(range(1, length)):
    if (df.loc[i, 'UpperBoundary'] > (df.loc[i-1, 'Hedged'] - df.loc[i, 'CusUnits']) > df.loc[i, 'LowerBoundary']):  df.loc[i, 'Hedged'] = df.loc[i-1, 'Hedged']     
    else :df.loc[i, 'Hedged'] =df.loc[i-1, 'CusUnits']

标签: pythonperformanceloopsfor-loop

解决方案



for i in tqdm(range(1, length)):
    if (df.at[i, 'UpperBoundary'] > (df.at[i-1, 'Hedged'] - df.at[i, 'CusUnits']) > df.at[i, 'LowerBoundary']):  df.at[i, 'Hedged'] = df.at[i-1, 'Hedged']     
    else :df.at[i, 'Hedged'] =df.at[i-1, 'CusUnits']

the replacement of loc with at solved many thanks for you comments

推荐阅读