M/Ph3ZdZddlmZddlZddlZddlZddlZddlmZddl Z ddl m Z ddl mZmZmZmZddgZd aGd d ZeZd d dZddddZdZd-dZd.dZdZd/dZdZdZdZdZdZ dZ!d Z"Gd!d"Z#Gd#d$e#Z$d%Z%ed&d' d0d)Z&ed&d' d1d*Z'Gd+d,Z(dS)2z Run x12/x13-arima specs in a subprocess from Python and curry results back into python. Notes ----- Many of the functions are called x12. However, they are also intended to work for x13. If this is not the case, it's a bug. )deprecate_kwargN)warn)Bunch)X13NotFoundError IOWarningX13Error X13Warningx13_arima_select_orderx13_arima_analysis)z x13as.exex13aszx12a.exex12a x13as_ascii x13as_htmlceZdZdZdS)_freq_to_periodc|drdS|drdS|drdSdS)NM QW4) startswith)selfkeys S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tsa/x13.py __getitem__z_freq_to_period.__getitem__ sW >>#   2 ^^C  1 ^^C  2  N)__name__ __module__ __qualname__rrrrrs#rrrr)rrlognoneauto)TFNc|rdndS)Nyesnor"xs rr+-sQ0EEDrTcj|X|tr>tj|stj|}|s?tddda|tjdd}|stjdd}n.|,tjdd}|stjdd}tD]a}tj||} tj |tj tj |cS#t$rY^wxYwdS)a If x12path is not given, then either x13as[.exe] or x12a[.exe] must be found on the PATH. Otherwise, the environmental variable X12PATH or X13PATH must be defined. If prefer_x13 is True, only X13PATH is searched for. If it is false, only X12PATH is searched for. NX12PATHX13PATHstdoutstderrF) endswith _binary_namesospathisdirdirnamegetenvjoin subprocess check_callPIPEOSError)x12path prefer_x13binaryx12s r _find_x12rD0s;w// >>w}}W%% /goog..G  /%ddd+ ?i 2..G /i 2..G )Ir** /i 2..G  gll7F++   !#jo)3 : : : :JJJ    D us4,D## D0/D0cFt|}|std|S)Nzxx12a and x13as not found on path. Give the path, put them on PATH, or set the X12PATH or X13PATH environmental variable.)rDr)r@s r _check_x12rFUs:  G M LMM M Nrctjd|}d}t|dkrt||\}}n||d}d}||fS)z Takes something like (1 1 0)(0 1 1) and returns a arma order, sarma order tuple. Also accepts (1 1 0) and return arma order and (0, 0, 0) z \([0-9 ]*?\)c ttttjdd|dS)Nz[()]r/ )tuplemapintresubsplitr)s rcleanz_clean_order..cleanes6SbfVR3399#>>??@@@rr)rrr)rMfindalllenrK)orderrPsorders r _clean_orderrV^sm J . .EAAA 5zzA~~E5)) vveAh &=rFc|r|rtd|r|d|zg}n|r|d|zg}n||g}|r||gz }tj|tjtjS)Nz&Cannot specify both meta and datameta.z-m z-d r1) ValueErrorr<Popenr>STDOUT)r@specpathoutnamemetadatametaargss rrun_specr`qs CCABBB #)* #)*"    D#-#4 6 6 66rcd}|d|dd|ddz }||d|dd|ddz }n|d|dd|ddz }|S) N z maxorder = (rrIrQ) z maxdiff = (zdiff = (r")maxordermaxdiffdiffoptionss r_make_automdl_optionsrhsG >II!1!1$!7!788I 9 sAeZdZdZdddddddddggdd d ddd d fd ZdS) SeriesSpecag Parameters ---------- data appendbcst : bool appendfcst : bool comptype compwt decimals modelspan name period precision to_print to_save span start title type Notes ----- Rarely used arguments divpower missingcode missingval saveprecision trimzero Unnamed SeriesFNrQrr"r)rQrQr/iayiʚ;c tt||g\}}d|ddd}d|ddd}||||| |||dS)N"@O)data appendbcst appendfcstperiodstarttitlerj)rK_bool_to_yes_nor)rrrjrrcomptypecompwtdecimals modelspanr precisionto_printto_savespanrr series_typedivpower missingcode missingval series_names r__init__zSeriesSpec.__init__ s"%_z7A79":": J)49((( #U3B3Z### dz$.vU$;      r)rr r!__doc__rr"rrrrs^<#3u!q1a"brRTD# rrc t|dr:t|jdkrtd||jd}ddt t|j } t|j j }n8#ttf$r$ddlm}t||j }YnwxYw|j d}|dkr|j|j}}n$|d kr|j|j}}ntd t|d r |jpd }nd }t)|||||d |}|S)NrkrQz3Does not handle DataFrame with more than one columnrz({})rb) infer_freqrrzqOnly monthly and quarterly periods are supported. Please report or send a pull request if you want this extended.rjr.)rrjrrr)rnrSrkrXrwr;rKrxryr{rindexfreqstrAttributeErrorpandas.tseries.apiryearmonthquarterrjr) r*rrr start_daterstperiodrj series_specs rpandas_to_series_specrsq) qy>>A  &'' ' ailO ==3sAHOO,=,=#>#>?? @ @D 6 1 J '666111111 AG!4!456J ||#**:h 1#**o ' 'E KKMME y*GO<< :w / /   ''ME6 eF @ @ @C JrceZdZdZdZdS)rc \|D]\}}t|||dSN)rsetattr)rrrrs rrzX13ArimaAnalysisResult.__init__ds< ,,.. & &JC D#u % % % % & &rclddlm}|}|ddd\}}|j|dd|dd |j|dd|dd |j|d d|d d |j|d d|d d| |S)Nr) _import_mplrrQT)sharexF)axlegendObservedzSeas. AdjustedrTrend Irregular) statsmodels.graphics.utilsrsubplotsrplot set_ylabelrrr tight_layout)rrpltfigaxess rrzX13ArimaAnalysisResult.ploths+::::::kmmLLAdL33 T d1ge444 Q:&&& T!WU333 Q+,,, 475111 Q7### tAwu555 Q;'''  rN)rr r!rrr"rrrrcs2&&&rr)NTr)NFF)rrNNNTFNFFNNFNTN)rrNNNTFNNNFNTN))rstatsmodels.compat.pandasrr6r<rrMwarningsrrrstatsmodels.tools.toolsrstatsmodels.tools.sm_exceptionsrrrr __all__r5r_period_to_freqrrrDrFrVr`rhrtr~rrrrrrrr r rr"rrr+s655555  ))))))999999999999 $%9 :. "/##s##688 00""""J&6666$$(!!! @ @ @*////////d(((V!#566DHBG6;8