Python + Pandas + Dataframe + Csv:将计算值存储在列的末尾

时间:2019-03-26 13:43:12

标签: python pandas csv dataframe

我正在从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列)的末尾。该怎么做?

编辑: csv看起来像这样: enter image description here

如果我预测BelRd(2)列的值,那么我想将该值存储在同一BelRd(2)列的末尾。

修改2: 如果对于BelRd(d2)列,预测值为2,则该列应看起来像这样,在列末尾有一个额外的2:

enter image description here

1 个答案:

答案 0 :(得分: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)