U
|
®ý°d) ã @ sV d dl mZ d dlmZ d dlZd dlZG dd dZdddd Z e
|
d
|
krRe dS ) é )Ú annotations)ÚIterableNc @ sz e Zd ZdZd#ddddddd Zd
|
d d d dZd dddZd$dddddZdd Zdd Z d%d d dd d!d"Z
|
d S )&Ú TablePlotterzg
|
Layout some DataFrames in vertical/horizontal layout for explanation.
|
Used in merging.rst
|
ç®Gáz®×?ç Ð?ç @ÚfloatÚNone)Ú
|
cell_widthÚ cell_heightÚ font_sizeÚreturnc C s || _ || _|| _d S )N)r
|
r r )Úselfr
|
r r © r úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/util/_doctools.pyÚ__init__ s zTablePlotter.__init__z pd.DataFrameztuple[int, int])Údfr c C s" |j \}}||jj ||jj fS )zA
|
Calculate table shape considering index levels.
|
)ÚshapeÚcolumnsÚnlevelsÚindex)r r ÚrowÚcolr r r Ú_shape s
|
zTablePlotter._shape©r c s¤ |rPt t fdd|D |¡d }t fdd|D |¡d }nLt fdd|D |¡d g }t fdd|D |¡d g }||fS ) zQ
|
Calculate appropriate figure size based on left and right data.
|
c 3 s | ]} |¡d V qdS ©r N©r ©Ú.0r ©r r r Ú <genexpr>' s z*TablePlotter._get_cells.<locals>.<genexpr>r c 3 s | ]} |¡d V qdS ©é Nr r r r r r ( s r" c s g | ]} |¡d qS )r r r r r r Ú
|
<listcomp>* s z+TablePlotter._get_cells.<locals>.<listcomp>c s g | ]} |¡d qS )r" r r r r r r# + s )ÚmaxÚsumr )r ÚleftÚrightÚverticalÚvcellsÚhcellsr r r Ú
|
_get_cells! s &&&&zTablePlotter._get_cellsr Tz Iterable[str]Úbool©Úlabelsr( c s2 ddl m} ddlm} t|ts(|g} fdd|D } |¡} |||¡\}}|rp j| j | f} n j| j | f} |j
|
| d}
|
|r^| t ||¡} t fdd|D } t fd d|D } tt||D ]8\}\}}|
|
| |d|
f ¡} j|||d
|
| d qà| | dd
| d
f ¡} j||d d | d |
|
jddddd nÐt fdd||g D }d
|
t |¡ }| d|¡} d}t||D ]P\}} |¡}|
|
| d|||d
f ¡} j||||d ||d 7 }q¢| | d|d
f ¡} 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.
|
r )ÚgridspecNc s g | ]} |¡qS r )Ú_convr r r r r# ? s z%TablePlotter.plot.<locals>.<listcomp>)Úfigsizec 3 s | ]} |¡d V qdS r! r r r r r r N s z$TablePlotter.plot.<locals>.<genexpr>c 3 s | ]} |¡d V qdS r r r r r r r O s ç ð?)ÚtitleÚheightZResultgÍÌÌÌÌÌð?gÍÌÌÌÌÌì?g©?gffffffî?)ÚtopÚbottomr& r' c 3 s | ]} |¡d V qdS r r r r r r r X s r" g333333ë?)Z
|
matplotlibr/ Úmatplotlib.pyplotÚpyplotÚ
|
isinstanceÚlistr0 r+ r
|
r ÚfigureZGridSpecÚlenr$ Ú enumerateÚzipZ add_subplotÚ _make_tableZsubplotZsubplots_adjustÚnpr )r r& r' r. r( r/ Úpltr* r) r1 ZfigZgsZ max_left_colsZ max_left_rowsÚiÚ_leftZ_labelÚaxZmax_rowsr4 r Úspr r r Úplot. sD
|
|
|
zTablePlotter.plotc C s: t |tjr,|jdkr$|jdd}n| ¡ }| d¡}|S )zF
|
Convert each input to appropriate for table outplot.
|
NÚ )ÚnameÚNaN)r9 ÚpdZSeriesrH Zto_frameZfillna)r Údatar r r r0 i s
|
|
zTablePlotter._convc s² ¡ jj}|dkr* dd j¡ n*t|D ] } |d| j |¡¡ q2 jj}|dkr® j d¡} fddtd|D }t |¡}|j _t | g¡ | _ S )Nr" r ZIndexc s g | ]} j |¡jqS r )r Ú_get_level_valuesZ_values)r rB ©rK r r r# s z.TablePlotter._insert_index.<locals>.<listcomp>)
|
Úcopyr r ÚinsertÚrangerL r rJ Ú DataFrameÚconcat)r rK Ú idx_nlevelsrB Ú col_nlevelsr ÚvaluesZcol_dfr rM r Ú _insert_indexu s"
|
ÿ
|
zTablePlotter._insert_indexNÚstrz float | None)r3 r4 r c C s |d kr| d¡ d S ddlm} |jj}|jj}| |¡}|j||dd}| | j ¡ |d krpdt
|
|d }| ¡ } | d ¡ D ]`\\}
|
} } | d kr¤| d¡ n6|
|
|k rÀ| |k rÀ| d¡ n|
|
|k sÐ| |k rÚ| d
|
¡ | |¡ q|j|| j d | d ¡ d S ) NFr )Úplottingé )Úlocr2 r" Zcelldéÿÿÿÿz#AAAAAA)ÚsizeÚoff)Z set_visibleÚpandasrX r r r rV ÚtableZ set_fontsizer r< Z
|
propertiesÚitemsZ set_facecolorZ
|
set_heightÚ set_titleÚaxis) r rD r r3 r4 rX rS rT ÚtbÚpropsÚrÚcÚcellr r r r? s,
|
|
|
zTablePlotter._make_table)r r r )r T)N) Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r+ rF r0 rV r? r r r r r
|
s ü
|
; r r r c C s: dd l m} t }t dddgdddgdd d
|
gd ¡}t ddgdd
|
gd ¡}|j||gt ||g¡d dgdd | ¡ t ddgdd
|
gd¡}|j||gtj||gddd dgdd | ¡ tj ddddddg¡}tj ddg¡}tjddd d d d gd d d d!d"dgd#|d$}||_
|
|j||d%gd& | ¡ d S )'Nr é
|
é é é é é é é é )ÚAÚBÚC)ru rw Údf1Údf2Tr- )ÚXÚZr" )rb F)r" ru )r" rv )r" rw )é ru )r| rv )r| rw r| é é é é é é rY )Zv1Zv2)r Údf3)r. ) r7 r8 r rJ rQ rF rR ÚshowZ
|
MultiIndexÚ from_tuplesr )rA Úprx ry r ÚidxÚcolumnr r r Úmainª s, $$ ÿÿ,r Ú__main__) Ú
|
__future__r Útypingr Únumpyr@ r^ rJ r r rh r r r r Ú<module> s !
|
