将季节性免费数据添加到python

Question

我是python的新手，必须在没有季节性的情况下写入预测的CSV文件。所以我写了一个处理季节性问题的代码。

import numpy as np
import pandas as pd
from statsmodels.tsa.seasonal import seasonal_decompose

dateparse = lambda dates: pd.datetime.strptime(dates, '%Y-%m')
data = pd.read_csv('File1.csv', parse_dates=['Date'], index_col='Date', date_parser=dateparse)
ts = data['total_val']
ts_log = np.log(ts)
moving_avg = pd.rolling_mean(ts_log, 12)
decomposition = seasonal_decompose(ts_log)
seasonal = decomposition.seasonal
residual = ts_log - seasonal

现在我需要将结果写入同一文件中的新列。 我听说没有其他文件使用熊猫就可以这样做。任何人都可以帮忙吗？

Answer 1

将列添加到数据框：

data["residual"] = residual

写一个csv：

data.to_csv("newfile")