M/Ph# &dZddlmZmZddlmZmZddlZdOdZGdde Z d Z Gd d e Z Gd d Z edPiddddddddddddddddddddddd d!gd"d#d$dd%d!d&d!d'd!d(d!d)dd*dd+d,ZedPidd-dddddddddddddddd#dddd.d/d!d d!gd"d#d$dd%d0d&d0d)dd'd!d(d!d*dd+d,ZedPidddddddddddddddd1dd2ddd d3gd/d3d"d#d$dd%d4d&d4d)d5d'd4d(d4d*d5d+d6ZedPidd7dd8dd9dd:ddd;d<dd=dddddd>d d!gd/d!d"d#d$dd?d#d%d@d&d@d)dd'd@d(d@d*dd+d,dAdBdCdDdEdFdGdHdIZeeeeeJZedKddLMZdNZdS)Qa Provides a simple table class. A SimpleTable is essentially a list of lists plus some formatting functionality. Dependencies: the Python 2.5+ standard library. Installation: just copy this module into your working directory (or anywhere in your pythonpath). Basic use:: mydata = [[11,12],[21,22]] # data MUST be 2-dimensional myheaders = [ "Column 1", "Column 2" ] mystubs = [ "Row 1", "Row 2" ] tbl = SimpleTable(mydata, myheaders, mystubs, title="Title") print( tbl ) print( tbl.as_csv() ) A SimpleTable is inherently (but not rigidly) rectangular. You should create it from a *rectangular* (2d!) iterable of data. Each item in your rectangular iterable will become the data of a single Cell. In principle, items can be any object, not just numbers and strings. However, default conversion during table production is by simple string interpolation. (So you cannot have a tuple as a data item *and* rely on the default conversion.) A SimpleTable allows only one column (the first) of stubs at initilization, concatenation of tables allows you to produce tables with interior stubs. (You can also assign the datatype 'stub' to the cells in any column, or use ``insert_stubs``.) A SimpleTable can be concatenated with another SimpleTable or extended by another SimpleTable. :: table1.extend_right(table2) table1.extend(table2) A SimpleTable can be initialized with `datatypes`: a list of ints that provide indexes into `data_fmts` and `data_aligns`. Each data cell is assigned a datatype, which will control formatting. If you do not specify the `datatypes` list, it will be set to ``range(ncols)`` where `ncols` is the number of columns in the data. (I.e., cells in a column have their own datatype.) This means that you can just specify `data_fmts` without bothering to provide a `datatypes` list. If ``len(datatypes)K|]}t|kVdSNlen).0rowncolss W/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/iolib/table.py zcsv2st..ws. - -3s88u  - - - - - -z+All rows of CSV file must have same length.)dataheadersstubs) listopencsvreadernextappendranyOSError SimpleTable) csvfilerrtitlerowsfhrrrs @rcsv2str%[s 66D g ( ( (BB d??6llGG   G D==FFE ) ))LLQ(((KKABB((( )  % %%KK$$$ E>>E#$ QLLE - - - - - - ---ECDDD D' ? ? ??sB#CCCceZdZdZ ddZdZdZdZdd Zd Z dd Z dd Z dZ dZ dZdZdZdZdZdZdZddZdZdZedZdS)r aProduce a simple ASCII, CSV, HTML, or LaTeX table from a *rectangular* (2d!) array of data, not necessarily numerical. Directly supports at most one header row, which should be the length of data[0]. Directly supports at most one stubs column, which must be the length of data. (But see `insert_stubs` method.) See globals `default_txt_fmt`, `default_csv_fmt`, `default_html_fmt`, and `default_latex_fmt` for formatting options. Sample uses:: mydata = [[11,12],[21,22]] # data MUST be 2-dimensional myheaders = [ "Column 1", "Column 2" ] mystubs = [ "Row 1", "Row 2" ] tbl = text.SimpleTable(mydata, myheaders, mystubs, title="Title") print( tbl ) print( tbl.as_html() ) # set column specific data formatting tbl = text.SimpleTable(mydata, myheaders, mystubs, data_fmts=["%3.2f","%d"]) print( tbl.as_csv() ) with open('c:/temp/temp.tex','w') as fh: fh.write( tbl.as_latex_tabular() ) Nc  ||_||_|jzSimpleTable.__init__s> # ? "$'IINNbbs47||8L8LDO',,.. +0022',,.. )..00 X&&& X&&& x((( h''' W./// W./// w0$&&111 h0$&&111"  /     % Ns t$$ dD!!! ///&&rc*|Sr )as_textrAs r__str__zSimpleTable.__str__s||~~rc:tt|Sr )strtyperLs r__repr__zSimpleTable.__repr__s4::rc |jdi|S)Nr )as_html)rArIs r _repr_html_zSimpleTable._repr_html_st|''h'''rTc |j|fi|Sr )as_latex_tabular)rAcenterrIs r _repr_latex_zSimpleTable._repr_latex_s$t$V88x888rcn|r|d|d|r|d|dSdS)akReturn None. Adds headers and stubs to table, if these were provided at initialization. Parameters ---------- headers : list[str] K strings, where K is number of columns stubs : list[str] R strings, where R is number of non-header rows :note: a header row does not receive a stub! rheader_dec_below) dec_belowN)insert_header_row insert_stubs)rArrs rr?zSimpleTable._add_headers_stubssW  M  " "1g9K " L L L  (   a ' ' ' ' ' ( (rc| |j}n#t$rYnwxYw||||}t|||dS)z1Return None. Insert a row into a table. N)datatypetable)r_AttributeErrorr<rinsert)rAidxrr_s rrbzSimpleTable.insertsk   <!    iihdi;; D#s#####s  rZc 0d|D}tt|itd}|t |D]>\}}|||d|dkr|||_1d||_?dS)zReturn None. Insert a row of headers, where ``headers`` is a sequence of strings. (The strings may contain newlines, to indicated multiline headers.) c8g|]}|dS) )split)rheaders r z1SimpleTable.insert_header_row..s$@@@fv||D))@@@rr') fillvaluerhr_rN)rrr7reverse enumeraterbr[)rArownumrr[ header_rowsr#irs rr\zSimpleTable.insert_header_rows A@@@@ KCr0B0B0BCCDD oo . .FAs KKhK 7 7 7Avv)2V &&)-V &&  . .rc|j}t|}|D]r}|jdkr$|dd}|||1 ||t |V#t $rtdwxYwdS)zReturn None. Insert column of stubs at column `loc`. If there is a header row, it gets an empty cell. So ``len(stubs)`` should equal the number of non-header rows. rhr'emptyrkz'length of stubs must match table lengthN)r:iterr_rb insert_stubr StopIteration ValueError)rAlocrr:r empty_cells rr]zSimpleTable.insert_stubss  U  P PC|x''"U2888  3 ++++POOCe5555$PPP$%NOOOP P Ps #A//B c|j}|j}g}|D]Z}t|j}||d||}|D]}t ||_||_||[|S)zCReturn list of Row, the raw data as rows of cells. r)r_r`rG)r:r<rr,rr_rr) rAraw_datar:r<r#datarowdtypesnewrowcells rr=zSimpleTable._data2rowss  y  G4?++FT'F$OOOF " " $V  ! KK     rc$t|||S)z%DEPRECATED: just use the pad function)pad)rAswidthaligns rrzSimpleTable.pad1s1eU###rc V t|j|t d|D}d dkrdg|zS dg|z nJt tr g|z n.t |kr fdt|D g}t|D]3}t fd|D}| |4tt| }|S)z2Return list, the calculated widths of each column.c34K|]}t|VdSr r rrs rrz-SimpleTable._get_colwidths..:s(--CHH------r colwidthsrNc@g|]}|tzSr r )rrprequests rriz.SimpleTable._get_colwidths..Cs)GGGQwq3w<</0GGGrc3PK|] }t|jdfiV!dS)rN)rformat)rcfmt output_formats rrz-SimpleTable._get_colwidths..FsAOOa3xqx=@@C@@AAOOOOOOr) get_output_formatr8r.r6maxget isinstanceintrrangeziprr) rArrIr min_widthscolmaxwidthresultrrs ` @@r_get_colwidthszSimpleTable._get_colwidths5sL)-88 !-05577 8-------''+&& a<<3;  _cEkGG  % % Hi%'GG \\E ! !GGGG%,,GGGG : ( (COOOOO3OOOOOH   h ' ' ' 'c:w// rc J|g}t|D]\}}t|tr%||t |f?t|t rD||t t|f|||ft |} |j|S#t$r&|j |fi||j|<|j|cYSwxYw)z'Return list, the widths of each column.) sorteditemsrrrtupler7r@KeyErrorr)rArrI call_argskvkeys r get_colwidthszSimpleTable.get_colwidthsKs="O 8>>++,, ) )DAq!T"" )  !U1XX////At$$ )  !U6!''))+<+<%=%=!>????  !Q((((I (?3' ' ( ( (#64#6}$C$C9A$C$CDOC ?3' ' ' ' (s% C22-D"!D"c t|} |j|}n #t$rt d|zwxYw|||S-Return dict, the formatting options. Unknown format: %s)rr8r.rrvr6rArrIrs r_get_fmtzSimpleTable._get_fmt]s~*-88  C%m499;;CC C C C1MABB B C 8 s 1Ac 6|jdi|}|jdi|S)zXReturn string, the table in CSV format. Currently only supports comma separator.r)rr )rrK)rArIrs ras_csvzSimpleTable.as_csvjs3dm..X..t|""c"""rc |jd i|fd|D}t|d}dd}|r|d||z|j}|r@t |j|dd}|d|d d }|r|||zd |S) z!Return string, the table as text.r)c*g|]}|jdiS)text)r as_stringrrrs rriz'SimpleTable.as_text..vs+GGG3-#-66#66GGGrtable_dec_above=r title_alignrtable_dec_below-rfr))rrrrbr"rrjoin)rArIformatted_rowsrowlenrr"rrs @rrKzSimpleTable.as_textqsdm..X..GGGG$GGG^B'((''"3S99  ?  ! !!_v%= > > >   , FCGGM3,G,GHHE  ! !!U + + +''"3S99  <  ! !/F": ; ; ;yy(((rc |jdi|dg}|jrd|jz}|||fd|D|dd|S)zReturn string. This is the default formatter for HTML tables. An HTML table formatter must accept as arguments a table and a format dictionary. r*zzc32K|]}|jdiVdS)r*Nr*rrs rrz&SimpleTable.as_html..s5KKsmcm::c::KKKKKKrz
%s
rfr)rr"rextendr)rArIrr"rs @rrSzSimpleTable.as_htmlsdm//h//78 : )+dj8E  ! !% ( ( (KKKKdKKKKKKj)))yy(((rc |jdi|}g}|r|d|dpd}|dpd}d}d}||gzD]} | |krd} n | jdi|} | |kr]|r*|||d| r/|d| z|s||| |kr"|| jdd di|| }|jrd |jz} || |r|d d |d dS)ziReturn string, the table as a LaTeX tabular environment. Note: will require the booktabs package.r+z\begin{center}rr'rNz \end{tabular}z\begin{tabular}{%s}rz%%\caption{%s}z \end{center}rfz$$ )r+r )rr get_alignsrr"rreplace) rArWrIrrrr prev_alignslastralignsr"s rrVzSimpleTable.as_latex_tabularsdm00x00  5  ! !"3 4 4 4/06B/06B 4&= ! !Cd{{'77377$$<"))/:::"))*:;;;?"))*@6*IJJJ&?&--o>>>d{{%%!CM???3??AAA KK : )% 2E  ! !% ( ( (  3  ! !/ 2 2 2yy((00s;;;rc\t||D]\}}||dS)aReturn None. Extend each row of `self` with corresponding row of `table`. Does **not** import formatting from ``table``. This generally makes sense only if the two tables have the same number of rows, but that is not enforced. :note: To extend append a table below, just use `extend`, which is the ordinary list method. This generally makes sense only if the two tables have the same number of columns, but that is not enforced. N)rr)rAr`row1row2s r extend_rightzSimpleTable.extend_rights@dE**  JD$ KK      rcB|D]}|D]}||}|||_dS)zReturn None. Labels cells based on `func`. If ``func(cell) is None`` then its datatype is not changed; otherwise it is set to ``func(cell)``. Nrk)rAfuncrr~labels r label_cellszSimpleTable.label_cellssL  * *C * *T $$)DM * * *rcd|DS)Ncg|] }|j Sr rrs rriz$SimpleTable.data..s)))S)))rr rLs rrzSimpleTable.datas))D))))r) NNr'NNNNNNN)Tr )rZ)__name__ __module__ __qualname____doc__r>rMrQrTrXr?rbr\r]r=rrrrrrKrSrVrrpropertyrr rrr r |s2>@EI7;>!>!>!>!@(((9999(((" $ $ $ $.... PPP"$$$$,((($   ###)))0))) .<.<.<.<`    * * ***X***rr c|dkr||}n1|dkr||}n||}|S)zCReturn string padded with spaces, based on alignment parameter.lr)ljustrjustrW)rrrs rrrsL || GGENN # GGENN HHUOO HrcZeZdZdZ ddZdZdZdZd Zdd Z d Z e d Z dS)r;zjProvides a table row as a list of cells. A row can belong to a SimpleTable, but does not have to. rN row_dec_belowc |_|_|tn|j_|_t _|_t fd|DdS)aw Parameters ---------- seq : sequence of data or cells table : SimpleTable datatype : str ('data' or 'header') dec_below : str (e.g., 'header_dec_below' or 'row_dec_below') decoration tag, identifies the decoration to go below the row. (Decoration is repeated as needed for text formats.) Nc32K|]}|VdS))rNr )rr~rGrAs rrzRow.__init__..s2FF$XXd555FFFFFFr) r_r`r9r:_fmtr7 special_fmtsr[rr>)rAseqr_r`rGr[rIs` ` rr>z Row.__init__s!   } ;   FF" dFFFFF#FFFGGGGGrc t|}||jvrt|j|<|j||dS)z Return None. Adds row-instance specific formatting for the specified output format. Example: myrow.add_format('txt', row_dec_below='+-') N)rrr7r6)rArrIs r add_formatzRow.add_formatsR *-88  1 1 1/3vvD m , -(//99999rc|j}t||s|}||d|}|||dS)zGReturn None. Inserts a stub cell in the row at `loc`. stub)r_rN)r:rrb)rArwrr:s rrtzRow.insert_stubsR $&& :D5D999D Crc t|} t|}n #t$rt d|zwxYw ||jj|n#t$rYnwxYw||j |||j |d}||||S)rrN) r default_fmtsr.rrvr6r`r8rarrr)rArrIr special_fmts rrz Row._get_fmt&s*-88  C}-2244CC C C C1MABB B C  JJtz0? @ @ @ @    D  49 8'++M4@@  " JJ{ # # # s1A%A88 BBc f|jfi|dfd|DS)zcReturn string, sequence of column alignments. Ensure comformable data_aligns in `fmt_dict`.r'c34K|]}|jfiVdSr ) alignment)rr~rrs rrz!Row.get_aligns..@s7MM~t~m;;s;;MMMMMMr)rrrs ` @rrzRow.get_aligns<sIdmM66X66wwMMMMMMMMMMMrr)c |j|fi|} |jj|fi|}n%#t$r|d}YnwxYw|dt |z}|d}|dd}|dd}g}t ||D]*\} } | j| fd|i|} || +|| |z|z} |j | |fi|} | S) aReturn string: the formatted row. This is the default formatter for rows. Override this to get different formatting. A row formatter must accept as arguments a row (self) and an output format, one of ('html', 'txt', 'csv', 'latex'). rN)rcolseprow_prer'row_postr) rr`rrarrrrrr_decorate_below) rArrIrrrrrformatted_cellsr~rcontent formatted_rows rrz Row.as_stringBsRdmM66X66 -0 0FF#FFII - - - ,,III -  s4yy(IX'')R((77:r**tY// , ,KD%!dk%LL}LLLG  " "7 + + + +&++o">">>I ,,]M44/244 s$AAc B||jd}||}n~t|}|dkrHt|}t|}t ||\}} |dz||z|d| zz}n!|dkr |dz|z}nt d|z|S)zLThis really only makes sense for the text and latex output formats. Nr)rfr+zI cannot decorate a %s header.)rr[rrdivmodrv) rA row_as_stringrrIr[rrow0lendec_lenrepeataddons rrzRow._decorate_below`sLL66  "FF-m<.ws+++d +++rr rLs rrzRow.dataus++d++++r)rNNrr) rrrrr>rrtrrrrrrr rrr;r;sCG*HHHH4 : : :,NNN <*,,X,,,rr;cneZdZdZd dZdZdZdZedZ dd Z d Z d Z e e e ZdS)r9zQProvides a table cell. A cell can belong to a Row, but does not have to. r'Nc t|tr%|j|_|j|_|j|_n!||_||_t |_|j|||_dSr ) rr9rr_ _datatyperr7r6r)rArr_rrIs rr>z Cell.__init__~sk dD ! !  DI!]DN DIIDI%DNDI """rcd|jzS)N%srrLs rrMz Cell.__str__sdirc t|} t|}n #t$rt d|zwxYw ||jjj|n#t$rYnwxYw ||jj n#t$rYnwxYw||j |||Sr) rrr.rrvr6rr`r8rarrs rrz Cell._get_fmts*-88  C}-2244CC C C C1MABB B C  JJtx~4]C D D D D    D   JJtx} % % % %    D  49 8 s-1A*A== B  B B.. B;:B;c |j|fi|}|j}|dd}t|tr||t |z}nd|dkr,|dp|dd}n2||vrd|z}||d}nt d|z|S) N data_alignsrr stubs_align stub_alignrz%s_alignUnknown cell datatype: %s)rr_rrrrrv)rArrIrr_rr label_aligns rrzCell.alignmentsdmM66X66=ggmS11 h $ $ E3{+;+; ; !H%EENE rc||_dSr )r)rAvals r set_datatypezCell.set_datatypes r)r'NNr)rrrrr>rMrr staticmethodrrrrrr_r rrr9r9zs       0 \+*+*+*+*Zx l33HHHrr9rr)rrrrrrrr'rrZrrrrrrrrrr header_align header_fmtstub_fmtrhrrxrrmissingz--r,rz"%s"z z z %sz %sz z --ltxz\toprulez \bottomrulez\midrulestrip_backslashTz \\z & empty_alignz \textbf{%s}r z\#z\$z\%z\&z$>$z\_z$|$)#$%&>_|r*r)r+rr*r+)htmrr!cl|dvr/ t|}n #t$rtd|zwxYw|S)Nr+zunknown output format %s)output_format_translationsrrv)rs rrrsY;;; I6}EMM I I I7-GHH H I s 1)FFNr )rstatsmodels.compat.pythonrr itertoolsrrrr%rr rr;r9r7r-r2r4r0rr.rr rrr1sRRh32222222(((((((( @@@@Bh*h*h*h*h*$h*h*h*V    D,D,D,D,D,$D,D,D,N}4}4}4}4}4}4}4}4B$ CC    BRS$d 3 ff!&s'(),t-.T/0 412 34r56 "78 D9>$ DD B  R T$d 3Tff$s%&'*v+,V-.r/0 412 34 "56 D7<4DDT  $   d 3 JYoo]$s%&'*}+,]-.{/0 =12 34 +56 M7<D%%% % K%#N % ![ % $ %D%W%%d% 5%ff%T%$s%%&'%()%,~-%.^/%0r1%2 >3%4 5%6 "7%8 D9%< =%Nt    "T r