3
���h� � @ sp d dl mZmZ d dlZd dlZG dd� d�Zedk�rld dl j
Z e� Z ej
dddgd d
d gd d
dgd��Zej
ddgd dgd��Ze jeegejeeg�ddgdd� e j� ej
ddgd dgd��Ze jeegejeegdd�ddgdd� e j� ejjd(d)d*d+d,d-g�Zejjd.d/g�Zej
dd d d d d gd d d!d"d#dgd$�ed%�Zee_e jeed&gd'� e j� dS )0� )�Optional�TupleNc @ s� e Zd ZdZdeeed�dd�Zejee e f d�d d
�Z
ee e f d �d d
�Z de d�dd�Z
dd� Zdd� Zd eee d�dd�ZdS ) � TablePlotterzg
Layout some DataFrames in vertical/horizontal layout for explanation.
Used in merging.rst
�G�z��?� �?� @)�
cell_width� cell_height� font_sizec C s || _ || _|| _d S )N)r r r
)�selfr r r
� r �7/tmp/pip-build-5_djhm0z/pandas/pandas/util/_doctools.py�__init__ s zTablePlotter.__init__)�df�returnc 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 ��� j|�d �}t � fdd�|D ��� j|�d }nLt � fdd�|D �� j|�d g �}t� fdd�|D �� j|�d g �}||fS ) zQ
Calculate appropriate figure size based on left and right data.
c 3 s | ]}� j |�d V qdS )r N)r )�.0�l)r r r
� <genexpr>% s z*TablePlotter._get_cells.<locals>.<genexpr>r c 3 s | ]}� j |�d V qdS )� N)r )r r )r r r
r & s r c s g | ]}� j |�d �qS )r )r )r r )r r r
�
<listcomp>( s z+TablePlotter._get_cells.<locals>.<listcomp>c s g | ]}� j |�d �qS )r )r )r r )r r r
r ) s )�max�sumr )r �left�right�vertical�vcells�hcellsr )r r
�
_get_cells s &&&&zTablePlotter._get_cellsNT)r! c s6 ddl j}ddlj}t|t�s$|g}� fdd�|D �}� j|�}� j|||�\}}|rl� j| � j | f} n� j| � j | f} |j
| d�}
|�r^|j t |�|�} t
� fdd�|D ��} t
� fdd�|D ��}
xJtt||��D ]8\}\}}|
j| |d| �f �}� j|||d |
d
� q�W |j| dd�| d�f �}� j||d d | d
� |
jd
dddd� n�t
� fdd�||g D ��}d tj
|� }|j d|�} d}x^t||�D ]P\}}� j|�}|
j| d|||d �f �}� j||||d
� ||d 7 }�q�W |j| 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 Nc s g | ]}� j |��qS r )�_conv)r r )r r r
r = s z%TablePlotter.plot.<locals>.<listcomp>)�figsizec 3 s | ]}� j |�d V qdS )r N)r )r r )r r r
r L s z$TablePlotter.plot.<locals>.<genexpr>c 3 s | ]}� j |�d V qdS )r N)r )r r )r r r
r M s g �?)�title�heightZResultg�������?g�������?g�������?gffffff�?)�topZbottomr r c 3 s | ]}� j |�d V qdS )r N)r )r r )r r r
r V s r g333333�?)Zmatplotlib.gridspec�gridspec�matplotlib.pyplot�pyplot�
isinstance�listr% r$ r r �figureZGridSpec�lenr � enumerate�zipZ add_subplot� _make_tableZsubplotZsubplots_adjust�npr )r r r �labelsr! r* �pltr# r" r& ZfigZgsZ
max_left_colsZ
max_left_rows�ir �label�axZmax_rowsr( �spr )r r
�plot, sD
zTablePlotter.plotc C s: t |tj�r,|jdkr$|jdd�}n|j� }|jd�}|S )zF
Convert each input to appropriate for table outplot.
N� )�name�NaN)r- �pdZSeriesr= Zto_frameZfillna)r �datar r r
r% g s
zTablePlotter._convc s� � j � � � jj}|dkr*� jdd� j� n.x,t|�D ] }� j|d|� �� jj|�� q4W � jj}|dkr�� jjd�}� fdd�td|�D �}tj|�}|j� _tj |� g�� |� _� S )Nr r �Indexc s g | ]}� j j|�j�qS r )r �_get_level_valuesZ_values)r r7 )r@ r r
r � s z.TablePlotter._insert_index.<locals>.<listcomp>)
�copyr r �insert�rangerB r r? � DataFrame�concat)r r@ � idx_nlevelsr7 � col_nlevelsr �valuesZcol_dfr )r@ r
�
_insert_indexs s
zTablePlotter._insert_index)r' r( c
C s |d kr|j d� d S dd lj}|jj}|jj}| j|�}|j||dd�}|j| j � |d krndt
|�d }|j � } xp| d j � D ]`\\}
} } | d kr�| j d� n6|
|k r�| |k r�| j d� n|
|k s�| |k r�| j
d� | j|� q�W |j|| j d � |jd
� d S ) NFr � )�locg �?r Zcelldz#AAAAAA)�size�off���)Z set_visibleZpandas.plotting�plottingr r r rK �tableZ set_fontsizer
r0 Z
properties�itemsZ
set_facecolorZ
set_height� set_title�axis)
r r9 r r' r( rQ rH rI �tb�props�r�cZcellr r r
r3 � s,
zTablePlotter._make_table)r r r )NT)N)�__name__�
__module__� __qualname__�__doc__�floatr r? rF r �intr r$ �boolr; r% rK �strr r3 r r r r
r s
; r �__main__�
� � � � � � � � )�A�B�C)rl rn �df1�df2T)r5 r! )�X�Zr )rU Frl rm rn � � � � � � � rL )Zv1Zv2)r �df3)r5 )r rl )r rm )r rn )rs rl )rs rm )rs rn )r rl )r rm )Ztypingr r Znumpyr4 Zpandasr? r rZ r+ r, r6 �prF ro rp r; rG �showrz Z
MultiIndex� from_tuples�idxr r r r r r
�<module> s, !
$$$, |