3
���hw2 � @ sT d Z ddlmZmZmZmZ ddlZddlm Z ddl
m Z m Z G dd� de �Z
dS )z-
Module for formatting output data in Latex.
� )�IO�List�Optional�TupleN)�
ABCMultiIndex)�DataFrameFormatter�TableFormatterc
@ s� e Zd ZdZd eee eeee eee ee d�dd�Ze e dd�dd �Z
e e e e e d
�d d �Z
e e e e e eed
f e e d�dd�Ze e e e dd�dd�Zed�dd�Zdd� Zed�dd�Zedd � �ZdS ) �LatexFormattera�
Used to render a DataFrame to a LaTeX tabular/longtable environment output.
Parameters
----------
formatter : `DataFrameFormatter`
column_format : str, default None
The columns format as specified in `LaTeX table format
<https://en.wikibooks.org/wiki/LaTeX/Tables>`__ e.g 'rcl' for 3 columns
longtable : boolean, default False
Use a longtable environment instead of tabular.
See Also
--------
HTMLFormatter
NF)� formatter�
column_format� longtable� multicolumn�multicolumn_format�multirow�caption�labelc C sR || _ | j j| _| j j| _|| _|| _|| _|| _|| _|| _|| _ | j j
| _
d S )N) �fmt�frame� bold_rowsr r r
r r r r �escape) �selfr
r r r
r r r r � r �9/tmp/pip-build-5_djhm0z/pandas/pandas/io/formats/latex.py�__init__ s
zLatexFormatter.__init__)�buf�returnc s� t | jj�dks t | jj�dkrPdt| j�j� d| jj� d| jj� �}|gg}n
| jj� }dd� }| jjovt| jjt ��r | jjj
d| jj | jj | jj
d�}d d
� ��fd d �|D �}| jjj� � fd
d�|D �}| jjj}t|��rdd� |D �}||| jjjd d� �< ||dd� }| jdk�rf| jjj}djt||��} | jj�r�d| jjj }
|
| } n(t| jt��s�tdt| �� ���n| j} | j�r�| j|| � n | j|| � |jd� | jjj�| jjj� � } | jj �r�| jj �r�| d7 } t t |� �} g | _ �x�t | �D �]t\}
}|
| k�r�| jj!�r�|jd� | j�r�|jd� |jd� |jdt |�� d�� |jd� |jd� |jd� |jd � | j"�r�d d� |D �}nd d� |D �}| j#�r�| jj�r܇fd d�t |�D �}|
� k �r| jj!�r| j$�r| j%|��}|
| k�r:| jj�r:| j&�r:�dk�r:| j'|�|
| �}|jd j|�� |jd!� | j&�r |
t | �d k �r | j(||
t |�� �q W | j�r�| j)|� n
| j*|� dS )"zp
Render a DataFrame to a LaTeX tabular, longtable, or table/tabular
environment output.
r zEmpty z
Columns: z
Index: c S s t | jtj�rdS dS d S )N�r�l)�
issubclass�type�np�number)Zdtyper r r � get_col_typeF s z1LatexFormatter.write_result.<locals>.get_col_typeF)Zadjoin�sparsify�names�na_repc s< xt | �D ]
� � r
P q
W | d g� fdd�| dd � D � S )Nr c s g | ]}|r|n
d t � � �qS )� )�len)�.0�i)�padr r �
<listcomp>[ s zDLatexFormatter.write_result.<locals>.pad_empties.<locals>.<listcomp>� )�reversed)�xr )r* r � pad_emptiesW s z0LatexFormatter.write_result.<locals>.pad_emptiesc 3 s | ]}� |�V qd S )Nr )r( r) )r/ r r � <genexpr>] s z.LatexFormatter.write_result.<locals>.<genexpr>c s& g | ] }d t |d � g� | �qS )r& r, ���)r' )r( r) )�clevelsr r r+ a s z/LatexFormatter.write_result.<locals>.<listcomp>c S s g | ]}|r|nd �qS )z{}r )r( r) r r r r+ f s r, N� r z*column_format must be str or unicode, not z \toprule
z \midrule
z \endhead
z
\multicolumn{z"}{r}{{Continued on next page}} \\
z
\endfoot
z \bottomrule
z
\endlastfoot
c S sp g | ]h}|rh|d krh|j dd�j dd�j dd�j dd�j d d
�j d d �j d
d�j dd�j dd�j dd�nd �qS )z{}�\z\textbackslash �_z\_�%z\%�$z\$�#z\#�{z\{�}z\}�~z\textasciitilde �^z\textasciicircum �&z\&)�replace)r( r. r r r r+ � s c S s g | ]}|r|nd �qS )z{}r )r( r. r r r r+ � s c s4 g | ],\}}|� k r,|j � dkr,d|� d�n|�qS )r3 �{}z\textbf{r: )r3 r? )�strip)r( �jr. )�ilevelsr r r+ � s z & z \\
)+r' r �columns�indexr �__name__r Z_to_str_columns�
isinstancer �formatr# Zhas_index_namesr% �nlevelsr$ �anyr �dtypesZ_values�join�map�str�AssertionErrorr �_write_longtable_begin�_write_tabular_begin�writeZshow_index_names�list�zip�clinebuf� enumerate�headerr r r
�_format_multicolumnr �_format_multirow� _print_cline�_write_longtable_end�_write_tabular_end)r r Z info_lineZstrcolsr" �outZcnamesZ new_namesrJ r Z index_formatrH Zstrrowsr) �rowZcrowr )r2 rB r/ r � write_result6 s� &
&
zLatexFormatter.write_result)r] rB r c sr t |d|� ��d�d� � ���fdd�}x8||d� D ](}|j� rX� rN|� |� d�q8�d7 �q8W � rn|� �S )z�
Combine columns belonging to a group to a single multicolumn entry
according to self.multicolumn_format
e.g.:
a & & & b & c &
will become
\multicolumn{3}{l}{a} & b & \multicolumn{2}{l}{c}
Nr, r3 c s> �dkr0�j d�d�d�j� d� j� � d�� n
�j � � d S )Nr, z
\multicolumn{�dz}{r: )�appendr r@ r )�coltext�ncol�row2r r r �
append_col� s $z6LatexFormatter._format_multicolumn.<locals>.append_col)rR r@ )r r] rB rd �cr )ra rb rc r r rW � s
z"LatexFormatter._format_multicolumn.)r] rB r) �rowsr c C s� x�t |�D ]�}|| j� r
d}x0||d d� D ] }|| j� sJ|d7 }q0P q0W |dkr
d|d�d|| j� � d�||< | jj|| d |d g� q
W |S )z�
Check following rows, whether row should be a multirow
e.g.: becomes:
a & 0 & \multirow{2}{*}{a} & 0 &
& 1 & & 1 &
b & 0 & \cline{1-2}
b & 0 &
r, Nz
\multirow{r_ z}{*}{r: )�ranger@ rT r` )r r] rB r) rf rA Znrowr r r r rX � s
z LatexFormatter._format_multirow)r r) �icolr c sV x:| j D ]0}|d � kr|jd|d d�d|d�d�� qW � fdd�| j D �| _ d S )
zB
Print clines after multirow-blocks are finished.
r z\cline{r, r_ �-z}
c s g | ]}|d � kr|�qS )r r )r( r. )r) r r r+ s z/LatexFormatter._print_cline.<locals>.<listcomp>N)rT rQ )r r r) rh Zclr )r) r rY s $zLatexFormatter._print_cline)r c C s~ | j dk s| jdk rh| j dkr$d}nd| j � d�}| jdkrBd}nd| j� d�}|jd|� |� d�� n |jd|� d �� dS )
a�
Write the beginning of a tabular environment or
nested table/tabular environments including caption and label.
Parameters
----------
buf : string or file handle
File path or object. If not specified, the result is returned as
a string.
column_format : str
The columns format as specified in `LaTeX table format
<https://en.wikibooks.org/wiki/LaTeX/Tables>`__ e.g 'rcl'
for 3 columns
Nr3 z
\caption{r: z
\label{z\begin{table}
\centering�
z\begin{tabular}{z}
)r r rQ )r r r Zcaption_Zlabel_r r r rP s
z#LatexFormatter._write_tabular_beginc C s8 |j d� |j d� | jdk s(| jdk r4|j d� n dS )a
Write the end of a tabular environment or nested table/tabular
environment.
Parameters
----------
buf : string or file handle
File path or object. If not specified, the result is returned as
a string.
z \bottomrule
z\end{tabular}
Nz \end{table}
)rQ r r )r r r r r r[ 2 s
z!LatexFormatter._write_tabular_endc C sv |j d|� d�� | jdk s&| jdk rr| jdkr2n|j d| j� d�� | jdkrRn|j d| j� d�� |j d� n dS )a�
Write the beginning of a longtable environment including caption and
label if provided by user.
Parameters
----------
buf : string or file handle
File path or object. If not specified, the result is returned as
a string.
column_format : str
The columns format as specified in `LaTeX table format
<https://en.wikibooks.org/wiki/LaTeX/Tables>`__ e.g 'rcl'
for 3 columns
z\begin{longtable}{z}
Nz \caption{r: z\label{z\\
)rQ r r )r r r r r r rO E s
z%LatexFormatter._write_longtable_beginc C s | j d� dS )z�
Write the end of a longtable environment.
Parameters
----------
buf : string or file handle
File path or object. If not specified, the result is returned as
a string.
z\end{longtable}
N)rQ )r r r r rZ h s z#LatexFormatter._write_longtable_end)NFFNFNN)rE �
__module__� __qualname__�__doc__r r rM �boolr r r^ r �intrW r rX rY rP r[ rO � staticmethodrZ r r r r r
s$ . )
"#r )rm Ztypingr r r r Znumpyr Zpandas.core.dtypes.genericr Zpandas.io.formats.formatr r r r r r r �<module> s
|