Итерирование размера окна в скользящей функции, добавление к новому фрейму данных?

Question

Я пытаюсь выполнить итеративный расчет из фрейма данных, каждый раз повторяя / изменяя размер окна, затем добавляя его в новый фрейм данных, каждый раз с новым столбцом.

код выглядит следующим образом:

window_list = [39, 52, 65, 78, 91, 104]

for i in window_list: 
    window = i
    df_SP_final = pd.DataFrame(index=df_SP.index)
    df_SP_final.insert(loc=0,column=i,value=0.5*(((df_SP.pcent - df_SP.pcent.rolling(window=window,min_periods=min_periods).mean())

Вывод - это просто 104 столбец. Похоже, цикл for работает неправильно?

Answer 1

Вы повторно инициализируете свой df_SP_final каждый раз, когда вы пытаетесь открыть новое окно. Я думаю, что вы хотели сделать:

window_list = [39, 52, 65, 78, 91, 104]
df_SP_final = pd.DataFrame(index=df_SP.index)

for window in window_list: 
    df_SP_final.insert(loc=0,column=i,value=0.5*(((df_SP.pcent - df_SP.pcent.rolling(window=window,min_periods=min_periods).mean())