o 0GfW@sddlmZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl mZddlmZmZGdd d Zd d Zd d ddddddZddZ  d-ddZd.ddZd/ddZd d!Z   d0d#d$Zd1d%d&Z   d2d)d*Zd3d+d,ZdS)4) FUTURE_STACK)lzipN)reduce) SimpleTable) fmt_latexfmt_txtc@seZdZddZddZddZddZd d Z d%ddZd&ddZ d'ddZ ddZ d(ddZ  d)ddZ ddZd d!Zd*d#d$ZdS)+SummarycCs"g|_g|_g|_d|_d|_dS)NF)tablessettings extra_txttitle _merge_latexselfr:lib/python3.10/site-packages/statsmodels/iolib/summary2.py__init__s  zSummary.__init__cC|SNas_textrrrr__str__szSummary.__str__cCstt|d|dS)Nz """ z """)strtyperrrrr__repr__szSummary.__repr__cCr)z$Display as HTML in IPython notebook.as_htmlrrrr _repr_html_zSummary._repr_html_cCr)z9Display as LaTeX when converting IPython notebook to PDF.)as_latexrrrr _repr_latex_"rzSummary._repr_latex_T%.4frcCs*||||d}|j||j|dS)a 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 : str Formatting to float data columns align : str Data alignment (l/c/r) indexheader float_formatalignN)r appendr )rdfr%r&r'r(r rrradd_df&s  zSummary.add_dfcCs"t|}|j|dd||ddS)aAdd the contents of a Numpy array to summary table Parameters ---------- array : numpy array (2D) float_format : str Formatting to array if type is float align : str Data alignment (l/c/r) Fr$N)pd DataFramer+)rarrayr(r'tablerrr add_array=s  zSummary.add_arraylc sfdd|D}fdd|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 : str Data alignment (l/c/r) float_format : str Formatting to float data columns cg|]}t|qSr _formatter.0xr'rr ]z$Summary.add_dict..cr3rr4r6r9rrr:^r;rcSst||gSr)nphstackr8yrrrfsz"Summary.add_dict..r(N) keysvaluesr=r.rshapevstacksplitrr0) rdZncolsr(r'rCZvalsdatapadrr9radd_dictMs 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 indented. N)r r))rstringrrradd_textiszSummary.add_textNcCsNt|tr ||_dS|dur"|jjj}|tvrt|}d||_dSd|_dS)aInsert 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. Nz Results: r<) isinstancerr model __class____name__ _model_types)rr resultsrOrrr add_titleos    zSummary.add_title皙?c Csdt|||jd}t|}|dur||_|dur||d<|j|dd|j||d|j||ddS)aLTry to construct a basic summary instance. Parameters ---------- results : Model results instance alpha : float significance level for the confidence intervals (optional) float_format: str Float formatting for summary of parameters (optional) title : str Title of the summary table (optional) xname : list[str] of length equal to the number of parameters Names of the independent variables (optional) yname : str Name of the dependent variable (optional) )alphause_tNDependent Variable:r2rBr9)r rS)summary_paramsrW summary_modelr%rKr+rT) rrSrVr'r xnameynameZparaminforrradd_baseszSummary.add_basec s|j}|j}|j}|j}t||\}}d}t||||}dd|D} d| } | d} || d<| |d| } |dur[|}t |krZdt dt |d|}nd }fd d|D} d d| D} d| } d|| | g} | S) z%Generate ASCII Summary Table =cSg|]}|qSrrr6rrrr:z#Summary.as_text.. rN r1r<csg|]}t|qSr)textwrapwrapr6Zwidestrrr:cSsg|]}d|qS)rb)joinr6rrrr:r;) r r r r _measure_tables_simple_tablesrhrGr)lenint) rr r r r pad_col pad_indexZ rule_equal simple_tablestabtxtoutrrfrrs0       zSummary.as_textcCsZ|j}|j}t||}dd|D}d|}dd|jD}d|}d||g}|S)z$Generate HTML Summary Table cSr`rrr6rrrr:raz#Summary.as_html..rbcSsg|]}|ddqS)rb
)replace)r7strrrr:rgrs)r r rjrhr )rr r rorpZtemp_txtrqrrrrrrs   zSummary.as_htmlr<c Cs|j}|j}|j}|durd|d}nd}d|d}t||}dd|D}d|}d }|jr9t|d |}d |||d f}d |}d|j} |d| } | S)zGenerate LaTeX Summary Table Parameters ---------- label : str Label of the summary table that can be referenced in a latex document (optional) Nz \caption{}z \caption{}z\label{cSr`r)Zas_latex_tabularr6rrrr:raz$Summary.as_latex..z z8\\hline\n\\hline\n\\end{tabular}\n\\begin{tabular}{.*}\nz \\midrule\nz \begin{table}z \end{table}rbz \newline z \bigskip ) r r r rjrhrresubr ) rZlabelr r r rorpZ to_replaceZ non_captionedrqrrrrrr s$       zSummary.as_latex)TTr"r#)r#r")r1r2r"NN)rUr"NNN)r<)rQ __module__ __qualname__rrrrr!r+r0rKrMrTr^rrr rrrrr s&    %r c Cst||}dd|D}dd|D}t|}g}g}tt|D],}t||jddd} t|||| } || ||| | } ||| q!||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. cSr`rrr6rrrr:raz#_measure_tables..cSsg|] }t|dqSr)rk splitlinesr6rrrr:sr)rjmaxrangerkrErlr)) r r rorplengthZlen_maxZpad_seprniZnseprJZlen_newrrrris  rizOrdinary least squareszGeneralized least squaresz$Generalized least squares with AR(p)zWeighted least squareszRobust linear modelzNegative binomial modelzGeneralized linear model)ZOLSZGLSZGLSARZWLSZRLMZNBinZGLMc Csdd}i}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<|jr]dnd }d!d|d"|d#<d$d|d%|d#<d&d|d'<d(d|d)<d*d|d+<d,d|d-<d.d|d/<d0d|d1<d2d|d3<d4d|d5<d6d|d7<d8d|d9<d:d|d;<i}|D]\}}z||||<WqtttfyYqw|S).time_nowcS |jjjSr)rOrPrQr8rrrrA* zsummary_model..zModel:cSrr)familyZ__classrQrrrrrA+rz Model Family:cSs |jjjjSr)rlinkrPrQrrrrrA,s zLink Function:cSs|jjSr)rO endog_namesrrrrrA-rXzDate:cS d|jSNz%#6d)ZnobsrrrrrA/rzNo. Observations:cSrr)Zdf_modelrrrrrA0rz Df Model:cSrr)Zdf_residrrrrrA1rz Df Residuals:cS |jdS)NZ convergedZ mle_retvalsrrrrrA2rz Converged:cSr)NZ iterationsrrrrrrA3rzNo. Iterations:cSs|jSr)methodrrrrrA4szMethod:cSr)NZnormZ fit_optionsrrrrrA5rzNorm:cSr)NZ scale_estrrrrrrA6rz Scale Est.:cSr)NZcovrrrrrrA7rz Cov. Type:r<z (uncentered)cSrNz%#8.3f)rsquaredrrrrrA:r R-squared:cSrr) rsquared_adjrrrrrA;rzAdj. R-squaredcSrr)Z prsquaredrrrrrA<rzPseudo R-squared:cSrNz%8.4f)ZaicrrrrrA=rzAIC:cSrr)ZbicrrrrrA>rzBIC:cSrNz%#8.5g)ZllfrrrrrA?rzLog-Likelihood:cSrr)ZllnullrrrrrA@rzLL-Null:cSrr)Z llr_pvaluerrrrrAArz LLR p-value:cSrr)ZdeviancerrrrrABrz Deviance:cSrNz%#6.3g)Z pearson_chi2rrrrrACrz Pearson chi2:cSr)Nz%#8.4g)ZfvaluerrrrrADrz F-statistic:cSrr)Zf_pvaluerrrrrAErzProb (F-statistic):cSrr)ZscalerrrrrAFrzScale:)Z k_constantitemsAttributeErrorKeyErrorNotImplementedError)rSrr]Z rsquared_typerrkeyfuncrrrrZ sL                        rZrUTFr"c Cs t|tr|\}}}} } } n|j}|j}|j} |j} ||} t||| | gj } t | | g} t | } |rOdddddt |dt d|ddg| _nddd d dt |dt d|ddg| _|sz |jjj| _W| Sty|jj| _Y| Sw|| _| S) axcreate a summary table of parameters from results instance Parameters ---------- res : results instance some required information is directly taken from the result instance yname : {str, None} optional name for the endogenous variable, default is "y" xname : {list[str], 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_header : bool If false (default), then the header row is added. If true, then no header row is added. float_format : str float formatting options (e.g. ".3g") Returns ------- params_table : SimpleTable instance zCoef.zStd.Err.tzP>|t|[r1r]zzP>|z|)rNtupleparamsbsetvaluespvaluesconf_intr=r.Tr>r,r-rcolumnsrOrIZ param_namesr%rZ exog_names) rSr\r[rVrWZ skip_headerr'rrrrrrIrrrrYRs6      rYc st|}|jddD]}||fdd||<q d|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}|jdit}|rt|dtj }t|dtj }t ||d} | r| fdd} t j|| gd d}t |}t|jjg|_|S)z.(r)g?r*rUg{Gz?rr)rr<)zR-squared Adj.r<crrrrr9rrrAr)Zaxisr)rYrZapplyiloclocstackrgetattrr=nanr,SeriesZnotnullanyconcatr-rrOr) resultr'stars include_r2rescolidxrrZr2rr9r _col_paramss2((((  rc Cs|duri}g}g}|D])}t||trq z ||||Wnty/|dYnw||q tjt|jj|i|d}|S)z!Stack model info in a column Nr<)r%) rNdictr)rr,r-rrOr)r info_dictrrr%rrrr _col_infos  rcCsdtt|t|kr |Sddlm}|t}g}|D]}||d7<||d||q|S)Nr) defaultdictIrc)rkset collectionsrrr))Z list_of_namesrZ name_counterr&_namerrr _make_uniques rrcst|ts|g}fdd|D}|rt|} n tdd|D} tt|D] } | | g|| _q)dd} t|dj} |ddD]} | jD] }|| vrU| |qJqEd D]}|| vrl| || t| |qYt | |}| | }r|j d t}|j|d kj fd dD}fd dD}||fd d|j dD}|7|r|D]}|q|j dd}g}|j d} td| jdd D]+} || | | d| jdkr| | | | dkr|dq|| | dq||_rfdd|D}ndd|D}t|tdd|DD] \}}|g|_q,t | |}tt||g}|j|_t|j |j |_|}|d}t}d|_|j|ddd|dry|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 : str, optional float format for coefficients and standard errors Default : '%.4f' model_names : list[str], 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, default None 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":lambda x:(x.nobs), "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[str], 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 regressor_order will be included. include_r2 : bool, optional Includes R2 and adjusted R2 in the summary table. csg|] }t|dqS))rr'r)rr6)r'rrrrr:s  zsummary_col..cSg|]}|jdqSr|rr6rrrr:r;cSs|j|ddddS)NZouterT)ZhowZ right_indexZ left_index)Zmerger?rrrmerg szsummary_col..mergrrNrr1csg|]}|vr|qSrrr6)varnamesrrr: rgcg|]}|vr|qSrrr6)regressor_orderrrr:!rgcrrrr6) new_orderrrr:#s)levelr<cs$g|]}t||jjjqSr)rgetrOrPrQr6)rrrr:7s  cSsg|] }t|t|ddqS)Zdefault_model_infosN)rrr6rrrr::scSrr|r)r7r*rrrr:=r;Tr2)r&r(zStandard errors in parentheses.z* p<.1, ** p<.05, ***p<.01)rNlistrrrkrr%r)removeinsertrZreindexZget_level_valuestolistr,rZ value_countsrrEzipr-r=rFZIndexZfillnar rr+rM)rSr'Z model_namesrrrZ drop_omittedrZcolsZcolnamesrrr%rrZspecialZsummZvcZorderedZ unorderedotherZuorr*namer]datZsmryr)r'rrrrrrr summary_cols %        &        rc Cs8z ||}W|Sttfyt|}Y|Swr) ValueError TypeErrorrstrip)elementr'rrrrrr5Rs  r5r#-c s<|} z | fdd} Wnty!| fdd} Ynw|r/dd| jD} nd} |r@fdd| jD} nfdd| jdddfD| jdddf<d} tt | | | t t d } || j d d <|| j d d <|| j d d <|| j d d<|| j d d<|| j d d<dt|d| j d d<| S)Nc t|Srr4rr9rrrA_rz$_df_to_simpletable..crrr4rr9rrrAarcSsg|]}t|qSr)rr6rrrr:craz&_df_to_simpletable..c g|] }t|tdqSrcrrlr6rnrrr:gs crrrr6rrrr:isr)headersstubsZltx_fmtZtxt_fmtZlatexZ data_alignsZ header_alignrqtable_dec_abovetable_dec_belowheader_dec_belowrcrZcolsep)copymaprZapplymaprrr%rrr=r.rrZoutput_formatsrl)r*r(r'r&r%rrrrmrnrrrrur)r'rnr_df_to_simpletableZs4  rc Csg}|r |ddnd}|durdgt|}|dur"dgt|}t|D](\}}||d}||d} ||d} |t|| || |||||dq&|S)Nrr'r"r%r&r()r(r'r&r%rmrn)rk enumerater)r) r r rmrnror'rvr%r&r(rrrrjxs"     rj)NNrUTFr")r"TFr)r"rFNrFT)r") r#r"TTrNrrrry)Zstatsmodels.compat.pandasrZstatsmodels.compat.pythonrr functoolsrrwrdZnumpyr=Zpandasr,r/rZtableformattingrrr rirRrZrYrrrrr5rrjrrrrsL   m 2 = '  z