a < b@sPddlmZddlZddlZGdddZedkrLddlm Z eZ e gdgdgdd Z e d d gd d gdZe je egee egddgdde e d d gd d gdZe je egeje egddddgdde ejgdZejddgZej gdgddedZee_e jeedgde dS) ) annotationsNc@steZdZdZd ddddddZd d d d d Zd dddZd!ddddZddZddZ d"dddddZ dS)# TablePlotterzg Layout some DataFrames in vertical/horizontal layout for explanation. Used in merging.rst Gz??@float cell_width cell_height font_sizecCs||_||_||_dS)Nr)selfr r r r e/Users/vegardjervell/Documents/master/model/venv/lib/python3.9/site-packages/pandas/util/_doctools.py__init__szTablePlotter.__init__z pd.DataFrameztuple[int, int])dfreturncCs"|j\}}||jj||jjfS)zA Calculate table shape considering index levels. )shapecolumnsnlevelsindex)r rrowcolr r r_shapes zTablePlotter._shape)rcs|rPttfdd|D|d}tfdd|D|d}nLtfdd|D|dg}tfdd|D|dg}||fS) zQ Calculate appropriate figure size based on left and right data. c3s|]}|dVqdSrNr.0rr r r %z*TablePlotter._get_cells..rc3s|]}|dVqdSNrrrr rr&rr!csg|]}|dqS)rrrrr r (rz+TablePlotter._get_cells..csg|]}|dqS)r!rrrr rr")r)maxsumr)r leftrightverticalvcellshcellsr rr _get_cellss &&&&zTablePlotter._get_cellsNTbool)r'c s2ddlm}ddlm}t|ts(|g}fdd|D}|}|||\}}|rpj|j |f} nj|j |f} |j | d} |r^| t ||} t fdd|D} t fdd|D} tt||D]8\}\}}| | |d| f}j|||d | d q|| dd| df}j||d d |d | jd ddddnt fdd||gD}d t |}| d|} d}t||D]P\}}|}| | d|||df}j||||d ||d7}q|| d|df}j||d |d | jddddd| S)a Plot left / right DataFrames in specified layout. Parameters ---------- left : list of DataFrames before operation is applied right : DataFrame of operation result labels : list of str to be drawn as titles of left DataFrames vertical : bool, default True If True, use vertical layout. If False, use horizontal layout. rNcsg|]}|qSr )_convrrr rr"=rz%TablePlotter.plot..)figsizec3s|]}|dVqdSr rrrr rrLrz$TablePlotter.plot..c3s|]}|dVqdSrrrrr rrMr?titleheightResultg?g?g?gffffff?)topZbottomr%r&c3s|]}|dVqdSrrrrr rrVrr!g333333?)Zmatplotlib.gridspecgridspecmatplotlib.pyplotpyplot isinstancelistr,r*r r figureZGridSpeclenr# enumeratezipZ add_subplot _make_tableZsubplotZsubplots_adjustnpr)r r%r&labelsr'r4pltr)r(r-ZfigZgsZ max_left_colsZ max_left_rowsillabelaxZmax_rowsr1rspr rrplot,sD       zTablePlotter.plotcCs:t|tjr,|jdur$|jdd}n|}|d}|S)zF Convert each input to appropriate for table outplot. N)nameNaN)r7pdZSeriesrHZto_frameZfillna)r datar r rr,gs    zTablePlotter._convcsjj}|dkr*ddjn*t|D] }|d|j|q2jj}|dkrjd}fddtd|D}t|}|j_t |g|_S)Nr!rIndexcsg|]}j|jqSr )r_get_level_valuesZ_values)rrArKr rr"sz.TablePlotter._insert_index..) copyrrinsertrangerMrrJ DataFrameconcat)r rK idx_nlevelsrA col_nlevelsrvaluesZcol_dfr rNr _insert_indexss"    zTablePlotter._insert_indexstrz float | Noner/c Cs|dur|ddSddlm}|jj}|jj}||}|j||dd}||j |durpdt |d}| } | d D]`\\} } } | dkr| dn6| |kr| |kr| dn| |ks| |kr| d | |q|j||j d |d dS) NFr )locr.r!Zcelldz#AAAAAA)sizeoff)Z set_visibleZpandas.plottingplottingrrrrWtableZ set_fontsizer r:Z propertiesitemsZ set_facecolorZ set_height set_titleaxis) r rDrr0r1r^rTrUtbpropsrccellr r rr=s,        zTablePlotter._make_table)rrr)NT)N) __name__ __module__ __qualname____doc__rrr*rFr,rWr=r r r rrs  ; r__main__) )) )ABCrmrorsru)rvrxdf1df2T)r?r')XZr!)rbF)r!rvr!rw)r!rx)rv)rrw)rrxr}r~)r!r)rrrYrm)v1v2)rdf3)r?) __future__rZnumpyr>ZpandasrJrrhr5r6r@prRryrzrFrSshowrZ MultiIndex from_tuplesidxrrr r r rs0 !  $