2016-12-06 83 views
1

我是python的新手,必须写入存在的CSV文件,而无需季节性预测。所以我写了一个处理季节性问题的代码。在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 

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

+0

看一看[新用户大熊猫教训]的第4课(HTTP:// pandas.pydata.org/pandas-docs/stable/tutorials.html#lessons-for-new-pandas-users)。你可能还想花一些时间[这篇Pycon美国2015年的谈话](http://pandas.pydata.org/talks.html#pycon-us-2015) - 如果你计划使用熊猫,它很长,但是值得。 – wwii

回答

0

一列添加到数据帧:

data["residual"] = residual 

写一个CSV:

data.to_csv("newfile") 
+0

这非常有帮助,谢谢! – Hilaria