ó áp7]c@s§dZddlmZddlZddlZdd„Zd„Zd„Z d„Z ddd „Z d „Z d „Zd „Zd „Zd„Zd„Zd„ZdS(s7 Compatibility tools for various data structure inputs iÿÿÿÿ(trangeNtMcCsžddlm}m}t|j||fƒs=tdƒ‚n|jjdk r^|jj}nt j |jƒ}|j |ƒsštdj ||ƒƒ‚ndS(Niÿÿÿÿ(t PeriodIndext DatetimeIndexs0The index must be a DatetimeIndex or PeriodIndexsExpected frequency {}. Got {}( tpandasRRt isinstancetindext ValueErrortfreqtNonetfreqstrtpdt infer_freqt startswithtformat(txRRRt inferred_freq((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_check_period_index s cCst|tjƒS(N(RR t DataFrame(tobj((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt is_data_framescCsddlm}t||ƒS(Niÿÿÿÿ(t DesignMatrix(tpatsyRR(RR((s5lib/python2.7/site-packages/statsmodels/tools/data.pytis_design_matrixscCs"t|tjƒo!|jjdk S(N(RtnptndarraytdtypetnamesR (R((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_structured_ndarray scCsgt|tjƒrŠt|ƒrH|dkr9|jj}nt|ƒ}n|}|dkrágt|j dƒD]}d|^qn}qánWt |ƒrÀ|j ƒ}|j }|j }|j}n!tdjdt|ƒjƒƒ‚t|tƒsÿt|ƒ}nt|ƒ|j dkr'tdƒ‚n|dk rZt|ƒt|ƒkrZtdƒ‚n|||fS(sC Convert passed data structure to form required by estimation classes Parameters ---------- data : ndarray-like colnames : sequence or None May be part of data structure rownames : sequence or None Returns ------- (values, colnames, rownames) : (homogeneous ndarray, list) isY_%dsCannot handle input type {typ}ttyps;length of colnames does not match number of columns in datas8length of rownames does not match number of rows in dataN(RRRRR RRtstruct_to_ndarrayRtshapeRtdropnatvaluestcolumnsRt TypeErrorRttypet__name__tlisttlenR(tdatatcolnamestrownamesR!ti((s5lib/python2.7/site-packages/statsmodels/tools/data.pytinterpret_data$s,   0      $cCs+|jtt|jjƒffdtjƒS(NR$(tviewtfloatR'RRRR(tarr((s5lib/python2.7/site-packages/statsmodels/tools/data.pyRVscCs4t|ƒtjko3t|ƒtjkp3|dkS(N(R$RRR (tendogtexog((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_using_ndarray_typeZscCs.t|tjƒo-t|tjƒp-|dkS(N(RRRR (R0R1((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_using_ndarray_scCs,ddlm}t||ƒp+t||ƒS(Niÿÿÿÿ(t data_klasses(tstatsmodels.compat.pandasR4R(R0R1tklasses((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_using_pandasdscCs5y&tj|ƒ}tj|ƒ}tSWntSXdS(N(RtasarraytTruetFalse(R0R1((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_array_likeis cCs"t|ƒo!t|ƒp!|dkS(N(RR (R0R1((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt_is_using_patsyrs cCst|tjjƒS(s, Returns true if data is a recarray (RRtcoretrecarray(R(((s5lib/python2.7/site-packages/statsmodels/tools/data.pyt _is_recarrayxs(t__doc__tstatsmodels.compat.pythonRtnumpyRRR RRRRR R,RR2R3R7R;R<R?(((s5lib/python2.7/site-packages/statsmodels/tools/data.pyts      2