我正在从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列)的末尾。该怎么做?
如果我预测BelRd(2)列的值,那么我想将该值存储在同一BelRd(2)列的末尾。
修改2: 如果对于BelRd(d2)列,预测值为2,则该列应看起来像这样,在列末尾有一个额外的2:
答案 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)