M/PhWddlmZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl mZddlmZmZGdd Zd Zd d d dddddZdZ d#dZd$dZd%dZdZ d&dZd'dZ d(d!Zd)d"ZdS)*) FUTURE_STACK)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 Z/var/www/html/test/jupyter/venv/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.rs 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 DataFramer1)rarrayr.r-tables r add_arrayzSummary.add_array=sC U## Eu!-U  < < < < z$Summary.add_dict..]s#>>> 1l++>>>rc0g|]}t|Sr<r=r?s rrBz$Summary.add_dict..^s#@@@ 1l++@@@rrc.tj||gSr)nphstackrAys rz"Summary.add_dict..fs29aV#4#4rr.N) keysvaluesrFr5rshapevstacksplitrr7) rdncolsr.r-rLvalsdatapads ` radd_dictzSummary.add_dictMs ?>>>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_textis  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: rD) isinstancerrmodel __class____name__ _model_types)rrresultsr\s r add_titlezSummary.add_titleosd 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:r9rK)r-)rr`)summary_paramsre summary_modelr+rVr1ra) rr`rdr-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@rAs rrBz#Summary.as_text.. 222qqyy{{222r rN r8rDc:g|]}tj|Sr<)textwrapwrap)r@rAwidests rrBz#Summary.as_text..s%;;;Ax}Q'';;;rc8g|]}d|S)rs)joinrqs rrBz#Summary.as_text..s")))tyy||)))r) r rrr_measure_tables_simple_tablesrzrPr/lenint) rr rrrpad_col pad_index rule_equal simple_tablestabtxtoutrxs @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!rqs rrBz#Summary.as_html..rrrrsc:g|]}|ddS)rs
)replace)r@sts rrBz#Summary.as_html..s&HHH2BJJtY//HHHrr)r rr|rzr)rr rrrtemp_txtrrs rr"zSummary.as_htmls=&vx88 22M222iinnHHHHHnnX&&nnc3Z(( rrDcv|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_tabularrqs rrBz$Summary.as_latex..s$;;;q!!##;;;rz z8\\hline\n\\hline\n\\end{tabular}\n\\begin{tabular}{.*}\nz \\midrule\nz \begin{table}z \end{table}rsz \newline z \bigskip ) r rrr|rzrresubr) rlabelr rrrr 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')r8r9r'NN)rbr'NNN)rD)r^ __module__ __qualname__rrrr#r&r1r7rVrYrarmrr"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<rrqs rrBz#_measure_tables..s . . .1199;; . . .rc\g|])}t|d*Sr)r} splitlinesrqs rrBz#_measure_tables..s- 2 2 2c!,,..#$$ 2 2 2rr)r|maxranger}rNr~r/) r rrrlengthlen_maxpad_seprinseprUlen_news rr{r{s#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^rAs rrJzsummary_model..*sqw09rzModel:c$|jjjSr)family__classr^rs rrJzsummary_model..+sah&6&?rz Model Family:c.|jjjjSr)rlinkr]r^rs rrJzsummary_model..,sqx}'>'GrzLink Function:c|jjSr)r\ endog_namesrs rrJzsummary_model..-s AG,?rrfzDate:cd|jzSNz%#6d)nobsrs rrJzsummary_model../s &16/rzNo. Observations:cd|jzSr)df_modelrs rrJzsummary_model..0s &1:"5rz Df Model:cd|jzSr)df_residrs rrJzsummary_model..1s fqz&9rz Df Residuals:c|jdS)N converged mle_retvalsrs rrJzsummary_model..2s 1=#=rz Converged:c|jdS)N iterationsrrs rrJzsummary_model..3s  l(CrzNo. Iterations:c|jSr)methodrs rrJzsummary_model..4srzMethod:c|jdS)Nnorm fit_optionsrs rrJzsummary_model..5s amF3rzNorm:c|jdS)N scale_estrrs rrJzsummary_model..6s AM+$>rz Scale Est.:c|jdS)Ncovrrs rrJzsummary_model..7s 1=#7rz Cov. Type:rDz (uncentered)cd|jzSNz%#8.3f)rsquaredrs rrJzsummary_model..:s 1:8Mr R-squared:cd|jzSr) rsquared_adjrs rrJzsummary_model..;s X=VrzAdj. R-squaredcd|jzSr) prsquaredrs rrJzsummary_model..<s (Q[*@rzPseudo R-squared:cd|jzSNz%8.4f)aicrs rrJzsummary_model..= Wqu_rzAIC:cd|jzSr)bicrs rrJzsummary_model..>rrzBIC:cd|jzSNz%#8.5g)llfrs rrJzsummary_model..?s 15(8rzLog-Likelihood:cd|jzSr)llnullrs rrJzsummary_model..@s AH!4rzLL-Null:cd|jzSr) llr_pvaluers rrJzsummary_model..As X %<rz LLR p-value:cd|jzSr)deviancers rrJzsummary_model..Bs (QZ"7rz Deviance:cd|jzSNz%#6.3g) pearson_chi2rs rrJzsummary_model..Cs h&?rz Pearson chi2:cd|jzS)Nz%#8.4g)fvaluers rrJzsummary_model..Ds X%8rz F-statistic:cd|jzSr)f_pvaluers rrJzsummary_model..Es Hqz,ArzProb (F-statistic):cd|jzSr)scalers rrJzsummary_model..Fs x!'1rzScale:) k_constantitemsAttributeErrorKeyErrorNotImplementedError)r`rrl rsquared_typerkeyfuncs rrhrh s ... 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))DDrbTFr'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|[r8r]zzP>|z|)r[tupleparamsbsetvaluespvaluesconf_intrFr5TrGr3r4rcolumnsr\rT param_namesr+r exog_names) r`rjrirdre skip_headerr-rrrrrrTs rrgrgRsu:'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?czt|}|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}|jdit }|rt|d tj }t|dtj }tj ||d} | r0| fd} tj|| gd }tj|}t!|jjg|_|S)z.s L1,<r(r)g?r*rbg{Gz?rr)rrD)zR-squared Adj.rDc|zSrr<rs rrJz_col_params..s L1$4r)axisr<)rgr applyiloclocstackrgetattrrFnanr3Seriesnotnullanyconcatr4rr\r) resultr-stars include_r2rescolidxrrr2s ` r _col_paramsr)sP  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 #) # #l # #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 NrD)r+) r[dictr/rr3r4rr\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) defaultdictIrt)r}set collectionsr/rr/) list_of_namesr/ name_counterr,_names r _make_uniquer6s 3}  #m"4"444'''''';s##L F99Us" eckL$778888 Mrr<ct|ts|g}fd|D}|rt|} ntd|D} tt |D]} | | g|| _d} t|dj} |ddD]%} | jD]}|| vr| |&dD]>}|| vr8| || t | |?t| |}| | }r|j d tj}|j|dkj fd D}fd D}||zfd |j dD}|z |r|D]}|| d }g}|j d} td| jddD]z} || | | dz| jdkr+| | | | dzkr|d \|| | dz{||_rfd|D}n d|D}t'|td|DD] \}}|g|_t| |}tjt+j||g}|j|_tj|j |j z|_|}|d }t3}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@rAr-r$r#s rrBzsummary_col..sG A A A34 \#- / / / A A Arc(g|]}|jdSrr rqs rrBzsummary_col..s < < .merg swwqg4DwIIIrrrNrr8cg|]}|v| Sr<r<)r@rAvarnamess rrBzsummary_col.. s???h1rcg|]}|v| Sr<r<)r@rAregressor_orders rrBzsummary_col..!s#EEE1A_,D,DQ,D,D,Drcg|]}|v| Sr<r<)r@rA new_orders rrBzsummary_col..#s-(((qY&&&&&r)levelrDc vg|]5}t||jjj6Sr<)r-getr\r]r^)r@rAr,s rrBzsummary_col..7sXIII;<!Y]]17+<+E+46677IIIrc Lg|]!}t|t|dd"S)default_model_infosN)r-rrqs rrBzsummary_col..:s;! !WQ(=tDDEErc(g|]}|jdSrr:)r@r0s rrBzsummary_col..=s+I+I+IbBJqM+I+I+IrTr9)r,r.zStandard errors in parentheses.z* p<.1, ** p<.05, ***p<.01)r[listr6rr}r r+r/removeinsertrreindexget_level_valuestolistr3r value_countsrrNzipr4rFrOIndexfillnar rr1rY)r`r- model_namesr#r,rE drop_omittedr$colscolnamesrrAr+r&rspecialsummvcordered unorderedotheruor'r0namerldatsmryrGrCs ` ``` ` @@r summary_colrfsJ gt $ $) A A A A A A8? A A AD> ,, < r>RsPW$  "'ll 99;;s  ++r(-c  |} | fd} n(#t$r| fd} YnwxYw|r$d| jD} nd} |r& fd| jD} n+ fd| jdddfD| jdddf<d} ttj | | | tt} || j dd <|| j dd <|| j d d <|| j d d <|| j d d <|| j d d<dt|dzz| j d d<| S)Nc$t|Srr=rs rrJz$_df_to_simpletable.._s 1l ; ;rc$t|Srr=rs rrJz$_df_to_simpletable..asZ<%@%@rc,g|]}t|Sr<)rrqs rrBz&_df_to_simpletable..cs888a3q66888rcTg|]$}t|tdzz%Srtrr~r@rArs rrBz&_df_to_simpletable..gs/KKK1Q#i..3..KKKrcTg|]$}t|tdzz%Srrrsrts rrBz&_df_to_simpletable..is?333a&&3y>>C#77333rr)headersstubsltx_fmttxt_fmtlatex data_aligns header_alignrtable_dec_abovetable_dec_belowheader_dec_belowrtrcolsep)copymaprapplymapr rSr+rrrFr5rr output_formatsr~)r0r.r-r,r+r}r~rrrrdrvrwrs ` ` r_df_to_simpletablerZs ''))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+rr)r} enumerater/r) r rrrrr-rvr+r,r.s rr|r|xsM2:F8A;~..L#F #C#f++% &!!II1 G$!X& G$/=I7=U8? :CA, HHH I I I I r)NNrbTFr')r'TFr)r'r<FNr<FT)r') r(r'TTrlNrlrrr)statsmodels.compat.pandasrstatsmodels.compat.pythonrr functoolsrrrvnumpyrFpandasr3r6rtableformattingrr r r{r_rhrgr)r-r6rfr>rr|r<rrrs222222****** ////////iiiiiiiiX+++402?/,11  ///dFJ399999z$$$$N*   EJAFwwwwtCGHLBC<r