M/PhuyddlmZmZmZddlZddlmZddlZddlZ ddl m Z ddl m Z mZmZmZddlmZdd Zdd Z ddZddZddZ d dZ d!dZ d"dZ d#dZd$dZd%dZGddZdS)&)lmaplrangelzipN) zip_longest) SimpleTable)fmt_2 fmt_2cols fmt_paramsgen_fmt) _model_typescHtj|}|dkr0t|dkst|dkrd|zSd|zS|dkr0t|dkst|dkrd|zSd|zStd | ) Nrg@g-C6?z%9.3gz%9.3fz%10.4gz%10.4fz1`prec` argument must be either 3 or 4, not {prec}prec)npsqueezeabs ValueErrorformat)xrs Y/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/iolib/summary.pyforgrs 1 A qyy FFcMMs1vv}}Q; Q;   FFcMMs1vv}}a< a< L &d&++-- -cftj|r |dz dzdzS|dz|krd|zSd|zS)amconvert 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 r NaNr z%#6dz%#8.2f)risnan)rwidths rd_or_fr"&sI" x{{) S 5((Av{{z!|r皙?textc !"#$%&'|dkrt|jjj}|Yt |t |jkr4t dt |jt|||\}}tj }tj d|g}tj d|} |jjj} |j } |j } |j} d| gfd| gfd|fd | gfg}t|\}}|}d}t!||||t" }d }| g|| g| gf}t!||||t" }|||"||||||d }|j|j|j|j|j|jd }t+d |z dzdz}dddd|dzgdddd|dzgdddd|dzgdddd|dzgdddd|dzgdddd|dzgd}|}|j$||!|j&t1t |}|| '|| %t3$fd|D&fd|D'fd|D%fd|D!fd|D}t!||| |dt4 #"#fd}"#fd}||d} |d kr1 | | S#t6$r| d!cYSwxYwdS)"a Parameters ---------- yname : str optional, Default is `Y` xname : list[str] optional, Default is `X.#` for # in p the number of regressors Confidance interval : (0,1) not implimented title : str optional, Defualt is 'Generalized linear model' returns : str '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() >>> 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. rNz`User supplied xnames must have the same number of entries as the number of model parameters ({})%H:%M:%S %a, %d %b %Y Model type:Date:Dependent Variable:zdf modeltitletxt_fmt)zMethod:Time:Number of Obs:zdf resid)OLSGLSGLSARWLSRLMGLMr d%coefstd errtP>|t|z Conf. IntervalzP>|z|)r0r1r2r3r5r4c&g|] }d|zS)z%#6.4g.0iparamss r zsummary..s"AAA1F1I.AAArc&g|] }d|zSz%#6.4fr?rArBstd_errs rrDzsummary.."BBBAGAJ/BBBrc&g|] }d|zSrFr?)rArBtstats rrDzsummary..s"@@@!E!H-@@@rc&g|] }d|zSrFr?)rArB prob_stats rrDzsummary..s"DDDaIaL1DDDrc@g|]}dt|zS)z (%#5g, %#5g))tuplerArBconf_ints rrDzsummary..s*NNNx{););;NNNrcLtdztz}|S)z4 print summary table for ols models  strtable general_tableparameter_tables r ols_printerzsummary..ols_printers)M""4'O(<(<< rcLtdztz}|S)NrSrTrVs r glm_printerzsummary..glm_printers'M""4'O(<(<< r)r0r5printr0)r model __class____name__lenrCrr _getnamestime localtimestrftimenobsdf_modeldf_residrrr extend_rightr:pvaluesrUrQbserrrKeyError)(selfynamexnamer,alphareturns model_infotime_now time_of_daydate modeltyperfrgrhgen_leftgen_stubs_left gen_data_left gen_title gen_headergen_table_leftgen_stubs_rightgen_data_rightgen_table_righttstats prob_statsalp param_header params_stubsexog_len params_datarZr\printersrQrXrYrCrMrHrKs( @@@@@@@rsummaryr@snn zzTZ1:; SZZ3t{+;+;;; &s4;'7'7 8 8:: :T5%00LE5~H=X667K = 2 2D $-I 9D}H}H ,4&!&.hZ(H %0$:!NMIJ !+!/'0)0 ###N#O!k!fjN ".","1(1*1 $$$O 000"M VVXXVVXXvvxxVVXXVVXXVVXX F < J qwm  S CIsGS;L5LMIsGS;L5LMIsGS;L5LMIsGS;L5LMIsGS;L5LMIsGS;L5LM LL [F}}U##HhGc%jj!!H 9 E9%IAAAAAAABBBBBBB@@@@x@@@DDDD8DDDNNNNXNNN PPK "+".y"9".(,*/ $$$O #;77H' %&8I&(( ( % % %"8E?$$ $ $ $ %s6MM#"M#c|%t|jdd |jj}nd}|Nt|jdd |jj}n+dt t |jD}||fS)z0extract names from model or construct names N endog_namesy exog_namescg|]}d|zS)zvar_%dr?rArBs rrDz_getnames.. sCCCaX\CCCr)getattrr^rrrangerarC)rmrnros rrbrbs } 4:}d 3 3 ?J*EEE } 4:|T 2 2 >J)EECC5T[1A1A+B+BCCCE %<rc||}}tj}tjd|gtjd|t|\}t dfdfdfdfdfd fd fd fd fd fdfdfdfdfdfdfdfdfdfdfg } |jjjdz}|+gd} j} | dn#YnxYwg}|} d} g} |D].\}}|| |}| ||f/| }|r5g}|D].\}}|| |}| ||f/|}d||zD}|gks J||rt|t|kr'|dgt|t|z zz }nFt|t|kr&|dgt|t|z zz }d|D}t|\}}t|| || t}ng}t|\}}t|| || t}|||}|S) zgenerate top table(s) TODO: this still uses predefined model_methods ? allow gleft, gright to be 1 element tuples instead of filling with None? r&r'rnror*c gSNr?rnsrzsummary_top..&s 5'rDep. Variable:c gSrr?rsrrzsummary_top..'s eWrzModel:c(jjjgSr)r^r_r`resultssrrzsummary_top..(sgm5>?rr)c gSrr?)rusrrzsummary_top..)s TFrr.cSrr?)rtsrrzsummary_top..*sKrr/cjgSr)rfrsrrzsummary_top..+s gl^rNo. Observations:c.tjgSr)r"rfrsrrzsummary_top..,s )=)=(>rz Df Model:c.tjgSr)r"rgrsrrzsummary_top..-s(8!9!9 :rz Df Residuals:c.tjgSr)r"rhrsrrzsummary_top...sVG,<%=%=$>rzLog-Likelihood:cdjzgS)Nz%#8.5g)llfrsrrzsummary_top../sx'+'=&>rNzRegression Results))rN)r(N)r)N)rN)z Df model:N)z Df resid:N)zLog-LikelihoodNcg|] \}}|| Srr?rAkvs rrDzsummary_top..ZsEEECAa19a999r)rrc(g|]\}}dd|zz|fS)z%-21sz r?rs rrDzsummary_top..hs*BBB1ga(!,BBBrr+)rcrdrerbdictr^r_r`rappendrarrr ri)rr,gleftgrightrnrorw gen_rightrs default_itemsrrzr{ gen_left_itemvalue gen_right_missing_valuesr}r~rrxryr|rXrurts` ` @@r summary_toprs iH~H=X667K = 2 2DWE???LE5 //// 2 ____ - ???? @ NNNN # '''' ( 3333 4  > > > > ? :::: ; >>>> ? >>>> ?   M } '03GG))) +C OO4 5 5 5 5  D IJI(( e ='M$'))E$''''H $ - -KD%}+ d+--   tUm , , , , FE8i#7EEEN R       y>>CMM ) ) *XY)GH HII ^^c(mm + +  IX(FG GHCB BBB *5y*A'%n&0&5,5.7 (((%0$:!NM !+!/'0)2 ###N000"M s C!!C%TFc  t|tr |\} ntj|jtj|jtj|jtj|jtj|| j dkrtdggS|r2dddddt|dz ztd |dz z d zg}n1ddd d dt|dz ztd |dz z d zg}|rd }t|||\}}t|tkrtd|} tt|} tjtjtjtjtj  t!fd| Dfd| Dfd| Dfd| D fd| D fd| D} t| || |t"} | S)a#create a summary table for the parameters 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_headers : bool If false (default), then the header row is added. If true, then no header row is added. Returns ------- params_table : SimpleTable instance rzNo Model Parametersr8r9r:r;[r ]r<r=Nrz-xnames and params do not have the same lengthc>g|]}t|dSrrrr@s rrDz"summary_params..s*BBBAVAYQ///BBBrc:g|]}t|Sr?rrGs rrDz"summary_params..%;;;QWQZ((;;;rc:g|]}t|Sr?r)rArBtvaluess rrDz"summary_params..rrc&g|] }d|zS)z%#6.3fr?)rArBrjs rrDz"summary_params..rIrc>g|]}t|dfS)rrrPs rrDz"summary_params..)>>>Xac]++>>>rc>g|]}t|dfS)r rrPs rrDz"summary_params..rrr+) isinstancerOrasarrayrCrkrrjrQsizerrUrbrarrrr )rrnrorpuse_t skip_headerr,r_rexog_idxrrYrQrCrjrHrs @@@@@rsummary_paramsrs@'5!! 7@G<'7HHGN++*W[))*W_--*W_--:g..u5566 {a234555 B 3c%'ll*C%' NNS,@B  3c%'ll*C%' NNS,@B  U;;;HAu 5zzS[[  HIIILc%jj!!H Z  Fj!!Gj!!Gj!!Gz(##HBBBBBBB;;;;(;;;;;;;(;;;BBBBBBB>>>>X>>>>>>>X>>> @@K "+".".(-*4 $$$O rc<t|tr |\}}}}}} n1|j}|j}|j}|j}||} |rgd} ngd} t|||\} }ddlm } tj ||||| f} | | | |S)a( Create a summary table for the parameters 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_headers : bool If false (default), then the header row is added. If true, then no header row is added. Returns ------- params_table : SimpleTable instance )r8r9r:r;Conf. Int. LowConf. Int. Upp.)r8r9r<r=rrrr) DataFrame)columnsindex) rrOrCrkrrjrQrbpandasrr column_stack)rrnrorprrCrHrrjrQrrrrWs rsummary_params_framersB'5!! +@G<'7HH+//##E** ==== === U;;;HAu      OVWgwI J JE 9UL > > >>rcB|0dtjjjddD}|+dt t jD}djD}|rcfd|D}t|g|R}d|D}t|gdgt |zgt |zR}d |D}n|}|}tj t} d gjj dz| d <t||||| S) acreate summary table of regression parameters with several equations This allows interleaving of parameters with bse and/or tvalues Parameters ---------- result : result instance the result instance with params and attributes in extras extras : list[str] additional attributes to add below a parameter row, e.g. bse or tvalues endog_names : {list[str], None} names for rows of the parameter array (multivariate endog) exog_names : {list[str], None} names for columns of the parameter array (exog) alpha : float level for confidence intervals, default 0.95 title : None or string Returns ------- tables : list of SimpleTable this contains a list of all seperate Subtables table_all : SimpleTable the merged table with results concatenated for each row of the parameter array Ncg|]}d|zS)zendog_%dr?rs rrDz%summary_params_2d..6s(:::!zA~:::rr cg|]}d|zS)zvar%dr?rs rrDz%summary_params_2d..9sEEEagkEEErc&g|]}d|DS)c0g|]}t|dSrr)rArs rrDz0summary_params_2d...<s%666$41%%%666rr?)rArows rrDz%summary_params_2d..<s'PPP366#666PPPrcDg|]}dt|DS)c&g|]}d|DS)cfg|].}ddt|dzdzz/S)z%10s(rr))rstrip)rArs rrDz;summary_params_2d....>sM''' 3aa)>)>)@)@#@3#FG'''rr?)rAcols rrDz0summary_params_2d...>s?:::''"%''':::r)r)rAwhatresults rrDz%summary_params_2d..>sL ::#*64#8#8:::rcg|] }|D]}| Sr?r?rAjrBs rrDz%summary_params_2d..Ds%+++a++A++++rcg|] }|D]}| Sr?r?rs rrDz%summary_params_2d..Fs%---q1--a----rz%s data_fmts)headersstubsr,r-) runiquer^endogrrarCrcopydeepcopyr shaper) rextrasrrr, res_params extras_listdatarr-s ` rsummary_params_2drs:::y!344QRR8::: EE5V]1C1C+D+DEEE QP&-PPPJ $* J----++4+++[GRD[)9)9$9#:3v;;#FGGG--E---mJ''G 6&-"5a"88GK tZ#(#(%, . . ..rc 6|}|j}|jdkr;|jd} t||jdkrt dn1t|t|krt dd} t |t s|d} n|} |jjdd}g} t| D]} ||jdd| f|j dd| f|j dd| f|j dd| f| || f} d}t| || ||||}| |tt|D]}||| |_t#| |}| |fS) asummary table for parameters that are 2d, e.g. multi-equation models Parameters ---------- result : result instance the result instance with params, bse, tvalues and conf_int endog_names : {list[str], None} names for rows of the parameter array (multivariate endog) exog_names : {list[str], None} names for columns of the parameter array (exog) alpha : float level for confidence intervals, default 0.95 use_t : bool indicator whether the p-values are based on the Student-t distribution (if True) or on the normal distribution (if False) keep_headers : bool If true (default), then sub-tables keep their headers. If false, then only the first headers are kept, the other headerse are blanked out endog_cols : bool If false (default) then params and other result statistics have equations by rows. If true, then equations are assumed to be in columns. Not implemented yet. Returns ------- tables : list of SimpleTable this contains a list of all seperate Subtables table_all : SimpleTable the merged table with results concatenated for each row of the parameter array rr zendog_names has wrong lengthNrF)rnrorprr) keep_headers)rCndimrrarrlistr^rrrkrrjrQrrr, table_extend)rrrrprr endog_colsresrCn_equendog_basenametableseqrestupskiphtblerB table_alls rsummary_params_2dflatrTsF C ZF {a Q {  v|A . .;<< < / {  s6{{ * *;<< < k4 ( (0  $NN(Ni+ABB/ FEll  sz!!!B$'2 AAAbD8I+aaad#S\\%%8%8%<>fKO$.e5*/111  d3{## $ $))%a.q V,???I 9 rcddlm}t|ddD]\}}||}|djdkrq|j|dd_d|dd_|ddj|dd_|s |dkr|dddD] }d|_ |dkr|}|d}|dd | |d|_|S) aextend a list of SimpleTables, adding titles to header of subtables This function returns the merged table as a deepcopy, in contrast to the SimpleTable extend method. Parameters ---------- tables : list of SimpleTable instances keep_headers : bool If true, then all headers are kept. If falls, then the headers of subtables are blanked out. Returns ------- table_all : SimpleTable merged tables as a single SimpleTable instance r)rNheaderr rtxt-) row_dec_below) rr enumeratedatatyper,r _datatyper add_formatextend)rrriir:crr1s rrrs&6!!!9%%  A HQKK Q4=H $ $7AaDGL $AaDG A$q'+AaDGK R!VV1abb  AAFF 77II2B MM%sM 3 3 3   Q    IO rc|dkrEd}dt||ddt|dgzS|dkr|S|dkrdd|DS|dkrPtj|d }|d dD]}|||S|d krdd |DStd )Nr$cTt|dddS)NrSr r)rUrsplit)rs rrz summary_return..sCFFMM$q11!4rrSr rcsvc3>K|]}|VdSr)as_csv)rArs r z!summary_return..s*44444444rlatexrr htmlc3>K|]}|VdSras_html)rArWs rrz!summary_return..s*==U======rz3available output formats are text, csv, latex, html)joinrrUrrras_latex_tabularr)r return_fmtstrdroprWparts rsummary_returnr&sV44yygvcrc{33s6":6GGHHH x   u  yy44V444444 w   fQi((122J  D LL    %%''' v  yy==f======NOOOrcfeZdZdZdZdZdZdZdZ ddZ dd Z d Z d Z dZ dZdZdS)Summarya Result summary 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 : str extra lines that are added to the text output, used for warnings and explanations. c"g|_d|_dSr)r extra_txtrms r__init__zSummary.__init__s rc*|Sr)as_textr+s r__str__zSummary.__str__s||~~rcptt|dz|zdzS)Nz """ z """)rUtyper/r+s r__repr__zSummary.__repr__s+4::*T\\^^;gEErc*|S)z$Display as HTML in IPython notebook.rr+s r _repr_html_zSummary._repr_html_s||~~rc*|S)z9Display as LaTeX when converting IPython notebook to PDF.)as_latexr+s r _repr_latex_zSummary._repr_latex_s}}rNcdt||||||}|j|dS)au Add 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 : str, optional if None, then a default title is used. gleft : list[tuple], optional elements for the left table, tuples are (name, value) pairs If gleft is None, then a default table is created gright : list[tuple], optional elements for the right table, tuples are (name, value) pairs yname : str, optional optional name for the endogenous variable, default is "y" xname : list[str], optional optional names for the exogenous variables, default is "var_xx". Must match the number of parameters in the model. )r,rrrnroN)rrr)rmrr,rrrnrorWs radd_table_2colszSummary.add_table_2cols sA0CuE&"'u666 5!!!!!rr#Tc|jjdkrt|||||}n7|jjdkrt|||||\}}nt d|j|dS)acreate and add a table for the parameter estimates 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) Returns ------- None : table is attached r )rnrorprr)rrrprzparams has to be 1d or 2dN)rCrrrrrr)rmrrnrorprrWrs radd_table_paramszSummary.add_table_params%s0 :?a  "3e5).000EE Z_ ! !,Se8=38GGGHAuu899 9 5!!!!!rc:d||_dS)zadd 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. rSN)r!r*)rmetexts r add_extra_txtzSummary.add_extra_txtHs5))rcZt|jd}|j |dz|jz}|S)zreturn tables as string Returns ------- txt : str summary tables and extra text as one string r$r#N r&rr*)rmr s rr.zSummary.as_textSs5T[V<<< > %,/C rct|jd}|j!|dz|jddz}|S)aVreturn tables as string Returns ------- latex : str 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. rr@NrArSz \newline r&rr*replace)rmrs rr6zSummary.as_latexasEt{w??? > %FNT^%;%;D/%R%RRE rcZt|jd}|j |dz|jz}|S)zreturn tables as string Returns ------- csv : str concatenated summary tables in comma delimited format rr@NrArB)rmrs rrzSummary.as_csvus5T[U;;; > %,/C rct|jd}|j!|dz|jddz}|S)zreturn tables as string Returns ------- html : str concatenated summary tables in HTML format rr@Nz

rSz
rD)rmrs rr zSummary.as_htmlsFdkf=== > %,&)?)?g)N)NND rNNNNNNNr#T)r` __module__ __qualname____doc__r,r/r2r4r7r9r;r>r.r6rr r?rrr(r(sFFFDH*.""""8CF#!"!"!"!"F * * *   (        rr()r)r)NNrr#r$N)NNrH)NNr#TFNrI)NNNN)NNr#TTF)T)r$)statsmodels.compat.pythonrrrr itertoolsrrcnumpyrstatsmodels.iolib.tabler!statsmodels.iolib.tableformattingrr r r summary2r rr"rrbrrrrrrr&r(r?rrrSs8888888888 !!!!!! //////#"""""----$4:='+|%|%|%|%~$ppppfFJ,0SSSSlAD#:?:?:?:?zIM :.:.:.:.zLPDIMMMM`****ZPPPP,hhhhhhhhhhr