Source code for snowdrop.src.numeric.sa.test_x13

import os, sys
import pandas as pd
import statsmodels.api as api
import matplotlib.pyplot as plt

path = os.path.dirname(os.path.abspath(__file__))
working_dir = os.path.abspath(path+"../../../..") 
os.chdir(working_dir)

from snowdrop.src.numeric.sa.x13 import x13


[docs] def test1(): file_path = os.path.abspath(working_dir + "/data/temp/Rail_Road_Data.xlsx") df_railroad_adj,df_railroad = x13(file_path=file_path,freq="M") df_railroad.columns=["Railroad"] fig, ax = plt.subplots(figsize=(13,6)) ax.set_title("Actual and Seasonally Adjusted Series: US Railroad Traffic \n Monthly (Jan 2000 - Feb 2017)") ax.plot(df_railroad_adj.index,df_railroad_adj["Adjusted"],linewidth=2, marker='', markersize=3, zorder=1,label="Seasonally Adjusted") ax.plot(df_railroad_adj.index,df_railroad["Railroad"],linewidth=2, marker='', markersize=3, zorder=1,label="Actual") plt.show()
[docs] def test2(): df = pd.DataFrame.from_records(api.datasets.co2.load().data) df['date'] = df.date.apply(lambda x: x.decode('utf-8')) df['date'] = pd.to_datetime(df.date, format='%Y%m%d') df.set_index('date') dta = df.resample("M") dta = dta.fillna(method="ffill") res = api.tsa.seasonal_decompose(dta) fig = res.plot() fig.set_size_inches(10, 5) plt.tight_layout()
if __name__ == '__main__': """ The main program """ test1() #test2()