U mdM@s,dZddlZddlmZddlmZddZd-dd Zd.d d Z d/d d Z d0ddZ ddZ ddZ GdddZGdddZedkr(eddgddggddgddgggZeddgddggddgddgggZeddgddggddgddgggZeddgddggddgddggddgddgggZejed dddddfd!ed dddddffZedddgdddgdddggd!ddgdd"dgddd#gggZejd$d%Ze eeZejjeeded&d'Z e d!dd%d%Z"e e"Z#eeZ$e$%de$&e$&d(d)deddgddggddgddggddgddgggZ'eddgddggd*dgdd+gggZ(eddgddggd,dgd*dggd+dgddgggZ)ee'e(Z*e+e,e*e+e*-e+e*-ee+e*.e+e*/e+e*0ee)Z1e+e1/e+e10dS)1a9 Helper and filter functions for VAR and VARMA, and basic VAR class Created on Mon Jan 11 11:04:23 2010 Author: josef-pktd License: BSD This is a new version, I did not look at the old version again, but similar ideas. not copied/cleaned yet: * fftn based filtering, creating samples with fft * Tests: I ran examples but did not convert them to tests examples look good for parameter estimate and forecast, and filter functions main TODOs: * result statistics * see whether Bayesian dummy observation can be included without changing the single call to linalg.lstsq * impulse response function does not treat correlation, see Hamilton and jplv Extensions * constraints, Bayesian priors/penalization * Error Correction Form and Cointegration * Factor Models Stock-Watson, ??? see also VAR section in Notes.txt N)signal)lagmatc Csbt|}t|}|jdkr.|dddf}|jdkr@td|jd}|jd}|d}|jdkrtj||dddfddS|jdkrt|jdkrtj||ddSt|jd|d|f}t |D]8}tj|dd|f|dd|fdd|dd|f<q|S|jdkr^t|dddddf|}||| |jddddf}|SdS) aapply an autoregressive filter to a series x Warning: I just found out that convolve does not work as I thought, this likely does not work correctly for nvars>3 x can be 2d, a can be 1d, 2d, or 3d Parameters ---------- x : array_like data array, 1d or 2d, if 2d then observations in rows a : array_like autoregressive filter coefficients, ar lag polynomial see Notes Returns ------- y : ndarray, 2d filtered array, number of columns determined by x and a Notes ----- In general form this uses the linear filter :: y = a(L)x where x : nobs, nvars a : nlags, nvars, npoly Depending on the shape and dimension of a this uses different Lag polynomial arrays case 1 : a is 1d or (nlags,1) one lag polynomial is applied to all variables (columns of x) case 2 : a is 2d, (nlags, nvars) each series is independently filtered with its own lag polynomial, uses loop over nvar case 3 : a is 3d, (nlags, nvars, npoly) the ith column of the output array is given by the linear filter defined by the 2d array a[:,:,i], i.e. :: y[:,i] = a(.,.,i)(L) * x y[t,i] = sum_p sum_j a(p,j,i)*x(t-p,j) for p = 0,...nlags-1, j = 0,...nvars-1, for all t >= nlags Note: maybe convert to axis=1, Not TODO: initial conditions Nzx array has to be 1d or 2drZvalid)mode) npZasarrayndim ValueErrorshaperZconvolveminzerosrange) xanvarnlagsZntrimresultiZyfZyvalidrV/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/tsa/varma_process.py varfilter%s,9         6 "rrc CsP|j\}}}||krtdt|d||f}|d|dddddf<|dd |d|ddddf<|dkrtd|dD]d}t||f}td|D]0} |t||  ||| ddddf7}q|||ddddf<q|dkrLt|d|dD]D}t|ddj||d||dddddfjtdq|S)acreates inverse ar filter (MA representation) recursively The VAR lag polynomial is defined by :: ar(L) y_t = u_t or y_t = -ar_{-1}(L) y_{t-1} + u_t the returned lagpolynomial is arinv(L)=ar^{-1}(L) in :: y_t = arinv(L) u_t Parameters ---------- ar : ndarray, (nlags,nvars,nvars) matrix lagpolynomial, currently no exog first row should be identity Returns ------- arinv : ndarray, (nobs,nvars,nvars) Notes ----- .exogenous variables not implemented not testedrrNrz+waiting for generalized ufuncs or something)r printrr rdotNotImplementedError) arnobsversionrnvarsnvarsexZarinvrtmpprrrvarinversefilters" $. 6 r$c Cs|j\}}}|d}|jd}||kr.td|jd|krDtd|dkrdt|||f}|} n8t||jd} t|| |f}||| |jd| <||| d<t| | |D]B} td|D]2} || t|| | ddf||  7<qq|S)aEgenerate an VAR process with errors u similar to gauss uses loop Parameters ---------- ar : array (nlags,nvars,nvars) matrix lagpolynomial u : array (nobs,nvars) exogenous variable, error term for VAR Returns ------- sar : array (1+nobs,nvars) sample of var process, inverse filtered u does not trim initial condition y_0 = 0 Examples -------- # generate random sample of VAR nobs, nvars = 10, 2 u = numpy.random.randn(nobs,nvars) a21 = np.array([[[ 1. , 0. ], [ 0. , 1. ]], [[-0.8, 0. ], [ 0., -0.6]]]) vargenerate(a21,u) # Impulse Response to an initial shock to the first variable imp = np.zeros((nobs, nvars)) imp[0,0] = 1 vargenerate(a21,imp) rrrzu needs to have nvars columnsN)r rr rr maxrr) ru initvaluesrr r!Znlagsm1rZsarstartrr#rrr vargenerates$%   2r)c st|j}||||7<t|j}t|}||t|j||<|fddttD}||t |<|S)apad with zeros along one axis, currently only axis=0 can be used sequentially to pad several axis Examples -------- >>> padone(np.ones((2,3)),1,3,axis=1) array([[ 0., 1., 1., 1., 0., 0., 0.], [ 0., 1., 1., 1., 0., 0., 0.]]) >>> padone(np.ones((2,3)),1,1, fillvalue=np.nan) array([[ NaN, NaN, NaN], [ 1., 1., 1.], [ 1., 1., 1.], [ NaN, NaN, NaN]]) csg|]}t||qSrslice.0kZendindZstartindrr szpadone..) rarrayr emptyfillr r rlentuple) rfrontbackaxis fillvaluer shapearroutmyslicerr/rpadones      r=cspt|j}||||8<t|j}t|j||<|fddttD}|t|S)a;trim number of array elements along one axis Examples -------- >>> xp = padone(np.ones((2,3)),1,3,axis=1) >>> xp array([[ 0., 1., 1., 1., 0., 0., 0.], [ 0., 1., 1., 1., 0., 0., 0.]]) >>> trimone(xp,1,3,1) array([[ 1., 1., 1.], [ 1., 1., 1.]]) csg|]}t||qSrr*r,r/rrr0-sztrimone..)rr1r r r rr4r5)rr6r7r8r r:r<rr/rtrimones   r>cCs6|j\}}}tjt||dddddf| fS)z?make reduced lagpolynomial into a right side lagpoly array N)r rr_eye)rrrZnvarexrrrar2full4s rAcCs|dd S)zconvert full (rhs) lagpolynomial into a reduced, left side lagpoly array this is mainly a reminder about the definition rNr)rrrrar2lhs;srBc@s:eZdZdZddZddZddZdd Zdd d Zd S)_Vara<obsolete VAR class, use tsa.VAR instead, for internal use only Examples -------- >>> v = Var(ar2s) >>> v.fit(1) >>> v.arhat array([[[ 1. , 0. ], [ 0. , 1. ]], [[-0.77784898, 0.01726193], [ 0.10733009, -0.78665335]]]) cCs||_|j\|_|_dS)N)yr rr )selfrDrrr__init__Usz _Var.__init__cCs||_|j}t|j|ddd}|ddd|f|_|dd|df|_tjj|j|jdd}||_ |d ||||_ t |j |_ |d|_|d |_dS) aestimate parameters using ols Parameters ---------- nlags : int number of lags to include in regression, same for all variables Returns ------- None, but attaches arhat : array (nlags, nvar, nvar) full lag polynomial array arlhs : array (nlags-1, nvar, nvar) reduced lag polynomial for left hand side other statistics as returned by linalg.lstsq : need to be completed This currently assumes all parameters are estimated without restrictions. In this case SUR is identical to OLS estimation results are attached to the class instance Zbothin)ZtrimoriginalNrZrcondrrr)rr rrDZyredxredrlinalglstsqZ estresultsreshapeZarlhsrAarhatrssZxredrank)rErr ZlmatresrrrfitYs  z_Var.fitcCs t|dst|j|j|_|jS)z:calculate estimated timeseries (yhat) for sample yhat)hasattrrrDrNrRrErrrpredicts z _Var.predictcCsF|jddddftjt|jj|jdddddf|_dS)a covariance matrix of estimate # not sure it's correct, need to check orientation everywhere # looks ok, display needs getting used to >>> v.rss[None,None,:]*np.linalg.inv(np.dot(v.xred.T,v.xred))[:,:,None] array([[[ 0.37247445, 0.32210609], [ 0.1002642 , 0.08670584]], [[ 0.1002642 , 0.08670584], [ 0.45903637, 0.39696255]]]) >>> >>> v.rss[0]*np.linalg.inv(np.dot(v.xred.T,v.xred)) array([[ 0.37247445, 0.1002642 ], [ 0.1002642 , 0.45903637]]) >>> v.rss[1]*np.linalg.inv(np.dot(v.xred.T,v.xred)) array([[ 0.32210609, 0.08670584], [ 0.08670584, 0.39696255]]) N)rOrrKinvrrJTZparamcovrTrrrcovmats*z _Var.covmatrNcCs*|dkrt||jf}t|j||jdS)acalculates forcast for horiz number of periods at end of sample Parameters ---------- horiz : int (optional, default=1) forecast horizon u : array (horiz, nvars) error term for forecast periods. If None, then u is zero. Returns ------- yforecast : array (nobs+horiz, nvars) this includes the sample and the forecasts N)r')rr r r)rNrD)rEZhorizr&rrrforecastsz _Var.forecast)rN) __name__ __module__ __qualname____doc__rFrQrUrXrYrrrrrCCs ) rCc@sdeZdZdZdddZdddZddd Zdd d Zd dZddZ dddZ dddZ ddZ dS) VarmaPolyaclass to keep track of Varma polynomial format Examples -------- ar23 = np.array([[[ 1. , 0. ], [ 0. , 1. ]], [[-0.6, 0. ], [ 0.2, -0.6]], [[-0.1, 0. ], [ 0.1, -0.1]]]) ma22 = np.array([[[ 1. , 0. ], [ 0. , 1. ]], [[ 0.4, 0. ], [ 0.2, 0.3]]]) NcCs||_||_|j\}}}||||_|_|_|dd|ft|k |_ |jdkrrt|d|_d|_ n|dt|k |_ |jd|_ ||k|_ |dd |_ dS)Nr)N.Tr)rmar rnvarallr rr@all isstructured isindependentZmalagsZhasexogZarm1)rErr_rr`r rrrrFs "   zVarmaPoly.__init__rcCsD|dk r|}n(|dkr|j}n|dkr.|j}ntd|d|jS)z4stack lagpolynomial vertically in 2d array Nrr_no array or name givenr)rr_r rMr`rErnamerrrvstackszVarmaPoly.vstackcCsN|dk r|}n(|dkr|j}n|dkr.|j}ntd|ddd|jjS)z6stack lagpolynomial horizontally in 2d array Nrr_rdrrr)rr_r swapaxesrMr`rWrerrrhstackszVarmaPoly.hstackverticalcCst|dk r|}n(|dkr|j}n|dkr.|j}ntd|d|j}|j\}}tj||d}||ddd|f<|S)zDstack lagpolynomial vertically in 2d square array with eye Nrr_rdr)r.)rr_r rMr`r rr@)rErrfZ orientationZastackedZlenpkr amatrrr stacksquares zVarmaPoly.stacksquarecCs2t|jdd|jddfd}|d|jS)z;stack ar and lagpolynomial vertically in 2d array rNrr)r concatenaterr_rMr`rErrrrvstackarma_minus1s$zVarmaPoly.vstackarma_minus1cCs:t|jdd|jddfd}|ddd|jS)zustack ar and lagpolynomial vertically in 2d array this is the Kalman Filter representation, I think rNrrr)rrmrr_rhrMr`rnrrrhstackarma_minus1s$zVarmaPoly.hstackarma_minus1cCsz|dk r|}n.|jr,||jdd }n|jdd }||}ttj|ddd}||_t |dk S)a>check whether the auto-regressive lag-polynomial is stationary Returns ------- isstationary : bool *attaches* areigenvalues : complex array eigenvalues sorted by absolute value References ---------- formula taken from NAG manual Nrr) rb reduceformrrlrsortrKeigvalsZ areigenvaluesabsrarErrkZevrrrgetisstationarys zVarmaPoly.getisstationarycCs|dk r|}n*|jr*||jdd}n|jdd}|jddkrZtgtj|_dS||}t tj |ddd}||_t |dk S)a>check whether the auto-regressive lag-polynomial is stationary Returns ------- isinvertible : bool *attaches* maeigenvalues : complex array eigenvalues sorted by absolute value References ---------- formula taken from NAG manual NrrTr)rcrqr_r rr1complexZ maeigenvaluesrlrrrKrsrtrarurrrgetisinvertible<s zVarmaPoly.getisinvertiblecCs|jdkrtd|j\}}}t|}z"tj|dd|ddf}Wn"tjjk rltddYnXt|D]}t |||||<qv|S)z. this assumes no exog, todo rzapoly needs to be 3drNzmatrix not invertiblezask for implementation of pinv) r r r rZ empty_likerKrVZ LinAlgErrorrr)rEZapolyrr!r rZa0invZlagrrrrq^s   "  zVarmaPoly.reduceform)N)Nr)Nr)Nrrj)N)N) rZr[r\r]rFrgrirlrorprvrxrqrrrrr^s      "r^__main__g?ggg333333g?g?grg?g333333?g?irrrIig?g333333?gffffff)r)N)rrrr)rrr)2r]numpyrZscipyrZstatsmodels.tsa.tsatoolsrrr$r)r=r>rArBrCr^rZr1Za21Za22Za23Za24r?r@Za31Za32randomZrandnutZar2srKrLrPrMZbhatrNvrQrYZar23Zma22Zar23nsZvprvarsrgrprvrxZvp2rrrrs  _ 4 = # s@      B