Добавить заголовок файла к csv
Поэтому я перебираю несколько CSV-файлов и использую FBProphet, чтобы получить прогноз для них. Затем я записываю все прогнозы в один CSV-файл, используя.to_csv. Однако я хотел бы получить имя файла, чтобы разделить их на CSV и не уверен, есть ли способ сделать это при использовании.to_csv.
filenames=['example1.csv','example2.csv','example3.csv']
for f in filenames:
df=pd.read_csv(f)
df.columns=['ds','y']
df['ds']=pd.to_datetime(df['ds'])
m=Prophet(seasonality_mode='multiplicative')
m.fit(df)
future=m.make_future_dataframe(periods=6,freq='M')
forecast=m.predict(future)
forecast[['ds','yhat','yhat_lower','yhat_upper']].tail()
fig1=m.plot(forecast)
fig2=m.plot_components(forecast)
forecast.to_csv('final_data.csv',mode='a',header=True)
Итак, результаты, которые я получаю,
ds yhat yhat_lower yhat_upper
example1 forecast
ds yhat yhat_lower yhat_upper
example2 forecast
ds yhat yhat_lower yhat_upper
example 3 forecast
Результаты, которые я хотел бы получить:
example 1 filename
ds yhat yhat_lower yhat_upper
example1 forecast
example 2 filename
ds yhat yhat_lower yhat_upper
example2 forecast
example 3 filename
ds yhat yhat_lower yhat_upper
example 3 forecast