Оптимизация для l oop in pandas - PullRequest
Новая
       13

Оптимизация для l oop in pandas

0 голосов
/ 08 апреля 2020

У меня есть следующий код, который выполняет работу, которую я хочу, но довольно медленный: он создает новые столбцы в исходном фрейме данных с for для l oop, повторяющимся для i

Вот упрощенная выдержка из код: 

df['pm_0']=1000

for i in range(1,41):
    df['ch_enc'+str(i)]=0.05*df['pm_'+str(i-1)]
    df['dc'+str(i)]=0.01*df['pm_'+str(i-1)]
    df['pm_'+str(i)]=df['pm_'+str(i-1)]-df['ch_enc'+str(i)]-df['dc'+str(i)]

Я хотел бы найти более скоростной способ сделать то же самое. Я думал о списочном понимании или массивах numpy, но на самом деле не знаю, может ли это сработать в этом случае. У вас есть идея или метод, который я мог бы попробовать?

Большое спасибо за вашу помощь или предложения

...