'YHh pdZddlmZddlZddlZddZdZdZ dZ ddZ d Z d Z d Zd Zd ZdZdZy)z7 Compatibility tools for various data structure inputs )NP_LT_2NcJddlm}m}t|j||fs t d|jj |jj}ntj|j}|j|st dj||y)Nr) DatetimeIndex PeriodIndexz0The index must be a DatetimeIndex or PeriodIndexzExpected frequency {}. Got {}) pandasrr isinstanceindex ValueErrorfreqfreqstrpd infer_freq startswithformat)xr rr inferred_freqs O/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/tools/data.py_check_period_indexr s1 agg {; <KLLww||  agg.  # #D )8??@MOP P *c6t|tjSN)rr DataFrameobjs r is_data_framers c2<< ((rc&ddlm}t||S)Nr) DesignMatrix)patsyrr)rrs ris_design_matrixrs" c< ((rcjt|tjxr|jjduSr)rnpndarraydtypenamesrs r_is_structured_ndarrayr%!s% c2:: & F399??$+FFrcpt|tjr@tj|}|t |j dDcgc]}d|z }}nnt |r5|j}|j}|j}|j}n.tdjt|jt|ts t|}t!||j dk7r t#d|"t!|t!|k7r t#d|||fScc}w)aA Convert passed data structure to form required by estimation classes Parameters ---------- data : array_like colnames : sequence or None May be part of data structure rownames : sequence or None Returns ------- (values, colnames, rownames) : (homogeneous ndarray, list) zY_%dzCannot handle input type {typ})typz;length of colnames does not match number of columns in dataz8length of rownames does not match number of rows in data)rr!r"asarrayrangeshaperdropnavaluescolumnsr TypeErrorrtype__name__listlenr )datacolnamesrownamesr-is rinterpret_datar8%s$ #D!  ,1&,,q/,BCq CHC t {{}<<::8DJ$7$78: : h %> 8} Q'./ /H V <+, , 8X %%/Ds D3c|jtt|jjfft j S)N)r0)viewfloatr3r#r$r!r")arrs rstruct_to_ndarrayr=Rs. 88US1342::8 FFrc~t|tjuxr!t|tjuxs|duSr)r0r!r"endogexogs r_is_using_ndarray_typerBVs5 K2:: % 7 $Z2:: % 58rczt|tjxr t|tjxs|duSr)rr!r"r?s r_is_using_ndarrayrD[s1 ubjj ) ; bjj ) 9TT\t|xrt|xs|duSr)rr?s r_is_using_patsyrMns% U # 5 d # 3tt|6rctr$t|tjjSt|tj jS)z, Returns true if data is a recarray )rrr!corerecarrayrec)r4s r _is_recarrayrRts3$ 0 011$00r)M)NN)__doc__statsmodels.compat.numpyrnumpyr!rr rrrr%r8r=rBrDrIrKrMrRrrrXsV- P)) G*&ZG8 < E 6 1r