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

要以制表符分隔，可以使用to_csv的sep参数：

  

output.to_csv（文件名，sep ='\ t'）

要使用特定的编码（例如'utf-8'），请使用encoding参数：

  

output.to_csv（文件名，sep ='\ t'，编码='utf-8'）