VfdVddlmZddlZddlmZddlZddlZddlZddl Z ddl m Z ddl mZmZGddZd Zd d d d ddddZdZ d"dZd#dZd$dZdZ d%dZd&dZ d'd Zd(d!ZdS)))lzipN)reduce) SimpleTable) fmt_latexfmt_txtcveZdZdZdZdZdZdZ dd Zdd Z dd Z dZ ddZ ddZ dZdZddZdS)SummarycLg|_g|_g|_d|_d|_dS)NF)tablessettings extra_txttitle _merge_latexselfs :lib/python3.11/site-packages/statsmodels/iolib/summary2.py__init__zSummary.__init__s,   !c*|SNas_textrs r__str__zSummary.__str__s||~~rcptt|dz|zdzS)Nz """ z """)strtyperrs r__repr__zSummary.__repr__s+4::*T\\^^;gEErc*|S)z$Display as HTML in IPython notebook.as_htmlrs r _repr_html_zSummary._repr_html_s||~~rc*|S)z9Display as LaTeX when converting IPython notebook to PDF.)as_latexrs r _repr_latex_zSummary._repr_latex_!s}}rT%.4frc|||||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 s radd_dfzSummary.add_df%sN$#f$05BB 2 X&&&&&rcbtj|}||dd||dS)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 DataFramer0)rarrayr-r,tables r add_arrayzSummary.add_array<sC U## Eu!-U  < < < < z$Summary.add_dict..\s#>>> 1l++>>>rc0g|]}t|Sr;r<r>s rrAz$Summary.add_dict..]s#@@@ 1l++@@@rrc.tj||gSr)nphstackr@ys rz"Summary.add_dict..es29aV#4#4rr-N) keysvaluesrEr4rshapevstacksplitrr6) rdncolsr-r,rKvalsdatapads ` radd_dictzSummary.add_dictLs ?>>>QVVXX>>>@@@@QXXZZ@@@xT4(()) :a=5 A % %4:a=501C9dBHSRH:-=$>$>?@@Dxe$$44d;; t5)))))rc:|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)rr.)rstrings radd_textzSummary.add_texths  f%%%%%rNct|tr ||_dS|3|jjj}|t vr t |}d|z|_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: rC) isinstancerrmodel __class____name__ _model_types)rrresultsr[s r add_titlezSummary.add_titlensd eS ! ! DJJJ" /8L(((/E(50  r皙?ct|||j}t|}|||_|||d<||d||||||dS)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:r8rJ)r,)rr_)summary_paramsrd summary_modelr*rUr0r`) rr_rcr,rxnameynameparaminfos radd_basezSummary.add_bases&we7=IIIW%%  EK  */D& ' d# &&& E 555 UG44444rcx |j}|j}|j}|j}t ||\}} dz}t ||||}d|D} d| } | d} || d<| |d| } |A|}t| kr+dt dz t|dz z z|z}nd} fd |D} d | D} d| } d|| | g} | S) z%Generate ASCII Summary Table =c6g|]}|Sr;rr?r@s rrAz#Summary.as_text.. 222qqyy{{222r rN r7rCc:g|]}tj|Sr;)textwrapwrap)r?r@widests rrAz#Summary.as_text..s%;;;Ax}Q'';;;rc8g|]}d|S)rr)joinrps rrAz#Summary.as_text..s")))tyy||)))r) r r rr_measure_tables_simple_tablesryrOr.lenint) rr r rrpad_col pad_index rule_equal simple_tablestabtxtoutrws @rrzSummary.as_textsR= N %4VX%F%F"Fc\ &vx)LL 22M222iinniiooA :iinn  E5zzF""c&1*s5zzA~"=>>>FE;;;;;;;))S)))iinniiS)** rc|j}|j}t||}d|D}d|}d|jD}d|}d||g}|S)z$Generate HTML Summary Table c6g|]}|Sr;r rps rrAz#Summary.as_html..rqrrrc:g|]}|ddS)rr
)replace)r?sts rrAz#Summary.as_html..s&HHH2BJJtY//HHHrr)r r r{ryr)rr r rrtemp_txtrrs rr!zSummary.as_htmls=&vx88 22M222iinnHHHHHnnX&&nnc3Z(( rrCcv|j}|j}|j}| d|zdz}nd}d|zdz}t||}d|D}d|}d}|jrt j|d |}d |||d f}d |}d |j} |dz| z} | 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{c6g|]}|Sr;)as_latex_tabularrps rrAz$Summary.as_latex..s$;;;q!!##;;;rz z8\\hline\n\\hline\n\\end{tabular}\n\\begin{tabular}{.*}\nz \\midrule\nz \begin{table}z \end{table}rrz \newline z \bigskip ) r r rr{ryrresubr) rlabelr r rrr to_replace non_captionedrrs rr$zSummary.as_latexs=    5(3.EE!EU"S(&vx88 ;;];;;kk#A    :&^S99C(%^K  -00 ""4>22o-3 r)TTr&r')r'r&)r7r8r&NN)rar&NNN)rC)r] __module__ __qualname__rrrr"r%r0r6rUrXr`rlrr!r$r;rrr r s"""FFF@F''''.<<<< ****8&&&     "HL#'5555:###J$%%%%%%rr ct||}d|D}d|D}t|}g}g}tt|D]|}t||jddz d} t |||z | z } || ||| | zz} ||| z }||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. c6g|]}|Sr;rrps rrAz#_measure_tables..s . . .1199;; . . .rc\g|])}t|d*Sr)r| splitlinesrps rrAz#_measure_tables..s- 2 2 2c!,,..#$$ 2 2 2rr)r{maxranger|rMr}r.) r r rrlengthlen_maxpad_seprinseprTlen_news rrzrzs#6844M . . . . .C 2 2c 2 2 2F&kkGGI 3s88__,,6!9?1%)1--7VAY&$.//s)dSj(7*++++ Is6{{ **rzOrdinary least squareszGeneralized least squaresz$Generalized least squares with AR(p)zWeighted least squareszRobust linear modelzNegative binomial modelzGeneralized linear model)OLSGLSGLSARWLSRLMNBinGLMcd}i}d|d<d|d<d|d<d|d <||d <d |d <d |d<d|d<d|d<d|d<d|d<d|d<d|d<d|d<|jrdnd}d|d |zd!z<d"|d#|zd!z<d$|d%<d&|d'<d(|d)<d*|d+<d,|d-<d.|d/<d0|d1<d2|d3<d4|d5<d6|d7<d8|d9<i}|D]1\}} ||||<#tttf$rY.wxYw|S):z8 Create a dict with information about the model chtj}|dS)Nz%Y-%m-%d %H:%M)datetimenowstrftime)argskwdsrs rtime_nowzsummary_model..time_now$s)##%%||,---rc$|jjjSr)r[r\r]r@s rrIzsummary_model..)sqw09rzModel:c$|jjjSr)family__classr]rs rrIzsummary_model..*sah&6&?rz Model Family:c.|jjjjSr)rlinkr\r]rs rrIzsummary_model..+sqx}'>'GrzLink Function:c|jjSr)r[ endog_namesrs rrIzsummary_model..,s AG,?rrezDate:cd|jzSNz%#6d)nobsrs rrIzsummary_model...s &16/rzNo. Observations:cd|jzSr)df_modelrs rrIzsummary_model../s &1:"5rz Df Model:cd|jzSr)df_residrs rrIzsummary_model..0s fqz&9rz Df Residuals:c|jdS)N converged mle_retvalsrs rrIzsummary_model..1s 1=#=rz Converged:c|jdS)N iterationsrrs rrIzsummary_model..2s  l(CrzNo. Iterations:c|jSr)methodrs rrIzsummary_model..3srzMethod:c|jdS)Nnorm fit_optionsrs rrIzsummary_model..4s amF3rzNorm:c|jdS)N scale_estrrs rrIzsummary_model..5s AM+$>rz Scale Est.:c|jdS)Ncovrrs rrIzsummary_model..6s 1=#7rz Cov. Type:rCz (uncentered)cd|jzSNz%#8.3f)rsquaredrs rrIzsummary_model..9s 1:8Mr R-squared:cd|jzSr) rsquared_adjrs rrIzsummary_model..:s X=VrzAdj. R-squaredcd|jzSr) prsquaredrs rrIzsummary_model..;s (Q[*@rzPseudo R-squared:cd|jzSNz%8.4f)aicrs rrIzsummary_model..< Wqu_rzAIC:cd|jzSr)bicrs rrIzsummary_model..=rrzBIC:cd|jzSNz%#8.5g)llfrs rrIzsummary_model..>s 15(8rzLog-Likelihood:cd|jzSr)llnullrs rrIzsummary_model..?s AH!4rzLL-Null:cd|jzSr) llr_pvaluers rrIzsummary_model..@s X %<rz LLR p-value:cd|jzSr)deviancers rrIzsummary_model..As (QZ"7rz Deviance:cd|jzSNz%#6.3g) pearson_chi2rs rrIzsummary_model..Bs h&?rz Pearson chi2:cd|jzS)Nz%#8.4g)fvaluers rrIzsummary_model..Cs X%8rz F-statistic:cd|jzSr)f_pvaluers rrIzsummary_model..Ds Hqz,ArzProb (F-statistic):cd|jzSr)scalers rrIzsummary_model..Es x!'1rzScale:) k_constantitemsAttributeErrorKeyErrorNotImplementedError)r_rrk rsquared_typerkeyfuncs rrgrgs ... D99DN??DGGD "?"?D DM 9 9D 55D99D==DCCD ((DO33DM>>D77D!,ABB/M.M.MD} $s *+3V3VD M )C /0 @ @D ,,DL,,DL88D 44D<    D  JsC))DDraTFr&c t|tr |\}}}} } } n1|j}|j}|j} |j} ||} tj||| | gj } tj | | g} tj | } |r7dddddt|dz ztd|dz z dzg| _n6ddd d dt|dz ztd|dz z dzg| _|s9 |jjj| _n(#t&$r|jj| _Yn wxYw|| _| 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|[r7r]zzP>|z|)rZtupleparamsbsetvaluespvaluesconf_intrEr4TrFr2r3rcolumnsr[rS param_namesr*r exog_names) r_rirhrcrd skip_headerr,rrrrrrSs rrfrfQsu:'5!!+;B8gwk//##E** 8VS'73 4 4 6D 9dH% & &D <  D HS'c%!)nn,c!eai-.@.@3.FH  S'c%!)nn,c!eai-.@.@3.FH   2 +7DJJ 2 2 2 1DJJJ 2 Ks D!!D?>D?c~t|}|jddD]#}||fd||<$d|jdddfzdz|jdddf<|r|jdddfdk}|j||jd fd z|j||jd f<|jdddfd k}|j||jd fd z|j||jd f<|jdddfd k}|j||jd fd z|j||jd f<|jddddf}|}|rt |d tj}t |dtj}tj ||d} | r0| fd} tj || gd }tj|}t|jjg|_|S)z.s L1,<r(r)g?r*rag{Gz?rr))rrC)zR-squared Adj.rCc|zSrr;rs rrIz_col_params..s L1$4r)axis)rfr applyiloclocstackgetattrrEnanr2Seriesnotnullanyconcatr3rr[r) resultr,stars include_r2rescolidxrrr2s ` r _col_paramsr'sF  C{2A2>>s8>>"<"<"<"<==C38AAAqD>)C/CHQQQTN Jhqqq!tnr!'*wsCKN/B'Cc'IS[^#$hqqq!tns"'*wsCKN/B'Cc'IS[^#$hqqq!tns"'*wsCKN/B'Cc'IS[^#$ (111bqb5/C ))++C/6:rv66v~rv>> Y80<>>?? ::<<     /444455B)S"IA...C ,s  Cv|/001CK Jrcv|i}g}g}|D]~}t||tr ||||n%#t$r|dYnwxYw||t jt |jj|i|}|S)z!Stack model info in a column NrC)r*) rZdictr.rr2r3rr[r)r  info_dictrr*rs r _col_infor+s C E  ilD ) )    JJ|y|F++ , , , ,    JJrNNNNN  Q ,FL455s;5 I I IC Js$AA10A1ctt|t|kr|Sddlm}|t}g}|D]3}||xxdz cc<||dz||z4|S)Nr) defaultdictIrs)r|set collectionsr-rr.) list_of_namesr- name_counterr+_names r _make_uniquer4s 3}  #m"4"444'''''';s##L F99Us" eckL$778888 Mrr;cBt|ts|g}fd|D}|rt|} ntd|D} tt |D]} | | g|| _d} t | |} r| jd tj  } | j | dkj fdD}fdD}||zfd| jdD}|z |r|D]}|| j } g}| jd}td|jddD]z} ||| | d z|jdkr+|| || d zkr|d \||| d z{|| _rfd |D}n d |D}t#|td |DD] \}}|g|_d} t | |}tjt'j| |g}| j|_tj| j |j z|_|} | d } t/}d|_|| dd|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 : 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. c6g|]}t|S))r!r,r")r')r?r@r,r"r!s rrAzsummary_col..sG A A A34 \#- / / / A A Arc(g|]}|jdSrr rps rrAzsummary_col..s < < .merg &wwqg4"&(( (rrr7cg|]}|v| Sr;r;)r?r@varnamess rrAzsummary_col..s???h1rcg|]}|v| Sr;r;)r?r@regressor_orders rrAzsummary_col..s#EEE1A_,D,DQ,D,D,Drcg|]}|v| Sr;r;)r?r@ new_orders rrAzsummary_col..s-(((qY&&&&&rrrCc vg|]5}t||jjj6Sr;)r+getr[r\r])r?r@r*s rrAzsummary_col..+sXIII;<!Y]]17+<+E+46677IIIrc Lg|]!}t|t|dd"S)default_model_infosN)r+rrps rrAzsummary_col...s;! !WQ(=tDDEErc(g|]}|jdSrr8)r?r/s rrAzsummary_col..1s+I+I+IbBJqM+I+I+Irc4||dddSr:r?rGs rrAzsummary_col..merg4rBrTr8)r+r-zStandard errors in parentheses.z* p<.1, ** p<.05, ***p<.01)rZlistr4rr|r rr*get_level_valuestolistr2r value_countsrremoverMr.zipr3rErNIndexfillnar rr0rX)r_r, model_namesr!r*rF drop_omittedr"colscolnamesrrAsummvcordered unorderedotheruor%r*r/namerkdatsmryrHrDs ` ``` ` @@r summary_colrds1J gt $ $) A A A A A A8? A A AD> ,, < .Ws 1l ; ;rc$t|Srr<rs rrIz$_df_to_simpletable..YsZ<%@%@rc,g|]}t|Sr;)rrps rrAz&_df_to_simpletable..[s888a3q66888rcTg|]$}t|tdzz%Srsrr}r?r@rs rrAz&_df_to_simpletable.._s/KKK1Q#i..3..KKKrcTg|]$}t|tdzz%Srprqrrs rrAz&_df_to_simpletable..as?333a&&3y>>C#77333rr)headersstubsltx_fmttxt_fmtlatex data_aligns header_alignrtable_dec_abovetable_dec_belowheader_dec_belowrsrcolsep)copymaprapplymapr rQr*rrrEr4rroutput_formatsr})r/r-r,r+r*r{r|r}r~rrbrtrurs ` ` r_df_to_simpletablerRs ''))CBgg;;;;<< BBBll@@@@AAB 883;#5#5#7#7888 KKKK 8H8H8J8JKKK3333#&8AAAqD>333A RXc]]G5& 9 9 9B05Bg}-16Bg~..3Be]+2ABe./2ABe./3CBe/0),s7Q;/?/?)?BeX& Is1"AAc jg}|r|ddnd}|dgt|z}|dgt|z}t|D]d\}}||d}||d} ||d} |t|| || |||||e|S)Nrr,r&r*r+r-)r-r,r+r*r~r)r| enumerater.r) r r r~rrr,rvr*r+r-s rr{r{psM2:F8A;~..L#F #C#f++% &!!II1 G$!X& G$/=I7=U8? :CA, HHH I I I I r)NNraTFr&)r&TFr)r&r;FNr;FT)r&) r'r&TTrjNrjrrr)statsmodels.compat.pythonrr functoolsrrrunumpyrEpandasr2r5rtableformattingrrr rzr^rgrfr'r+r4rdr=rr{r;rrrs****** ////////iiiiiiiiX+++402?/,11  ///dFJ399999z$$$$N*   EJAFppppfCGHLBC<r