B Z`R @sddlmZmZmZmZmZmZmZmZm Z ddl m Z ddl Z ddl Z ddlZddlZddlZddlmZddlmZmZGdddeZd d Zd d d dddddZddZd,ddZd-ddZd.ddZdd Z d/d"d#Z!d0d$d%Z"d1d(d)Z#d2d*d+Z$dS)3) lrangeiterkeys iteritemslzipreduce itervalueszip string_typesrange) OrderedDictN) SimpleTable) fmt_latexfmt_txtc@s~eZdZddZddZddZddZd"d d Zd#ddZd$ddZ ddZ d%ddZ d&ddZ ddZ ddZd d!ZdS)'SummarycCs"g|_g|_g|_d|_d|_dS)NF)tablessettings extra_txttitle _merge_latex)selfr9lib/python3.7/site-packages/statsmodels/iolib/summary2.py__init__s zSummary.__init__cCs|S)N)as_text)rrrr__str__szSummary.__str__cCstt|d|dS)Nz """ z """)strtyper)rrrr__repr__szSummary.__repr__cCs|S)z$Display as HTML in IPython notebook.)as_html)rrrr _repr_html_szSummary._repr_html_T%.4frcCs*||||d}|j||j|dS)aAdd 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) )indexheader float_formatalignN)rappendr)rdfr#r$r%r&rrrradd_df#s  zSummary.add_dfcCs"t|}|j|dd||ddS)a 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) F)r#r$r%r&N)pd DataFramer))rarrayr&r%tablerrr add_array9s  zSummary.add_arraylc sfddt|D}fddt|D}tt||}|jd|dkr~||jd|}t|t|ddggg}t||}tdd|}|j ||dd S) aAdd 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) csg|]}t|qSr) _formatter).0x)r%rr Wsz$Summary.add_dict..csg|]}t|qSr)r1)r2r3)r%rrr4XsrcSst||gS)N)nphstack)r3yrrr`sz"Summary.add_dict..)r&N) rrr6r,rshapevstacksplitrr.) rdZncolsr&r%keysZvalsdatapadr)r%radd_dictIs zSummary.add_dictcCs|j|dS)zAppend 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')rstringrrradd_textcszSummary.add_textNcCsRt|tr||_n. rN r/r5csg|]}t|qSr)textwrapZwrap)r2r3)widestrrr4scSsg|]}d|qS)rV)join)r2r3rrrr4s) rrrr_measure_tables_simple_tablesrZr<r'lenint) rrrrrpad_col pad_indexZ rule_equalZ rule_dash simple_tablestabtxtoutr)rYrrs0     " zSummary.as_textcCs8|j}|j}|j}t||}dd|D}d|}|S)z$Generate HTML Summary Table cSsg|] }|qSr)r)r2r3rrrr4sz#Summary.as_html..rV)rrrr\rZ)rrrrrarbrrrrs  zSummary.as_htmlcCs|j}|j}|j}|dk r(d|d}nd}t||}dd|D}d|}d}|jrft|d |}d ||d f}d |}|S) z%Generate LaTeX Summary Table Nz \caption{}z \caption{}cSsg|] }|qSr)Zas_latex_tabular)r2r3rrrr4sz$Summary.as_latex..z \hline z8\\hline\n\\hline\n\\end{tabular}\n\\begin{tabular}{.*}\nz\midrule \midrule z \begin{table}z \end{table}rV)rrrr\rZrresub)rrrrrarbZ to_replacerdrrras_latexs    zSummary.as_latex)TTr!r")r"r!)r/r0r!)NN)rKr!NNN)rG __module__ __qualname__rrrr r)r.rArCrJrSrrrhrrrrrs     'rc Cst||}dd|D}dd|D}t|}g}g}xhtt|D]X}t||jddd} t|||| } || ||| | } ||| qDW||t|fS)zCompare 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. cSsg|] }|qSr)r)r2r3rrrr4sz#_measure_tables..cSsg|]}t|dqS)r)r] splitlines)r2r3rrrr4sr )r\maxr r]r:r^r') rrrarblengthZlen_maxZpad_sepr`iZnsepr@Zlen_newrrrr[s  r[zOrdinary least squareszGeneralized least squaresz$Generalized least squares with AR(p)zWeighted least squareszRobust linear modelzNegative binomial modelzGeneralized linear model)ZOLSZGLSZGLSARZWLSZRLMZNBinZGLMc Csdd}t}dd|d<dd|d<dd|d <d d|d <||d <d d|d<dd|d<dd|d<dd|d<dd|d<dd|d<dd|d<dd|d<dd|d<dd|d <d!d|d"<d#d|d$<d%d|d&<d'd|d(<d)d|d*<d+d|d,<d-d|d.<d/d|d0<d1d|d2<d3d|d4<d5d|d6<d7d|d8<t}xDt|D]8\}}y||||<Wntttfk rYnXq^W|S)9z3Create a dict with information about the model c_stj}|dS)Nz%Y-%m-%d %H:%M)datetimenowZstrftime)argskwdsrprrrtime_nows zsummary_model..time_nowcSs |jjjS)N)rErFrG)r3rrrr9szsummary_model..zModel:cSs |jjjS)N)familyZ__classrG)r3rrrr9sz Model Family:cSs |jjjjS)N)rtlinkrFrG)r3rrrr9szLink Function:cSs|jjS)N)rE endog_names)r3rrrr9szDependent Variable:zDate:cSs d|jS)Nz%#6d)Znobs)r3rrrr9szNo. Observations:cSs d|jS)Nz%#6d)Zdf_model)r3rrrr9sz Df Model:cSs d|jS)Nz%#6d)Zdf_resid)r3rrrr9sz Df Residuals:cSs |jdS)NZ converged) mle_retvals)r3rrrr9sz Converged:cSs |jdS)NZ iterations)rw)r3rrrr9szNo. Iterations:cSs|jS)N)method)r3rrrr9szMethod:cSs |jdS)NZnorm) fit_options)r3rrrr9szNorm:cSs |jdS)NZ scale_est)ry)r3rrrr9 sz Scale Est.:cSs |jdS)NZcov)ry)r3rrrr9!sz Cov. Type:cSs d|jS)Nz%#8.3f)Zrsquared)r3rrrr9"sz R-squared:cSs d|jS)Nz%#8.3f)Z rsquared_adj)r3rrrr9#szAdj. R-squared:cSs d|jS)Nz%#8.3f)Z prsquared)r3rrrr9$szPseudo R-squared:cSs d|jS)Nz%8.4f)Zaic)r3rrrr9%szAIC:cSs d|jS)Nz%8.4f)Zbic)r3rrrr9&szBIC:cSs d|jS)Nz%#8.5g)Zllf)r3rrrr9'szLog-Likelihood:cSs d|jS)Nz%#8.5g)Zllnull)r3rrrr9(szLL-Null:cSs d|jS)Nz%#8.5g)Z llr_pvalue)r3rrrr9)sz LLR p-value:cSs d|jS)Nz%#8.5g)Zdeviance)r3rrrr9*sz Deviance:cSs d|jS)Nz%#6.3g)Z pearson_chi2)r3rrrr9+sz Pearson chi2:cSs d|jS)Nz%#8.4g)Zfvalue)r3rrrr9,sz F-statistic:cSs d|jS)Nz%#6.3g)Zf_pvalue)r3rrrr9-szProb (F-statistic):cSs d|jS)Nz%#8.5g)Zscale)r3rrrr9.szScale:)r rAttributeErrorKeyErrorNotImplementedError)rIrsrRrdkeyfuncrrrrO sH                           rO皙?TF%.4fc Cst|tr|\}}}} } } n"|j}|j}|j} |j} ||} t||| | gj } t | | g} t | } |rdddddt |dt d|ddg| _n.ddd d dt |dt d|ddg| _|sy|jjj| _Wn tk r|jj| _YnXn|| _| S) acreate a summary table of parameters from results instance 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. float_format : string float formatting options (e.g. ".3g") Returns ------- params_table : SimpleTable instance zCoef.zStd.Err.tzP>|t|[r/r ]zzP>|z|)rDtupleparamsbsetvaluespvaluesconf_intr6r,Tr7r*r+rcolumnsrEr?Z param_namesr#rzZ exog_names) rIrQrPrLrMZ skip_headerr%rrrrrr?rrrrN9s,   (&rNcs^t|}x0|jddD]}||fdd||<qWd|jdddfd|jdddf<|r"|jdddfd k}|j||jd fd |j||jd f<|jdddfd k}|j||jd fd |j||jd f<|jdddfd k}|j||jd fd |j||jd f<|jddddf}|}t|}t|j j g|_|S)z.(r )g?r*g?g{Gz?) rNrZapplyiloclocstackr*r+rrErv)resultr%starsZrescolidxr)r%r _col_paramsvs (((( rcCs|dkr i}g}g}xT|D]L}t||tr.qy||||Wn|dYnX||qWtjt|jj|i|d}|S)z!Stack model info in a column Nr5)r#)rDdictr'r*r+rrErv)r info_dictrdr#rnrrr _col_infos rcCshtt|t|kr|Sddlm}|t}g}x2|D]*}||d7<||d||q6W|S)Nr) defaultdictIrW)r]set collectionsrrr')Z list_of_namesrZ name_counterr$_namerrr _make_uniques  rrcs.t|ts|g}fdd|D}|r2t|}ntdd|D}x$tt|D]} || g|| _qRWdd} t| |} r| jd fddD} fddD} | tt | }d d}|t | _| ||} d d| jD| _|r| j} t t| jdd d k}t |d | jd| _r^fdd|D}ndd|D}x.t|tdd|DD]\}}|g|_qWdd} t| |}t t | |g}| j|_t | j |j |_|} | d } t}d|_|j| ddd|dr*|d|S)a Summarize multiple results instances side-by-side (coefs and SEs) Parameters ---------- results : statsmodels results instance or list of result instances float_format : string, optional float format for coefficients and standard errors Default : '%.4f' model_names : list of strings, optional Must have same length as the number of results. If the names are not unique, a roman number will be appended to all model names stars : bool print significance stars info_dict : dict dict of functions to be applied to results instances to retrieve model info. To use specific information for different models, add a (nested) info_dict with model name as the key. Example: `info_dict = {"N":..., "R2": ..., "OLS":{"R2":...}}` would only show `R2` for OLS regression models, but additionally `N` for all other results. Default : None (use the info_dict specified in result.default_model_infos, if this property exists) regressor_order : list of strings, optional list of names of the regressors in the desired order. All regressors not specified will be appended to the end of the list. drop_omitted : bool, optional Includes regressors that are not specified in regressor_order. If False, regressors not specified will be appended to end of the list. If True, only regressors in regressors_list will be included. csg|]}t|dqS))rr%)r)r2r3)r%rrrr4szsummary_col..cSsg|]}|jdqS)r)r)r2r3rrrr4scSs|j|ddddS)NouterT)how right_index left_index)merge)r3r8rrrr9s zsummary_col..rcsg|]}|kr|qSrr)r2r3)varnamesrrr4scsg|]}|dgkr|qS)r5r)r2r3)regressor_orderrrr4scSstdd|DgS)NcSsg|]}|d|dgqS)ZcoefZstder)r2r3rrrr4sz1summary_col....)sum)rrrrr9scSsg|]}|ddqS)Nr)r2r3rrrr4sr/r r5cs$g|]}t||jjjqSr)rgetrErFrG)r2r3)rrrr4scSsg|]}t|t|ddqS)Zdefault_model_infosN)rgetattr)r2r3rrrr4scSsg|]}|jdqS)r)r)r2r(rrrr4scSs|j|ddddS)NrT)rrr)r)r3r8rrrr9s Tr0)r$r&zStandard errors in parentheses.z* p<.1, ** p<.05, ***p<.01)rDlistrr r]rrr#Zget_level_valuestolistr6uniquer*ZreindexrZSeriesrr:whererr+r;ZIndexZfillnarrr)rC)rIr%Z model_namesrrrZ drop_omittedZcolsZcolnamesrnZmergZsummZorderedZ unorderedorderfrr(namerRdatZsmryr)r%rrrrr summary_colsZ"       "    rcCs*y ||}Wnt|}YnX|S)N)rstrip)elementr%rdrrrr1s  r1r"rUc s|} | fdd} |r4dd| jD} nd} |rVfdd| jD} n2fdd| jdddfD| jdddf<d} tt| | | t t d} || j d d <|| j d d <|| j d d <|| j d d <|| j d d<dt |d| j d d<| S)Ncs t|S)N)r1)r3)r%rrr9!sz$_df_to_simpletable..cSsg|] }t|qSr)r)r2r3rrrr4#sz&_df_to_simpletable..cs g|]}t|tdqS)rW)rr^)r2r3)r`rrr4'scs g|]}t|tdqS)rW)rr^)r2r3)r`rrr4)sr)headersstubsZltx_fmtZtxt_fmtZlatexZ data_alignsrctable_dec_abovetable_dec_belowheader_dec_belowrWr Zcolsep) copyZapplymaprrr#rr r6r,rrZoutput_formatsr^)r(r&r%r$r#rrrr_r`rrrstr)r%r`r_df_to_simpletables$. rc Csg}d}|dkrdgt|}|dkr4dgt|}x\t|D]P\}}||d}||d} ||d} |t|| || |||||dq>W|S)Nz%.4frr#r$r&)r&r%r$r#r_r`)r] enumerater'r) rrr_r`rar%rnvr#r$r&rrrr\6s     r\)NNrTFr)rT)N)rrFNrF)r) r"rTTrUNrUrr)NN)%Zstatsmodels.compat.pythonrrrrrrrr r rr rorfrXZnumpyr6Zpandasr*r-r Ztableformattingrrobjectrr[rHrOrNrrrrr1rr\rrrrs>,  Y, <    a