U mdi@s$dZddlZdddZd ddZdS) z4 SARIMAX tools. Author: Chad Fulton License: BSD-3 NcCst|}|dkrdnd|}t||tksBtd||f|t}t|dkrftd||jdkr|jddkr|dddf}n|jdkrtd ||f|jdkr| }nt |dkrd}nd|k}t |dkdk}t|dk}|s|r.|rtd |t |dkdd}n t |}t |dkrLd}n2t|tdt |dkrv|d }n|}t|tott|dk}|rtd ||S) a+ Standardize lag order input. Parameters ---------- order : int or array_like Maximum lag order (if integer) or iterable of specific lag orders. title : str, optional Description of the order (e.g. "autoregressive") to use in error messages. Returns ------- order : int or list of int Maximum lag order if consecutive lag orders were specified, otherwise a list of integer lag orders. Notes ----- It is ambiguous if order=[1] is meant to be a boolean list or a list of lag orders to include, but this is irrelevant because either interpretation gives the same result. Order=[0] would be ambiguous, except that 0 is not a valid lag order to include, so there is no harm in interpreting as a boolean list, in which case it is the same as order=0, which seems like reasonable behavior. Examples -------- >>> standardize_lag_order(3) 3 >>> standardize_lag_order(np.arange(1, 4)) 3 >>> standardize_lag_order([1, 3]) [1, 3] Norderz%s orderz)Invalid %s. Non-integer order (%s) given.rz#Terms in the %s cannot be negative.zeInvalid %s. Must be an integer or 1-dimensional array-like object (e.g. list, ndarray, etc.). Got %s.zInvalid %s. Appears to be a boolean list (since it contains a 0 element and/or multiple elements) but also contains elements greater than 1 like a list of lag orders.z+Invalid %s. Cannot have duplicate elements.)nparrayallZastypeint ValueErroranyndimshapeitemlensumwheresortZarangetolist isinstancelistdiff)rtitleZ has_zerosZhas_multiple_onesZ has_gt_oneZ has_duplicaterT/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/tsa/arima/tools.pystandardize_lag_order sL&             rFcCs|dkr dnd|}zBtj|td}dd|D}t|rBtnt}tj||d}Wn tk rvtd|YnX|stt |stt |rtd|t t |}|j |fkr|d krdnd }td ||||j f|S) a Validate parameter vector for basic correctness. Parameters ---------- params : array_like Array of parameters to validate. length : int Expected length of the parameter vector. allow_infnan : bool, optional Whether or not to allow `params` to contain -np.Inf, np.Inf, and np.nan. Default is False. title : str, optional Description of the parameters (e.g. "autoregressive") to use in error messages. Returns ------- params : ndarray Array of validated parameters. Notes ----- Basic check that the parameters are numeric and that they are the right shape. Optionally checks for NaN / infinite values. Nz for %s)dtypecSsg|]}t|tqSr)rcomplex).0prrr sz"validate_basic..z,Parameters vector%s includes invalid values.z/Parameters vector%s includes NaN or Inf values.rszOSpecification%s implies %d parameter%s, but values with shape %s were provided.)rrobjectZravelr rfloat TypeErrorr isnanisinfZ atleast_1dZsqueezer )paramslengthZ allow_infnanrZ is_complexrpluralrrrvalidate_basicos.   r*)N)FN)__doc__numpyrrr*rrrrs e