p7]c@szddlZddlZddlmZmZmZmZmZddlZ ddl m Z ddl m Z mZmZmZddlmZddZd d Zddd d d ddZdddZddddddZddd eeddZddd edZdddddZddd eeedZedZd dZ de!fdYZ"dS(iN(trangetlrangetlmaptlzipt zip_longest(t SimpleTable(tgen_fmttfmt_2t fmt_paramst fmt_2colsi(t _model_typesicCs|dkrCt|dks0t|dkr8d|Sd|Sn[|dkrt|dksst|dkr{d|Sd|Sntd jd |dS( Nig@g-C6?s%9.3gs%9.3fis%10.4gs%10.4fs1`prec` argument must be either 3 or 4, not {prec}tprec(tabst ValueErrortformat(txR ((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytforg s $  $  icCsCtj|r|dddS|d|kr7d|Sd|SdS(smconvert number to string with either integer of float formatting This is used internally for nobs and degrees of freedom which are usually integers but can be float in some cases. Parameters ---------- x : int or float width : int only used if x is nan Returns ------- str : str number as formatted string it tNaNis%#6ds%#8.2fN(tnptisnan(Rtwidth((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytd_or_fs ig?ttextc' sz|dkr"t|jjj}n|d%k rjt|t|jkrjtdjt|jnt |||\}}t j }t j d|g}t j d|} |jjj} |j } |j} |j} d| gfd| gfd|fd| gfg}t|\}}|}d%}t|||d |d t}d&}| g|| g| gf}t|||d |d t}|j||i|jd6|jd6|jd6|jd6|jd6|jd6}i|jd6|jd6|jd6|jd6|jd6|jd6}td|dd}idddd|dgd6dddd|dgd6dddd|dgd6dddd|dgd6dddd|dgd6dddd|dgd6}|}|j}|j|}|j}tt|}|| } || }!tg|D]}"d||"^qGg|D]}"d ||"^qeg|D]}"d | |"^qg|D]}"d |!|"^qg|D]}"d!t||"^q}#t|#|| |d d%d tfd"}$fd#}%i|$d6|%d6}&|d$krvy|&| SWqvtk rr|&dSXnd%S('s Parameters ---------- yname : string optional, Default is `Y` xname : list of strings optional, Default is `X.#` for # in p the number of regressors Confidance interval : (0,1) not implimented title : string optional, Defualt is 'Generalized linear model' returns : string 'text', 'table', 'csv', 'latex', 'html' Returns ------- Default : returns='print' Prints the summarirized results Option : returns='text' Prints the summarirized results Option : returns='table' SimpleTable instance : summarizing the fit of a linear model. Option : returns='csv' returns a string of csv of the results, to import into a spreadsheet Option : returns='latex' Not implimented yet Option : returns='HTML' Not implimented yet Examples (needs updating) -------- >>> import statsmodels as sm >>> data = sm.datasets.longley.load(as_pandas=False) >>> data.exog = sm.add_constant(data.exog) >>> ols_results = sm.OLS(data.endog, data.exog).results >>> print ols_results.summary() ... Notes ----- conf_int calculated from normal dist. isaUser supplied xnames must have the same number of entries as the number of model parameters ({0})s%H:%M:%Ss %a, %d %b %Ys Model type:sDate:sDependent Variable:sdf modelttitlettxt_fmtsMethod:sTime:sNumber of Obs:sdf residtOLStGLStGLSARtWLStRLMtGLMiidt%tcoefsstd errttsP>|t|s Conf. IntervaltzsP>|z|s%#6.4gs%#6.4fs (%#5g, %#5g)cstdt}|S(s4 print summary table for ols models s (tstr(ttable(t general_tabletparameter_table(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt ols_printerscstdt}|S(Ns (R$(R%(R&R'(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt glm_printerstprintN(sMethod:sTime:sNumber of Obs:sdf resid(R tmodelt __class__t__name__tNonetlentparamsR Rt _getnamesttimet localtimetstrftimetnobstdf_modeltdf_residRRRt extend_rightR"tpvaluesR$tconf_inttbseRRttupleRtKeyError('tselftynametxnameRtalphatreturnst model_infottime_nowt time_of_daytdatet modeltypeR5R6R7tgen_lefttgen_stubs_leftt gen_data_leftt gen_titlet gen_headertgen_table_lefttgen_stubs_righttgen_data_righttgen_table_rightttstatst prob_statstalpt param_headert params_stubsR0R:tstd_errtexog_lenttstatt prob_stattit params_dataR(R)tprinters((R&R's8lib/python2.7/site-packages/statsmodels/iolib/summary.pytsummary7s7 '                          !*   cCs|dkr?t|jdddk r6|jj}q?d}n|dkrt|jdddk ru|jj}qgtt|jD]}d|^q}n||fS(s0extract names from model or construct names t endog_namestyt exog_namessvar_%dN(R.tgetattrR+R^R`RR/R0(R>R?R@RZ((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyR1s   /c s||}}tj}tjd|gtjd|tdd|\}tdfdfdfdfd fd fd fd fd fdfdfdfdfdfdfdfdfdfdfdfg } |d#kr@jjjd}n|d#krd$d%d&d'd(d)g}yj } |j d*WnnXg}n|} d#} g} xC|D];\}}|d#kr| |}n| j ||fqW| }|rJg}xC|D];\}}|d#kr*| |}n|j ||fqW|}ng||D]\}}|d#krU|^qU}|gkst ||rot |t |kr|d+gt |t |7}n<t |t |kr|d,gt |t |7}ng|D] \}}dd ||f^q}t |\}}t|| |d!| d"t}ng}t |\}}t|| |d!| d"t}|j||}|S(-sgenerate top table(s) TODO: this still uses predefined model_methods ? allow gleft, gright to be 1 element tuples instead of filling with None? s%H:%M:%Ss %a, %d %b %YR?R@sDependent Variable:csgS(N(((R?(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyttsDep. Variable:csgS(N(((R?(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRbRcsModel:csjjjgS(N(R+R,R-((tresults(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRbRcsDate:csgS(N(((RF(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb RcsTime:csS(N(((RE(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb!RcsNumber of Obs:cs jgS(N(R5((Rd(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb"RcsNo. Observations:cstjgS(N(RR5((Rd(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb#Rcs Df Model:cstjgS(N(RR6((Rd(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb$Rcs Df Residuals:cstjgS(N(RR7((Rd(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb%RcsLog-Likelihood:csdjgS(Ns%#8.5g(tllf((Rd(s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRb&RcsRegression Resultss Model type:s Df model:s Df resid:sLog-LikelihoodRs%-21ss RRN(sDep. Variable:N(s Model type:N(sDate:N(sNo. Observations:N(s Df model:N(s Df resid:N(sLog-LikelihoodN(RR(RR(R2R3R4R1tdictR.R+R,R-RetappendtAssertionErrorR/RRR R8(RdRtglefttgrightR?R@RHt gen_rightRDt default_itemsReRKRLt gen_left_titemtvaluet gen_right_tktvtmissing_valuesRNRORPRIRJRMR&((RFRdRER?s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt summary_tops          /$$-   c Cs=t|tr*|\}}}} } } n3|j}|j}|j} |j} |j|} |rdddddt|dtd|ddg} n8ddd d dt|dtd|ddg} |rd} nt |d |d |\} }t |t |kr*t d n|}t t |}t g|D]}t||dd^qLg|D]}t||^qrg|D]}t| |^qg|D]}d| |^qg|D]}t| |df^qg|D]}t| |df^q}t|| |d|dt}|S(s,create a summary table for the parameters Parameters ---------- res : results instance some required information is directly taken from the result instance yname : string or None optional name for the endogenous variable, default is "y" xname : list of strings or None optional names for the exogenous variables, default is "var_xx" alpha : float significance level for the confidence intervals use_t : bool indicator whether the p-values are based on the Student-t distribution (if True) or on the normal distribution (if False) skip_headers : bool If false (default), then the header row is added. If true, then no header row is added. Returns ------- params_table : SimpleTable instance R!sstd errR"sP>|t|t[iit]R#sP>|z|R?R@s-xnames and params do not have the same lengthR is%#6.3fiRRN(t isinstanceR<R0R;ttvaluesR9R:R$R.R1R/R RRRRR(RdR?R@RAtuse_tt skip_headerRR0RVRxR9R:RTt_RUtexog_idxRZR[R'((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytsummary_params{s>      / , )  &, cCst|tr*|\}}}}}} n3|j}|j}|j}|j}|j|} |r~ddddddg} nddddddg} t|d |d |\} }d d lm } t j ||||| f} | | d | d|S(s,create a summary table for the parameters Parameters ---------- res : results instance some required information is directly taken from the result instance yname : string or None optional name for the endogenous variable, default is "y" xname : list of strings or None optional names for the exogenous variables, default is "var_xx" alpha : float significance level for the confidence intervals use_t : bool indicator whether the p-values are based on the Student-t distribution (if True) or on the normal distribution (if False) skip_headers : bool If false (default), then the header row is added. If true, then no header row is added. Returns ------- params_table : SimpleTable instance R!sstd errR"sP>|t|sConf. Int. LowsConf. Int. Upp.R#sP>|z|R?R@i(t DataFrametcolumnstindex( RwR<R0R;RxR9R:R1tpandasR~Rt column_stack(RdR?R@RARyR0RVRxR9R:RTR{R~R%((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytsummary_params_frames        c Cs|dkr?gtj|jjdD]}d|^q&}n|dkrzgtt|jD]}d|^qa}ng|jD]+}g|D]}t|dd^q^q}|rg|D]Y} gt || D]=} g| D]*} ddt| ddj d ^q^q^q} t || } g| D]}|D] }|^qAq7} t |d gt|gt|}g|D]}|D] }|^qq}n |} |}t j t}d g|jjd|d stlatexiithtmlcss|]}|jVqdS(N(tas_html(RR%((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pys ss3available output formats are text, csv, latex, html(tjoinRR$RRRtas_latex_tabularR (Rt return_fmttstrdropR%tpart((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytsummary_returns   +     tSummarycBseZdZdZdZdZdZd d d d d dZd d de dZ dZ d Z d Z d Zd ZRS(sclass to hold tables for result summary presentation Construction does not take any parameters. Tables and text can be added with the `add_` methods. Attributes ---------- tables : list of tables Contains the list of SimpleTable instances, horizontally concatenated tables are not saved separately. extra_txt : string extra lines that are added to the text output, used for warnings and explanations. cCsg|_d|_dS(N(RR.t extra_txt(R>((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt__init__s cCs |jS(N(tas_text(R>((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt__str__scCs"tt|d|jdS(Ns """ s """(R$ttypeR(R>((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt__repr__scCs |jS(s$Display as HTML in IPython notebook.(R(R>((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt _repr_html_sc Cs>t|d|d|d|d|d|}|jj|dS(sradd a double table, 2 tables with one column merged horizontally Parameters ---------- res : results instance some required information is directly taken from the result instance title : string or None if None, then a default title is used. gleft : list of tuples elements for the left table, tuples are (name, value) pairs If gleft is None, then a default table is created gright : list of tuples or None elements for the right table, tuples are (name, value) pairs yname : string or None optional name for the endogenous variable, default is "y" xname : list of strings or None optional names for the exogenous variables, default is "var_xx" Returns ------- None : tables are attached RRiRjR?R@N(RtRRg(R>RRRiRjR?R@R%((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytadd_table_2colssg?c Cs|jjdkr9t|d|d|d|d|}nK|jjdkrxt|d|d|d|d|\}}n td |jj|d S( screate and add a table for the parameter estimates Parameters ---------- res : results instance some required information is directly taken from the result instance yname : string or None optional name for the endogenous variable, default is "y" xname : list of strings or None optional names for the exogenous variables, default is "var_xx" alpha : float significance level for the confidence intervals use_t : bool indicator whether the p-values are based on the Student-t distribution (if True) or on the normal distribution (if False) Returns ------- None : table is attached iR?R@RARyiR^R`sparams has to be 1d or 2dN(R0RR}RR RRg(R>RR?R@RARyR%R{((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytadd_table_paramss  cCsdj||_dS(sadd additional text that will be added at the end in text format Parameters ---------- etext : list[str] string with lines that are added to the text output. s N(RR(R>tetext((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyt add_extra_txt8s cCs<t|jdd}|jdk r8|d|j}n|S(sreturn tables as string Returns ------- txt : string summary tables and extra text as one string RRs N(RRRR.(R>R((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRCs cCsHt|jdd}|jdk rD|d|jjdd}n|S(sYreturn tables as string Returns ------- latex : string summary tables and extra text as string of Latex Notes ----- This currently merges tables with different number of columns. It is recommended to use `as_latex_tabular` directly on the individual tables. RRs s s \newline N(RRRR.treplace(R>R((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pytas_latexQs cCs<t|jdd}|jdk r8|d|j}n|S(sreturn tables as string Returns ------- csv : string concatenated summary tables in comma delimited format RRs N(RRRR.(R>R((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRes cCsHt|jdd}|jdk rD|d|jjdd}n|S(sreturn tables as string Returns ------- html : string concatenated summary tables in HTML format RRs

s s
N(RRRR.R(R>R((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRss  N(R-t __module__t__doc__RRRRR.RtTrueRRRRRR(((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyRs       "   (#RR2tstatsmodels.compat.pythonRRRRRtnumpyRtstatsmodels.iolib.tableRt!statsmodels.iolib.tableformattingRRRR tsummary2R RRR.R]R1RtRRR}RRRRRtobjectR(((s8lib/python2.7/site-packages/statsmodels/iolib/summary.pyts.  ( "   s P  ;  < O -