Python + Pandas + dataframe + Csv: сохранение вычисленного значения в конце столбца

Question

Я получаю данные из столбца CSV-файла. Для каждого значения столбца я прогнозирую данные с помощью моей модели. Код:

prod = pd.read_csv('/home/hp/products/prod[130].csv', engine='python')

#data_belrd = prod['BelRd(D2)'].tolist()

for i in range(3,11):
    data_belrd = prod.iloc[:, i].tolist()
    print(data_belrd)
    #print('-'*80)

    model = AR(data_belrd)
    model_fit = model.fit(2)

# make prediction
    y_belrd = model_fit.predict(len(data_belrd), len(data_belrd))
    print(y_belrd)
    print('-'*80)

Я хочу сохранить прогнозные данные y_belrd в конце соответствующего столбца (i-й столбец). Как это сделать?

Edit: CSV выглядит так:

Если я прогнозирую значение для столбца BelRd (2), то я хочу сохранить значение в конце того же столбца BelRd (2).

Редактировать 2: Если для столбца BelRd (d2) прогнозируемое значение равно 2, то столбец должен выглядеть следующим образом с дополнительными 2 в конце столбца:

Answer 1

Вопрос немного неясен.

Для сохранения в новом столбце в фрейме данных

df['new column'] = data

Для сохранения фрейма данных в CSV используйте:

df['Column Name'].to_csv('csv_filename.csv')

Чтобы добавить строку в конец вашего фрейма данных, используйте:

df.append({'column name': value}, ignore_index=True)  

Чтобы вставить значение в i-й столбец, используйте:

df.append({df.columns[i]: value}, ignore_index=True)