>h_>SrSSKrSSKrSSKrSSKJr SSK J r Sr SSjr SSjr SSjrSS jrS rSS jrS rS rSrSSjrSrSrSr"SS\5rSSjrSSjrg)z Utility functions models code N)_is_using_pandas) array_likec[U[5(aU$[U[5(aURS5$[U5$)Nlatin1) isinstancestrbytesdecode)ss jC:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\statsmodels/tools/tools.pyasstr2r s9!S Au  xx!!1v c>0n[U5H up4XBX1-'M U$)zX Helper function to create a dictionary mapping a column number to the name in tmp_arr. ) enumerate)tmp_arroffsetcol_mapicol_names r _make_dictnamesrs) G ) & * Nrc[R"U5nURS:Xa USS2S4nUb[R"U5nURS:Xa USS2S4n[R"[R "U5R U5)[R "U5R U5)5nXX4$[R "U5R U5)nX$)aa Returns views on the arrays Y and X where missing observations are dropped. Y : array_like X : array_like, optional axis : int Axis along which to look for missing observations. Default is 1, ie., observations in rows. Returns ------- Y : ndarray All Y where the X : ndarray Notes ----- If either Y or X is 1d, it is reshaped to be 2d. N)npasarrayndimarray logical_andisnanany)YXaxiskeepidxs r drop_missingr$ s( 1 Avv{ agJ} HHQK 66Q;!T' A.."((1+//$"7!7"$((1+//$"7!79z1:%%88A;??4((zrc[S5e)ax Construct a dummy matrix from categorical variables .. deprecated:: 0.12 Use pandas.get_dummies instead. Parameters ---------- data : array_like An array, Series or DataFrame. This can be either a 1d vector of the categorical variable or a 2d array with the column specifying the categorical variable specified by the col argument. col : {str, int, None} If data is a DataFrame col must in a column of data. If data is a Series, col must be either the name of the Series or None. For arrays, `col` can be an int that is the (zero-based) column index number. `col` can only be None for a 1d array. The default is None. dictnames : bool, optional If True, a dictionary mapping the column number to the categorical name is returned. Used to have information about plain arrays. drop : bool Whether or not keep the categorical variable in the returned matrix. Returns ------- dummy_matrix : array_like A matrix of dummy (indicator/binary) float variables for the categorical data. dictnames : dict[int, str], optional Mapping between column numbers and categorical names. Notes ----- This returns a dummy variable for *each* distinct variable. If a a DaataFrame is provided, the names for the new variable is the old variable name - underscore - category name. So if the a variable 'vote' had answers as 'yes' or 'no' then the returned array would have to new variables-- 'vote_yes' and 'vote_no'. There is currently no name checking. Examples -------- >>> import numpy as np >>> import statsmodels.api as sm Univariate examples >>> import string >>> string_var = [string.ascii_lowercase[0:5], ... string.ascii_lowercase[5:10], ... string.ascii_lowercase[10:15], ... string.ascii_lowercase[15:20], ... string.ascii_lowercase[20:25]] >>> string_var *= 5 >>> string_var = np.asarray(sorted(string_var)) >>> design = sm.tools.categorical(string_var, drop=True) Or for a numerical categorical variable >>> instr = np.floor(np.arange(10,60, step=2)/10) >>> design = sm.tools.categorical(instr, drop=True) With a structured array >>> num = np.random.randn(25,2) >>> struct_ar = np.zeros((25,1), ... dtype=[('var1', 'f4'),('var2', 'f4'), ... ('instrument','f4'),('str_instr','a5')]) >>> struct_ar['var1'] = num[:,0][:,None] >>> struct_ar['var2'] = num[:,1][:,None] >>> struct_ar['instrument'] = instr[:,None] >>> struct_ar['str_instr'] = string_var[:,None] >>> design = sm.tools.categorical(struct_ar, col='instrument', drop=True) Or >>> design2 = sm.tools.categorical(struct_ar, col='str_instr', drop=True) zcategorical has been removed)NotImplementedError)datacol dictnamesdrops r categoricalr+Gs` < ==rc[US5(aSSKJn U"USXS9$[R"U5nUR nUS:Xa USS2S4nOUR S:a [ S5e[R"USS 9S:HnU[R"US :gSS 9-nUR5(a~US :XaU$US :XapUS:Xa [ S 5e[R"URS5nSRXvVs/sHn[U5PM sn5n [ SU S35e[R"URS5U/nU(aUOUSSS2n[R"U5$s snf)a Add a column of ones to an array. Parameters ---------- data : array_like A column-ordered design matrix. prepend : bool If true, the constant is in the first column. Else the constant is appended (last column). has_constant : str {'raise', 'add', 'skip'} Behavior if ``data`` already has a constant. The default will return data without adding another constant. If 'raise', will raise an error if any column has a constant value. Using 'add' will add a column of 1s if a constant column is present. Returns ------- array_like The original values with a constant (column of ones) as the first or last column. Returned value type depends on input type. Notes ----- When the input is a pandas Series or DataFrame, the added column's name is 'const'. Nr) add_trendc)trendprepend has_constantrz)Only implemented for 2-dimensional arraysr"skipraisezdata is constant.,z Column(s) z are constant.)rstatsmodels.tsa.tsatoolsr-rrr ValueErrorptpallrarangeshapejoinrones column_stack) r'r0r1r-xris_nonzero_constcolumnsr.colss r add_constantrFs_8d##6S'UU 4A 66D qy agJ !DEEvvaa(A-qCxa00 6 !H W $qy !455))AGGAJ/xx1J K1JAQ1J KL :dV>!BCC  a A !DbD'A ??1  !LsE?c[USSS9n[USSS9nURS:Xa USSS24OUnURSURS:wa[SURS-5e[R "X/5n[R RU5[R RU5:wag g ) a True if (Q, P) contrast `c` is estimable for (N, P) design `d`. From an Q x P contrast matrix `C` and an N x P design matrix `D`, checks if the contrast `C` is estimable by looking at the rank of ``vstack([C,D])`` and verifying it is the same as the rank of `D`. Parameters ---------- c : array_like A contrast matrix with shape (Q, P). If 1 dimensional assume shape is (1, P). d : array_like The design matrix, (N, P). Returns ------- bool True if the contrast `c` is estimable on design `d`. Examples -------- >>> d = np.array([[1, 1, 1, 0, 0, 0], ... [0, 0, 0, 1, 1, 1], ... [1, 1, 1, 1, 1, 1]]).T >>> isestimable([1, 0, 0], d) False >>> isestimable([1, -1, 0], d) True r.r2)maxdimdrrNzContrast should have %d columnsFT)rrr>r:rvstacklinalg matrix_rank)r.rInews r isestimablerOs> 1c!$A1c"Affk$' qAwwqzQWWQZ:QWWQZGHH ))QF C yyS!RYY%:%:1%== rc |[R"U5nUR5n[RR US5up#n[R "U5nUR SnUR SnU[RRU5-n[[Xv55Hn X9U:a SX9- X9'MSX9'M [R"[R"U5[R"USS2[R4[R"U555n X4$)zm Return the pinv of an array X as well as the singular values used in computation. Code adapted from numpy. Frr?r4N)rr conjugaterLsvdcopyr>maximumreducerangemindot transposemultiplynewaxis) rBrcondur vts_origmncutoffrress r pinv_extendedres 1 A Ayy}}Q&HA" WWQZF  A  A RZZ&&q) )F 3q9  4&=ad7ADAD  &&b!2;;qBJJ/?/1||A$@ AC ;rch[R"U5n[R"U[RS9n[R"UR 5nU)nX3UR US:-X3'SUR U- UR U'[R UR U'U$)z Reciprocal of an array with entries less than or equal to 0 set to 0. Parameters ---------- x : array_like The input array. Returns ------- ndarray The array with 0-filled reciprocals. dtyperrQrr zeros_likefloat64rflatnan)rBoutnansposs r reciprrqs 1 A -- ,C 88AFF D %Cx166#;?+CH!&&+%CHHSMVVCHHTN Jrch[R"U5n[R"U[RS9n[R"UR 5nU)nX3UR US:g-X3'SUR U- UR U'[R UR U'U$)z Reciprocal of an array with entries less than 0 set to 0. Parameters ---------- x : array_like The input array. Returns ------- ndarray The array with 0-filled reciprocals. rgrrQri)rBrnronon_zeros r recipr0rt/s 1 A -- ,C 88AFF DuH!+qvvh/?1/DEHqvvh//CHHXVVCHHTN Jrc[RRUS-S5n[R"U5Vs/sH o SS2U4PM nn[R"[R "U55$s snf)z Erase columns of zeros: can save some time in pseudoinverse. Parameters ---------- matrix : ndarray The array to clean. Returns ------- ndarray The cleaned array. r2rN)raddrV flatnonzerorrZ)matrixcolsumrvals r clean0r{Gs^VV]]619a (F!#!7 8!7A!Q$>> from numpy import mean >>> from numpy.random import standard_normal >>> x = standard_normal((3,4,5)) >>> m = mean(x, axis=1) >>> m.shape (3, 5) >>> m = unsqueeze(m, 1, x.shape) >>> m.shape (3, 1, 5) >>> r)listreshape)r'r"oldshapenewshapes r unsqueezer{s#>H~HHN << !!rch[R"[R"U5US:g5n[R"US:g[R"U55nX#-n[R"[R"U5[R"U55n[RXT'U$)z Returns np.dot(left_matrix, right_matrix) with the convention that nan * 0 = 0 and nan * x = nan if x != 0. Parameters ---------- A, B : ndarray r)rrYr nan_to_numrm)ABshould_be_nan_1should_be_nan_2 should_be_nanCs r nan_dotrsyffRXXa[163Offa1frxx{3O#5M r}}Qq!12AvvA Hrc[USU5$)zx Gets raw results back from wrapped results. Can be used in plotting functions or other post-estimation type routines. _results)getattr)resultss r maybe_unwrap_resultsrs 7J 00rc,^\rSrSrSrU4SjrSrU=r$)Bunchiz Returns a dict-like object with keys accessible via attribute lookup. Parameters ---------- *args Arguments passed to dict constructor, tuples (key, value). **kwargs Keyword agument passed to dict constructor, key=value. c2>[TU]"U0UD6 XlgN)super__init____dict__)selfargskwargs __class__s r rBunch.__init__s $)&) r)r)__name__ __module__ __qualname____firstlineno____doc__r__static_attributes__ __classcell__)rs@r rrs rrcPUcU$[US5nURS:XaU(a XR4$US4$URS:a [S5eU(a UROSnU(a[ R "U5SS2S4U4$[R"U5U4$)a Parameters ---------- x : ndarray, Series, DataFrame or None Input to verify dimensions, and to transform as necesary ndarray : bool Flag indicating whether to always return a NumPy array. Setting False will return an pandas DataFrame when the input is a Series or a DataFrame. Returns ------- out : ndarray, DataFrame or None array or DataFrame with 2 dimensiona. One dimensional arrays are returned as nobs by 1. None is returned if x is None. names : list of str or None list containing variables names when the input is a pandas datatype. Returns None if the input is an ndarray. Notes ----- Accepts None for simplicity Nr2zx mst be 1 or 2-dimensional.) rrrDr:namerrpd DataFrame)rBndarray is_pandasrs r _ensure_2drs2 y D)Ivv{ ii< d7N !788166DDzz!}QW%t++||A$$rc[USSS9nUS:XawUSS2[R"US:gSS94nU[R"US-R S55- nUR U-n[R RXSS 9$US :Xa[R RUS S 9un[R"[R"U55nUc:USURS -[R"[5R-n[!XA:R 55$[R RXS9$)a Matrix rank calculation using QR or SVD Parameters ---------- m : array_like A 2-d array-like object to test tol : float, optional The tolerance to use when testing the matrix rank. If not provided an appropriate value is selected. method : {"ip", "qr", "svd"} The method used. "ip" uses the inner-product of a normalized version of m and then computes the rank using NumPy's matrix_rank. "qr" uses a QR decomposition and is the default. "svd" defers to NumPy's matrix_rank. Returns ------- int The rank of m. Notes ----- When using a QR factorization, the rank is determined by the number of elements on the leading diagonal of the R matrix that are above tol in absolute value. rar2rJipNrr3T)tol hermitianqrr)moder)r)rrrsqrtsumTrLrMscipyrabsdiagr>finfofloatepsint)rarmethodrabs_diags r rMrMs8 1c"A ~ aQQ'' ( a Q( ( CC!Gyy$$Q4$@@ 4 \\__QS_ )66"''!*% ;1+ *RXXe_-@-@@CHN'')**yy$$Q$00r)r)Nr)NFF)Tr5)gV瞯rs33 NP>h7t'T000'&>B!"H .1 D  (%V)1r