B ZV1 @szdZddlZddlmZddlmZddlmZm Z ddZ dd Z dddZ dddZ dddZdddZdddZdS)zCorrelation plot functions.N) sort_values)utils)acfpacfcCs|t|}|rdnd}|dkr0t| t|}n8t|rRt| t|d}nd}t|tj}|d}|||fS)NFTr) boolnparangelenZisscalarintZ asanyarrayZastypemax)xlagszero irregularnlagsr>> import statsmodels.api as sm >>> import pandas as pd >>> dta = sm.datasets.elnino.load_pandas().data >>> dta['YEAR'] = dta.YEAR.astype(int).astype(str) >>> dta = dta.set_index('YEAR').T.unstack() >>> dates = pd.to_datetime(list(map(lambda x : '-'.join(x) + '-1', ... dta.index.values))) >>> dta.index = pd.DatetimeIndex(dates, freq='MS') >>> fig = sm.graphics.tsa.month_plot(dta) .. plot:: plots/graphics_month_plot.py r) DataFrameN)_check_period_indexM)freq)Series PeriodIndex)indexjfmasrndcSs|jS)N)Zmonth)yrrr=szmonth_plot..)r7r)pandasr:statsmodels.tools.datar;r>r?r9groupby) r datesr7rr:r;r>r?r6rrr month_plots%  rNc Csxddlm}|dkr.ddlm}||ddn$ddlm}m}||||ddd}d d d d g}t|d d|||dS)a@ Seasonal plot of quarterly data Parameters ---------- x : array-like Seasonal data to plot. If dates is None, x must be a pandas object with a PeriodIndex or DatetimeIndex with a monthly frequency. dates : array-like, optional If `x` is not a pandas object, then dates must be supplied. ylabel : str, optional The label for the y-axis. Will attempt to use the `name` attribute of the Series. ax : matplotlib.axes, optional Existing axes instance. Returns ------- matplotlib.Figure r)r:N)r;Q)r=)r>r?)r@Zq1Zq2Zq3Zq4cSs|jS)N)Zquarter)rHrrrrI_szquarter_plot..)r7r)rJr:rKr;r>r?r9rL) r rMr7rr:r;r>r?r6rrr quarter_plot@s   rP) NNr$TFFr%TN)NNr$r+Tr,TN)NN)NNN)NNN)__doc__ZnumpyrZstatsmodels.compat.pandasrZstatsmodels.graphicsrZstatsmodels.tsa.stattoolsrrrr#r*r.r9rNrPrrrrs   V W ' 2