Добавить заголовок файла к csv - PullRequest
Новая
       6

Добавить заголовок файла к csv

0 голосов
/ 27 декабря 2018

Итак, я перебираю несколько 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
...