U
|
¬ý°d ã @ st d Z ddlmZ ddlmZ ddlmZmZ ddlZ ddl
|
m Z erPddl m Z G dd d Zd
|
d d
|
d d dZdS )z;
|
Module for formatting output data in console (to string).
|
é )Ú annotations)Úget_terminal_size)Ú TYPE_CHECKINGÚIterableN)Ú pprint_thing)ÚDataFrameFormatterc @ sØ e Zd ZdZd(ddddddZd d
|
d d Zd d
|
ddZd d
|
ddZed d
|
ddZ edd
|
ddZ
|
d d dddZ edd
|
dd Z d dd d d d Z d dd d d d!Zd"d d#d$d%Zd d dd&d'ZdS ))ÚStringFormatterz3Formatter for string representation of a dataframe.Nr z
|
int | NoneÚNone)ÚfmtÚ
|
line_widthÚreturnc C s || _ |j| _|j| _|| _d S ©N)r
|
ÚadjÚframer )Úselfr
|
r © r úOd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/io/formats/string.pyÚ__init__ s zStringFormatter.__init__Ústr)r c C s& | ¡ }| jjr"d || jjg¡}|S )NÚ )Ú_get_string_representationr
|
Zshould_show_dimensionsÚjoinZdimensions_info)r Útextr r r Ú to_string s zStringFormatter.to_stringzlist[list[str]]c C s | j ¡ }| j jr | |¡}|S r )r
|
Z get_strcolsZ is_truncatedÚ_insert_dot_separators©r Ústrcolsr r r Ú _get_strcols# s
|
|
z StringFormatter._get_strcolsc C sL | j jjr| jS | ¡ }| jd kr2| jjd| S | jrB| |¡S |
|
|¡S )Né )r ) r
|
r ÚemptyÚ_empty_info_liner r r ÚadjoinÚ_need_to_wrap_aroundÚ_join_multilineÚ _fit_strcols_to_terminal_widthr r r r r ) s
|
|
|
z*StringFormatter._get_string_representationc C s. dt | jj dt| jj dt| jj S )NzEmpty z
|
|
Columns: z
|
Index: )Útyper Ú__name__r ÚcolumnsÚindex©r r r r r 8 s ,ÿz StringFormatter._empty_info_lineÚboolc C s t | jjd kp| jjdkS )Nr )r* r
|
Zmax_colsr) r r r r" @ s z$StringFormatter._need_to_wrap_around)r r c C sD | j | j j¡}t|}| j jr,| ||¡}| j jr@| ||¡}|S r )r
|
Z_get_formatted_indexÚtr_frameÚlenÚis_truncated_horizontallyÚ _insert_dot_separator_horizontalÚis_truncated_verticallyÚ _insert_dot_separator_vertical)r r Z str_indexÚ index_lengthr r r r D s z&StringFormatter._insert_dot_separatorsÚintc C s | j jr| j jd S | j jS )Nr )r
|
r( Z
|
tr_col_numr) r r r Ú_adjusted_tr_col_numP s z$StringFormatter._adjusted_tr_col_num)r r1 r c C s | | jdg| ¡ |S )Nz ...)Úinsertr3 )r r r1 r r r r. T s z0StringFormatter._insert_dot_separator_horizontalc C s¼ |t | jj }| jj}t|D ]\}}| j || ¡}| jjrL|| jk}nd}|dks\|rbd} nd} |dkr|| jjr|d}
|
n|rd}d}
|
nd}
|
| jj | g||
|
d d } |
|
|| | ¡ q |S )
|
NFé z...z..r Úlefté Úright)Úmode) r, r
|
r+ Z
|
tr_row_numÚ enumerater r- r3 r( Zjustifyr4 ) r r r1 Z n_header_rowsZrow_numÚixÚcolZcwidthZ
|
is_dot_colÚdotsZdot_modeZdot_strr r r r0 Z s& z.StringFormatter._insert_dot_separator_verticalzIterable[list[str]])Ú strcols_inputr c sZ j }d}t|} jjrH| d¡}|t fdd|D ¡ ¡ | 8 } fdd|D }|d k sftt ||}t
|
|} jj r jj d k st jj d } n
|
t
|
j } g }
|
d} t|D ]\} } || |
} jjrà| d|¡ |dkr2| t
|
|kr"| |d k r"| dgdg| d ¡ n| dg| ¡ |
|
jj|f| ¡ | } q¸d |
|
¡S )
|
Nr r c s g | ]} j |¡qS r ©r r, ©Ú.0Úxr) r r Ú
|
<listcomp> s z3StringFormatter._join_multiline.<locals>.<listcomp>c s8 g | ]0}t |d kr0t fdd|D ¡ ¡ nd qS )r c s g | ]} j |¡qS r r? r@ r) r r rC s z>StringFormatter._join_multiline.<locals>.<listcomp>.<listcomp>)r, ÚnpÚarrayÚmax)rA r< r) r r rC s ÿz \z ú z
|
|
)r Úlistr
|
r( ÚpoprD rE rF ÚAssertionErrorÚ_binifyr, r/ Zmax_rows_fittedr r: r4 Úappendr r! r )r r> ZlwidthÚ adjoin_widthr ÚidxZ
|
col_widthsZcol_binsZnbinsZnrowsZstr_lstÚstartÚiÚendÚrowr r) r r# x s:
|
$
|
þ
|
|
|
z StringFormatter._join_multilinec s ddl m | jjd| d¡} |j ¡ ¡ }t \}}|| }|d } fdd|D }t|} d}
|
|dkrÆ| dkrÆ|
|
d7 }
|
t | d } |j
|
| } || } || d 8 }| | ¡}t|} qp| | j j
|
}t|d}|| j _ | j ¡ | ¡ }| jjd | S )
|
Nr ©ÚSeriesr Ú
|
c s g | ]} | t¡ ¡ qS r )Úapplyr, rF )rA ZelerS r r rC © s zBStringFormatter._fit_strcols_to_terminal_width.<locals>.<listcomp>é )r )r )ZpandasrT r r! Úsplitr r, rF r Úroundr( Zdropr
|
Úmax_cols_fittedÚtruncater )r r ÚlinesÚmax_lenÚwidthÚ_ZdifZadj_difZcol_lensZn_colsÚcounterZmidZmid_ixZcol_lenrZ r rS r r$ s.
|
|
|
|
|
|
z.StringFormatter._fit_strcols_to_terminal_width)N)r& Ú
|
__module__Ú __qualname__Ú__doc__r r r r Úpropertyr r" r r3 r. r0 r# r$ r r r r r s 'r z list[int]r2 )Úcolsr r c
|
C s d}g }d}t | d }t| D ]\\}}|| }||7 }||krV|d |koR|dk} n|d |koh|dk} | r | |¡ |}q | t | ¡ |S )Nr r rW )r, r: rL )
|
re r rM ZbinsZ
|
curr_widthZ i_last_columnrP ÚwZ
|
w_adjoinedÚwrapr r r rK Ã s
|
rK )rc Ú
|
__future__r Úshutilr Útypingr r ÚnumpyrD Zpandas.io.formats.printingr Zpandas.io.formats.formatr r rK r r r r Ú<module> s 0
|
