M/Phu8 ddlmZmZddlmZddlZddlZddZ ej dddddddfdZ Gd d Z dd Z dd ed edededeef dZdd ed ededefdZddZddZddZdS))AnyOptional)MappingNc|j}t|jD]}|dkrn|dz}t||}||jd|S)a Remove trailing singleton dimensions Parameters ---------- arr : ndarray Input array stop_dim : int Dimension where checking should stop so that shape[i] is not checked for i < stop_dim Returns ------- squeezed : ndarray Array with all trailing singleton dimensions (0 or 1) removed. Singleton dimensions for dimension < stop_dim are retained. N)ndimreversedshapemaxreshape)arrstop_dimlastss g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tools/validation/validation.py_right_squeezersi$ 8D ci  q55 E   tX  D ;;sy$' ( ((rFTc |r|dS| rdgng} |dks|r| dgz } n |dkr| dgz } tj||| } |"| j|kr|d|} t| n|| j|krt | |} n3| j|kr(tj| | jd|| jz zz} | j|kr+d } t| ||| j|qt| j|D][\} }|T| |krNt| d d }d } t| ||| j\| S) aD Convert array-like to a ndarray and check conditions Parameters ---------- obj : array_like An array, any object exposing the array interface, an object whose __array__ method returns an array, or any (nested) sequence. name : str Name of the variable to use in exceptions dtype : {None, numpy.dtype, str} Required dtype. Default is double. If None, does not change the dtype of obj (if present) or uses NumPy to automatically detect the dtype ndim : {int, None} Required number of dimensions of obj. If None, no check is performed. If the number of dimensions of obj is less than ndim, additional axes are inserted on the right. See examples. maxdim : {int, None} Maximum allowed dimension. Use ``maxdim`` instead of ``ndim`` when inputs are allowed to have ndim 1, 2, ..., or maxdim. shape : {tuple[int], None} Required shape obj. If None, no check is performed. Partially restricted shapes can be checked using None. See examples. order : {'C', 'F', None} Order of the array contiguous : bool Ensure that the array's data is contiguous with order ``order`` optional : bool Flag indicating whether None is allowed writeable : bool Whether to ensure the returned array is writeable Returns ------- ndarray The converted input. Examples -------- Convert a list or pandas series to an array >>> import pandas as pd >>> x = [0, 1, 2, 3] >>> a = array_like(x, 'x', ndim=1) >>> a.shape (4,) >>> a = array_like(pd.Series(x), 'x', ndim=1) >>> a.shape (4,) >>> type(a.orig) pandas.core.series.Series Squeezes singleton dimensions when required >>> x = np.array(x).reshape((4, 1)) >>> a = array_like(x, 'x', ndim=1) >>> a.shape (4,) Right-appends when required size is larger than actual >>> x = [0, 1, 2, 3] >>> a = array_like(x, 'x', ndim=2) >>> a.shape (4, 1) Check only the first and last dimension of the input >>> x = np.arange(4*10*4).reshape((4, 10, 4)) >>> y = array_like(x, 'x', ndim=3, shape=(4, None, 4)) Check only the first two dimensions >>> z = array_like(x, 'x', ndim=3, shape=(4, 10)) Raises ValueError if constraints are not satisfied >>> z = array_like(x, 'x', ndim=2) Traceback (most recent call last): ... ValueError: x is required to have ndim 2 but has ndim 3 >>> z = array_like(x, 'x', shape=(10, 4, 4)) Traceback (most recent call last): ... ValueError: x is required to have shape (10, 4, 4) but has shape (4, 10, 4) >>> z = array_like(x, 'x', shape=(None, 4, 4)) Traceback (most recent call last): ... ValueError: x is required to have shape (*, 4, 4) but has shape (4, 10, 4) NWCF)dtype requirementsz must have ndim <= )r)rz1{0} is required to have ndim {1} but has ndim {2}zNone, z*, z3{0} is required to have shape {1} but has shape {2}) nprequirer ValueErrorrr r formatzipstrreplace)objnamerrmaxdimr order contiguousoptional writeablereqsr msgactualreq req_shapes r array_liker-$sHCKt %C552D ||z|   #   *SD 9 9 9C  8f  66f66CS// !    8d?? t444CC X__*S#)ddSXo.F"FGGC 8t  ECSZZdCH==>> > sy%00 I IKFC6S==JJ..x?? K D)SY!G!GHHH Jrc eZdZdZdZddZdS) PandasWrappera Wrap array_like using the index from the original input, if pandas Parameters ---------- pandas_obj : {Series, DataFrame} Object to extract the index from for wrapping Notes ----- Raises if ``orig`` is a pandas type but obj and and ``orig`` have different numbers of elements in axis 0. Also raises if the ndim of obj is larger than 2. cj||_t|tjtjf|_dS)N) _pandas_obj isinstancepdSeries DataFrame _is_pandas)self pandas_objs r__init__zPandasWrapper.__init__s(%$Z")R\1JKKrNrctj|}|js|S|jd|z|z|jjdkrt d|jj}|||jd|z }|jdkr]|t|jdd}n t|tr|}n|d}| ||n|d|}tj |||S|jdkrX|t|jd d}|'g}|D] } | | |n| d|!|}tj||| St d ) a  Parameters ---------- obj : {array_like} The value to wrap like to a pandas Series or DataFrame. columns : {str, list[str]} Column names or series name, if obj is 1d. append : str String to append to the columns to create a new column name. trim_start : int The number of observations to drop from the start of the index, so that the index applied is index[trim_start:]. trim_end : int The number of observations to drop from the end of the index , so that the index applied is index[:nobs - trim_end]. Returns ------- array_like A pandas Series or DataFrame, depending on the shape of obj. rz;obj must have the same number of elements in axis 0 as origrNr"_)r"indexcolumns)r>r<z!Can only wrap 1 or 2-d array_like)rasarrayr6r r1rr<rgetattrr2rr3r4appendr5) r7r!r>rA trim_starttrim_endr<r"newcs rwrapzPandasWrapper.wraps,joo J 9Q<* $x /43C3I!3L L L!  &j%+a.8";;< 8q==t/>>GS)) "qz!!%vvd3E3EV3E3E9St5999 9 X]]!$"2ItDD! IIAJJvv1vHHHH<WEBBB B@AA Ar)NNrr)__name__ __module__ __qualname____doc__r9rFrrr/r/sJ  LLL6B6B6B6B6B6Brr/cf|r||S|rdnd}|r+t|tr|St|d|t|dr(t |jr|} t|S#t $r$td||wxYw)a Convert to bool or raise if not bool_like Parameters ---------- value : object Value to verify name : str Variable name for exceptions optional : bool Flag indicating whether None is allowed strict : bool If True, then only allow bool. If False, allow types that support casting to bool. Returns ------- converted : bool value converted to a bool N or Nonez must be a boolsqueezez({} must be a bool (or bool-compatible){})r2bool TypeErrorhasattrcallablerO Exceptionr)valuer"r&strict extra_texts r bool_likerXs*EM '/RJ B eT " " BLt@@J@@AA Aui   Xem%<%<   E{{     j))    s 3B.B0rUr"r&rVreturnc|r|dSt|ttjf}t |dr(t |jr|}t|ttjfr|st|S|s,|s* ||dzkrt|Sn#t$rYnwxYw|rdnd}td ||) Convert to int or raise if not int_like Parameters ---------- value : object Value to verify name : str Variable name for exceptions optional : bool Flag indicating whether None is allowed strict : bool If True, then only allow int or np.integer that are not bool. If False, allow types that support integer division by 1 and conversion to int. Returns ------- converted : int value converted to a int NrOrrMrNzJ{} must be integer_like (int or np.integer, but not bool or timedelta64){}) r2rPr timedelta64rRrSrOintintegerrTrQr)rUr"r&rVis_bool_timedeltarWs rint_liker`s.EMt"54*@AAui   Xem%<%<  %#rz*++4E5zz - !$$5zz!%    D '/RJ  #VD*55  sB00 B=<B=c4t||d|}|J|S)r[F)r&rV)r`)rUr"rV_ints rrequired_int_likercHs+* E4% ? ? ?D     Krc|r|dSt|t}t|ttjf}t |dr(t |jr|}t|ttj ttj fr|s|st|S|s>|r>   U||  wu~~ 199(( (  G %% %    D '/RJ  VD* % %  s?D DDc2|dSt|ts|rdnd}t|d||r|}|K||vrG|rdnd}dd|zdz}d|||}t ||S) a Check if object is string-like and raise if not Parameters ---------- value : object Value to verify. name : str Variable name for exceptions. optional : bool Flag indicating whether None is allowed. options : tuple[str] Allowed values for input parameter `value`. lower : bool Convert all case-based characters in `value` into lowercase. Returns ------- str The validated input Raises ------ TypeError If the value is not a string or None when optional is True. ValueError If the input is not in ``options`` when ``options`` is set. NrMrNz must be a string If not None, 'z', 'z{}{} must be one of: {})r2rrQlowerjoinrr)rUr"r&optionsrqrW options_textr)s r string_likerus: }t eS ! !@#+3ZZ 4>>*>>???  uG33(08__b V[[111C7 '.. l  oo Lrc|r|dSt|tr|r9t|ts$|rdnd}|rdnd}||d|}t||S)a Check if dict_like (dict, Mapping) or raise if not Parameters ---------- value : object Value to verify name : str Variable name for exceptions optional : bool Flag indicating whether None is allowed strict : bool If True, then only allow dict. If False, allow any Mapping-like object. Returns ------- converted : dict_like value NrorNz or dict_like (i.e., a Mapping)z must be a dict)r2rdictrQ)rUr"r&rVrW strict_textr)s r dict_likerys(EMt eW % %"5$//)18__b ;AI77r ?T??+??nn Lr)r)FF)F)FNT)FT)typingrrcollections.abcrnumpyrpandasr3rdoubler-r/rXrrPr]r`rcrmruryrKrrrs ######))))> )    ~~~~BJBJBJBJBJBJBJBJBZ& & & & TCH** **%)*;?* c]****ZSTc4////d++++\r