ó áp7]c @s®ddlmZmZmZmZmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlZddlZddlZddlmZddlmZmZdefd„ƒYZd „Zid d 6d d 6dd6dd6dd6dd6dd6Zd„Zdddeedd„Z ded„Z!dd„Z"d„Z#dd&edd'ed„Z$dd „Z%d!deed"dd"d#d#d$„ Z&ddd%„Z'dS((iÿÿÿÿ( tlrangetiterkeyst iteritemstlziptreducet itervaluestzipt string_typestrange(t OrderedDictNi(t SimpleTable(t fmt_latextfmt_txttSummarycBs­eZd„Zd„Zd„Zd„Zeeddd„Zddd„Zdd dd „Z d „Z ddd „Z d ddddd„Z d„Zd„Zd„ZRS(cCs1g|_g|_g|_d|_t|_dS(N(ttablestsettingst extra_txttNonettitletFalset _merge_latex(tself((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt__init__s     cCs |jƒS(N(tas_text(R((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt__str__scCs"tt|ƒƒd|jƒdS(Ns """ s """(tstrttypeR(R((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt__repr__scCs |jƒS(s$Display as HTML in IPython notebook.(tas_html(R((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt _repr_html_ss%.4ftrcCsFi|d6|d6|d6|d6}|jj|ƒ|jj|ƒdS(s§Add the contents of a DataFrame to summary table Parameters ---------- df : DataFrame header: bool Reproduce the DataFrame column labels in summary table index: bool Reproduce the DataFrame row labels in summary table float_format: string Formatting to float data columns align : string Data alignment (l/c/r) tindextheadert float_formattalignN(RtappendR(RtdfRR R!R"R((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pytadd_df#sc Cs8tj|ƒ}|j|dtdtd|d|ƒdS(s Add the contents of a Numpy array to summary table Parameters ---------- array : numpy array (2D) float_format: string Formatting to array if type is float align : string Data alignment (l/c/r) RR R!R"N(tpdt DataFrameR%R(RtarrayR"R!ttable((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt add_array9s itlc Csýgt|ƒD]}t||ƒ^q }gt|ƒD]}t||ƒ^q5}tjt||ƒƒ}|jd|dkrÂ||jd|} tj|tj| ddggƒgƒ}ntj||ƒ}t d„|ƒ}|j |d|ƒdS(sAdd the contents of a Dict to summary table Parameters ---------- d : dict Keys and values are automatically coerced to strings with str(). Users are encouraged to format them before using add_dict. ncols: int Number of columns of the output table align : string Data alignment (l/c/r) itcSstj||gƒS(N(tnpthstack(txty((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt`R,R"N( Rt _formatterRR-R(RtshapetvstacktsplitRR*( RtdtncolsR"R!R/tkeystvalstdatatpad((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pytadd_dictIs((.cCs|jj|ƒdS(s—Append a note to the bottom of the summary table. In ASCII tables, the note will be wrapped to table width. Notes are not indendented. N(RR#(Rtstring((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pytadd_textcscCskt|tƒr||_nLy9|jjj}|tkrFt|}nd||_Wnd|_nXdS(sInsert a title on top of the summary table. If a string is provided in the title argument, that string is printed. If no title string is provided but a results instance is provided, statsmodels attempts to construct a useful title automatically. s Results: R,N(t isinstanceRRtmodelt __class__t__name__t _model_types(RRtresultsR@((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt add_titleis   gš™™™™™©?c Cs˜t|d|d|jƒ}t|ƒ}|d k r?||_n|d k rX||dRRERNRRRe(((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyR s         % c Cst||ƒ}g|D]}|jƒ^q}g|D]}t|jƒdƒ^q5}t|ƒ}g}g}x€tt|ƒƒD]l} t|| jdddƒ} t||| | ƒ} |j| ƒ|| | | } |j|| ƒq‚W||t|ƒfS(sôCompare width of ascii tables in a list and calculate padding values. We add space to each col_sep to get us as close as possible to the width of the largest table. Then, we add a few spaces to the first column to pad the rest. ii( RRRRTt splitlinestmaxRR3RUR#( RRR\R/R]tlengthtlen_maxtpad_sepRYtitnsepR;tlen_new((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyRQçs)  sOrdinary least squarestOLSsGeneralized least squarestGLSs$Generalized least squares with AR(p)tGLSARsWeighted least squarestWLSsRobust linear modeltRLMsNegative binomial modeltNBinsGeneralized linear modeltGLMcCsîd„}tƒ}d„|d|t|t[iit]tzsP>|z|(R?ttupletparamstbsettvaluestpvaluestconf_intR-R(tTR.R&R'RtcolumnsR@R:t param_namesRR›t exog_names( RDRKRJRFRGt skip_headerR!R¦R§R¨R©RªR:((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyRH<s,     2 /  csÊt|ƒ}x5|jd D]&}||j‡fd†ƒ||(RDR!t model_namesR¸R¾tregressor_ordert drop_omittedR/tcolstcolnamesRmtmergtsummtvarnamestorderedt unorderedtordertfR»R$tnameRMtdattsmry((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyt summary_col¶sZ" (& %, #&$7+<  (   cCs.y||}Wnt|ƒ}nX|jƒS(N(Rtstrip(telementR!R_((s9lib/python2.7/site-packages/statsmodels/iolib/summary2.pyR2s Rt-ic  s˜|jƒ} | j‡fd†ƒ} |rUg| jjƒD]} t| ƒ^q:} nd} |ršg| jjƒD] } t| ƒt| ƒd^qq} nYg| jdd…dfD] } t| ƒt| ƒd^q´| jdd…dfs<@     Ö    1 <  a