Python сохраняет данные в файл / глобальную переменную в цикле

Question

У меня есть два цикла for, в которых я вызываю функцию прогноза.Функция возвращает значение прогноза, и я хочу записать его в файл или во фрейм данных, чтобы я мог сохранить все прогнозы и получить к ним доступ вне области цикла.Я немного застрял здесь.позвольте мне показать на примере.

 for country in df['country'].unique():
      for channel in df['channel'].unique():
          output = pd.DataFrame(columns=['date', 'country', 
                                          'channel_id','value'])
          x = df[df['country']==country & df['channel']==channel]
          pred = forecast(df)
          output = output.append(pred)
          output.to_csv('forecast.csv')

Это то, что я хочу сделать, но проблема, очевидно, в том, что данные теряются после каждой итерации.Пожалуйста, предложите способ сохранить его.

Answer 1

Чтобы разделить вкладкой, вы можете использовать аргумент sep to_csv:

output.to_csv (имя_файла, sep = '\ t')

Использоватьопределенная кодировка (например, 'utf-8') использует аргумент кодирования:

output.to_csv (имя_файла, sep = '\ t', encoding = 'utf-8')