o
    d<                    @  sZ  U d Z ddlmZ ddlmZ ddlmZmZ ddlm	Z	 ddl
mZ ddlmZ ddlZddlZdd	lmZ dd
lmZmZmZmZmZmZmZmZmZmZmZmZ ddlm Z  ddl!Z"ddl#m$Z$m%Z% ddl&m'Z' ddl(m)Z) ddl*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;m<Z<m=Z=m>Z> ddl?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZK ddlLmMZM ddlNmOZOmPZP ddlQmRZRmSZSmTZT ddlUmVZV ddlWmXZX ddlYmZ  m[Z\ ddl]m^Z^ ddl_m`Z`maZambZbmcZc ddldmeZe ddlfmgZg ddlhmiZi ddljmkZkmlZl ddlmmnZn er'ddlompZpmqZq d Zrd!esd"< d#Ztd$Zud!esd%< G d&d' d'ZvG d(d) d)ZwG d*d+ d+ZxG d,d- d-exZydd/d0Zzdd2d3Z{dd4d5Z|G d6d7 d7Z}G d8d9 d9Z~		ddd@dAZe	dddCdDZ		E			F	G	H		ddd[d\ZG d]d^ d^ZG d_d` d`eZG dadb dbeZG dcdd ddeZG dedf dfeZddidjZddmdnZdddsdtZ	o	dddvdwZ	dddzd{Zdd}d~ZG dd deZG dd deZ	o	ddddZ	F		ddddZddddZdddZ	GddddZdddZG dd dZddddZ	ddddZdddZdS )z
Internal module for formatting output data in csv, html, xml,
and latex files. This module also applies to display formatting.
    )annotations)contextmanager)
QUOTE_NONEQUOTE_NONNUMERIC)Decimal)partial)StringION)get_terminal_size)IOTYPE_CHECKINGAnyCallableFinal	GeneratorHashableIterableListMappingSequencecast)east_asian_width)
get_option
set_option)lib)NA)NaT	Timedelta	Timestampget_unit_from_dtypeiNaTperiods_per_day)NaTType)	ArrayLikeAxesColspaceArgTypeColspaceTypeCompressionOptionsFilePathFloatFormatTypeFormattersType
IndexLabelStorageOptionsWriteBuffer)is_categorical_dtypeis_complex_dtypeis_datetime64_dtypeis_extension_array_dtypeis_floatis_float_dtype
is_integeris_integer_dtypeis_list_likeis_numeric_dtype	is_scalaris_timedelta64_dtype)DatetimeTZDtype)isnanotna)CategoricalDatetimeArrayTimedeltaArray)StringDtype)PandasObject)extract_array)Index
MultiIndexPeriodIndexensure_index)DatetimeIndex)TimedeltaIndex)concat)check_parent_directorystringify_path)printing)	DataFrameSeriesa	  
        Parameters
        ----------
        buf : str, Path or StringIO-like, optional, default None
            Buffer to write to. If None, the output is returned as a string.
        columns : sequence, optional, default None
            The subset of columns to write. Writes all columns by default.
        col_space : %(col_space_type)s, optional
            %(col_space)s.
        header : %(header_type)s, optional
            %(header)s.
        index : bool, optional, default True
            Whether to print index (row) labels.
        na_rep : str, optional, default 'NaN'
            String representation of ``NaN`` to use.
        formatters : list, tuple or dict of one-param. functions, optional
            Formatter functions to apply to columns' elements by position or
            name.
            The result of each function must be a unicode string.
            List/tuple must be of length equal to the number of columns.
        float_format : one-parameter function, optional, default None
            Formatter function to apply to columns' elements if they are
            floats. This function must return a unicode string and will be
            applied only to the non-``NaN`` elements, with ``NaN`` being
            handled by ``na_rep``.

            .. versionchanged:: 1.2.0

        sparsify : bool, optional, default True
            Set to False for a DataFrame with a hierarchical index to print
            every multiindex key at each row.
        index_names : bool, optional, default True
            Prints the names of the indexes.
        justify : str, default None
            How to justify the column labels. If None uses the option from
            the print configuration (controlled by set_option), 'right' out
            of the box. Valid values are

            * left
            * right
            * center
            * justify
            * justify-all
            * start
            * end
            * inherit
            * match-parent
            * initial
            * unset.
        max_rows : int, optional
            Maximum number of rows to display in the console.
        max_cols : int, optional
            Maximum number of columns to display in the console.
        show_dimensions : bool, default False
            Display DataFrame dimensions (number of rows by number of columns).
        decimal : str, default '.'
            Character recognized as decimal separator, e.g. ',' in Europe.
    r   common_docstring)leftrightcenterjustifyzjustify-allstartendinheritzmatch-parentinitialunsetz
        Returns
        -------
        str or None
            If buf is None, returns the result as a string. Otherwise returns
            None.
    return_docstringc                   @  s>   e Zd Z				ddddZdddZdddZdddZdS )CategoricalFormatterNTNaNcategoricalr<   bufIO[str] | Nonelengthboolna_repstrfooterreturnNonec                 C  s8   || _ |d ur	|ntd| _|| _|| _|| _t| _d S N )r[   r   r\   r`   r^   rb   r   quoting)selfr[   r\   r^   r`   rb    ri   L/app/.heroku/python/lib/python3.10/site-packages/pandas/io/formats/format.py__init__   s   
zCategoricalFormatter.__init__c                 C  sP   d}| j r|r|d7 }|dt| j 7 }| j }|r |d7 }||7 }t|S )Nrf   , Length: 
)r^   lenr[   _repr_categories_infora   )rh   rb   
level_infori   ri   rj   _get_footer   s   
z CategoricalFormatter._get_footer	list[str]c                 C  s   t | j d d | j| jdS )N)float_formatr`   rg   )format_arrayr[   _internal_get_valuesr`   rg   rh   ri   ri   rj   _get_formatted_values   s   z*CategoricalFormatter._get_formatted_valuesc                 C  s|   | j }t|dkr| jr|  S dS |  }dd |D }d|}d| d g}| jr7|  }|r7|| td|S )	Nr   rf   c                 S  s   g | ]}|  qS ri   )strip.0iri   ri   rj   
<listcomp>       z2CategoricalFormatter.to_string.<locals>.<listcomp>rl   []rn   )r[   ro   rb   rr   rx   joinappendra   )rh   r[   
fmt_valuesvaluesresultrb   ri   ri   rj   	to_string   s   

zCategoricalFormatter.to_string)NTrZ   T)r[   r<   r\   r]   r^   r_   r`   ra   rb   r_   rc   rd   rc   ra   rc   rs   )__name__
__module____qualname__rk   rr   rx   r   ri   ri   ri   rj   rY      s    

	rY   c                   @  s^   e Zd Z										d'd(ddZd)ddZd*ddZd+d d!Zd,d#d$Zd*d%d&ZdS )-SeriesFormatterNTrZ   FseriesrM   r\   r]   r^   
bool | strheaderr_   indexr`   ra   namert   
str | Nonedtypemax_rows
int | Nonemin_rowsrc   rd   c                 C  st   || _ |d ur	|nt | _|| _|| _|| _|| _|| _|
| _|| _	|d u r*t
d}|| _|	| _t | _|   d S )Ndisplay.float_format)r   r   r\   r   r`   r   r^   r   r   r   r   rt   r   get_adjustmentadj_chk_truncate)rh   r   r\   r^   r   r   r`   r   rt   r   r   r   ri   ri   rj   rk   	  s   zSeriesFormatter.__init__c                 C  s   |  | j }| j}|ot| j|k}| j}|rItt|}|r"t||}|dkr0|}|jd | }n|d }t|jd | |j| d  f}|| _	nd | _	|| _
|| _d S )N      )r   r   ro   r   r   intminilocrH   
tr_row_num	tr_seriesis_truncated_vertically)rh   r   r   r   r   row_numri   ri   rj   r   )  s$   

"
zSeriesFormatter._chk_truncatec                 C  s<  | j j}d}t| j jdd d ur&t| j jtttfsJ |d| j jj 7 }| jdurC|d urC|r5|d7 }t	j
|dd}|d| 7 }| jd	u sP| jd
kr`| jr`|rV|d7 }|dt| j  7 }| jdur| jd urt| jjdd }|r|rz|d7 }|dt	
| 7 }t| jjr| jj }|r|d7 }||7 }t|S )Nrf   freqzFreq: Frl   	rn   )escape_charszName: Ttruncaterm   r   zdtype: rn   )r   r   getattrr   
isinstancerF   rD   rG   ZfreqstrrK   pprint_thingr^   r   ro   r   r   r-   _valuesrp   ra   )rh   r   rb   Zseries_nameZ
dtype_namerq   ri   ri   rj   rr   D  s@   
zSeriesFormatter._get_footertuple[list[str], bool]c                 C  sX   | j j}t|trtdd |jD }|jdd}||fS |jd u}|jdd}||fS )Nc                 s  s    | ]}|V  qd S Nri   r{   r   ri   ri   rj   	<genexpr>q  s    z7SeriesFormatter._get_formatted_index.<locals>.<genexpr>T)names)r   )r   r   r   rC   anyr   formatr   )rh   r   have_header	fmt_indexri   ri   rj   _get_formatted_indexm  s   

z$SeriesFormatter._get_formatted_indexrs   c                 C  s   t | jjd | j| j| jdS )N)rt   r`   leading_space)ru   r   r   rt   r`   r   rw   ri   ri   rj   rx   x  s   z%SeriesFormatter._get_formatted_valuesc                 C  s,  | j }|  }t|dkrt| jj d| dS |  \}}|  }| jr_d}| j	}t
t|}| j||d  }|dkrAd}	nd}	| jj|	g|dd	d }	||| |	 ||d d
 | jrs| jjdg|dd  |gR  }
n| jd|}
| jr|r|d d |
 }
|r|
d| 7 }
td
|
S )Nr   z([], )r      ...z..rQ   moderf   rn   )r   rr   ro   typer   r   r   rx   r   r   r   r   r   rR   insertr   adjoinr   ra   r   )rh   r   rb   r   r   r   Zn_header_rowsr   widthZdot_strr   ri   ri   rj   r     s2   
"
zSeriesFormatter.to_string)
NTTTrZ   FNTNN)r   rM   r\   r]   r^   r   r   r_   r   r_   r`   ra   r   r_   rt   r   r   r_   r   r   r   r   rc   rd   rc   rd   r   )rc   r   r   )	r   r   r   rk   r   rr   r   rx   r   ri   ri   ri   rj   r     s"    
 

)
	r   c                   @  s6   e Zd ZdddZddd	ZddddZdddZdS )TextAdjustmentrc   rd   c                 C  s   t d| _d S )Nzdisplay.encoding)r   encodingrw   ri   ri   rj   rk     s   zTextAdjustment.__init__textra   r   c                 C  s   t |S r   ro   rh   r   ri   ri   rj   ro     s   zTextAdjustment.lenrP   textsr   max_lenr   rs   c                 C  s   t j|||dS )Nr   )rK   rR   rh   r   r   r   ri   ri   rj   rR        zTextAdjustment.justifyspacec                 O  s"   t j|g|R | j| jd|S )N)ZstrlenZjustfunc)rK   r   ro   rR   )rh   r   listskwargsri   ri   rj   r     s   zTextAdjustment.adjoinNr   r   ra   rc   r   rP   )r   r   r   r   r   ra   rc   rs   )r   r   rc   ra   )r   r   r   rk   ro   rR   r   ri   ri   ri   rj   r     s
    

r   c                      s6   e Zd Zd fddZddd	Z	
ddddZ  ZS )EastAsianTextAdjustmentrc   rd   c                   s6   t    tdrd| _nd| _dddddd| _d S )Nz!display.unicode.ambiguous_as_wider   r   )ZNaNWFH)superrk   r   ambiguous_width_EAW_MAPrw   	__class__ri   rj   rk     s
   
z EastAsianTextAdjustment.__init__r   ra   r   c                   s(   t |ts	t|S t fdd|D S )zN
        Calculate display width considering unicode East Asian Width
        c                 3  s$    | ]} j t| jV  qd S r   )r   getr   r   )r{   crw   ri   rj   r     s    
z.EastAsianTextAdjustment.len.<locals>.<genexpr>)r   ra   ro   sumr   ri   rw   rj   ro     s
   
zEastAsianTextAdjustment.lenrP   r   Iterable[str]r   r   rs   c                   sT   fdd |dkr fdd|D S |dkr! fdd|D S  fdd|D S )	Nc                   s     |  t |  S r   r   )t)r   rh   ri   rj   _get_pad     z1EastAsianTextAdjustment.justify.<locals>._get_padrO   c                      g | ]	}|  |qS ri   )ljustr{   xr   ri   rj   r}         z3EastAsianTextAdjustment.justify.<locals>.<listcomp>rQ   c                   r   ri   )rQ   r   r   ri   rj   r}     r   c                   r   ri   )rjustr   r   ri   rj   r}     r   ri   r   ri   )r   r   rh   rj   rR     s   zEastAsianTextAdjustment.justifyr   r   r   )r   r   r   r   r   ra   rc   rs   )r   r   r   rk   ro   rR   __classcell__ri   ri   r   rj   r     s
    
r   rc   c                  C  s   t d} | r	t S t S )Nz display.unicode.east_asian_width)r   r   r   )Zuse_east_asian_widthri   ri   rj   r     s   r   dict[str, Any]c                  C  sL   ddl m}  tdr|  \}}nd}tdtdtdtdtd	|d
S )a  Get the parameters used to repr(dataFrame) calls using DataFrame.to_string.

    Supplying these parameters to DataFrame.to_string is equivalent to calling
    ``repr(DataFrame)``. This is useful if you want to adjust the repr output.

    .. versionadded:: 1.4.0

    Example
    -------
    >>> import pandas as pd
    >>>
    >>> df = pd.DataFrame([[1, 2], [3, 4]])
    >>> repr_params = pd.io.formats.format.get_dataframe_repr_params()
    >>> repr(df) == df.to_string(**repr_params)
    True
    r   )consolezdisplay.expand_frame_reprNdisplay.max_rowsdisplay.min_rowszdisplay.max_columnsdisplay.max_colwidthdisplay.show_dimensions)r   r   max_colsZmax_colwidthshow_dimensions
line_width)pandas.io.formatsr   r   Zget_console_size)r   r   _ri   ri   rj   get_dataframe_repr_params  s   r   c                  C  sN   t  \} }tddkr|ntd}tddkr|ntd}dd||tddS )a  Get the parameters used to repr(Series) calls using Series.to_string.

    Supplying these parameters to Series.to_string is equivalent to calling
    ``repr(series)``. This is useful if you want to adjust the series repr output.

    .. versionadded:: 1.4.0

    Example
    -------
    >>> import pandas as pd
    >>>
    >>> ser = pd.Series([1, 2, 3, 4])
    >>> repr_params = pd.io.formats.format.get_series_repr_params()
    >>> repr(ser) == ser.to_string(**repr_params)
    True
    r   r   r   Tr   )r   r   r   r   r^   )r	   r   )r   heightr   r   ri   ri   rj   get_series_repr_params  s   


r   c                   @  s  e Zd ZdZer
endZeee 7 Z																	dtdud(d)Zdvd+d,Zedwd-d.Z	edwd/d0Z
edwd1d2Zedwd3d4Zedxd5d6Zedwd7d8Zedwd9d:Zedwd;d<Zedwd=d>Zedyd@dAZdzdBdCZd{dEdFZd|dGdHZd}dJdKZd~dMdNZddOdPZddQdRZddSdTZdwdUdVZdwdWdXZdwdYdZZdyd[d\Zdd]d^Zdd_d`Z ddadbZ!dvdcddZ"ddgdhZ#ddkdlZ$ddmdnZ%ddodpZ&ddrdsZ'dS )DataFrameFormatterz;Class for processing dataframe formatting options and data.rf   NTrZ   F.framerL   columnsSequence[Hashable] | None	col_spaceColspaceArgType | Noner   bool | Sequence[str]r   r_   r`   ra   
formattersFormattersType | NonerR   r   rt   FloatFormatType | Nonesparsifybool | Noneindex_namesr   r   r   r   r   r   decimal	bold_rowsescaperc   rd   c                 C  s   || _ | || _| || _|| _|| _|| _| || _	| 
|| _|	| _| |
| _|| _|| _|| _|| _|| _|| _|| _|| _|  | _|  | _| j | _|   t | _d S r   )r   _initialize_columnsr   _initialize_colspacer   r   r   r`   _initialize_formattersr   _initialize_justifyrR   rt   _initialize_sparsifyr  show_index_namesr  r  r  r   r   r   r   _calc_max_cols_fittedmax_cols_fitted_calc_max_rows_fittedmax_rows_fittedtr_framer   r   r   )rh   r   r   r   r   r   r`   r   rR   rt   r  r  r   r   r   r   r  r  r  ri   ri   rj   rk   4  s.   

zDataFrameFormatter.__init__list[list[str]]c                 C  s*   |   }| jr| | j}|d| |S )zP
        Render a DataFrame to a list of columns (as lists of strings).
        r   )_get_strcols_without_indexr   r   r  r   )rh   strcolsZ	str_indexri   ri   rj   get_strcolsc  s
   zDataFrameFormatter.get_strcolsc                 C  s   | j du p| j dko| jS )NTr   )r   is_truncatedrw   ri   ri   rj   should_show_dimensionso  s   
z)DataFrameFormatter.should_show_dimensionsc                 C  s   t | jp| jS r   )r_   is_truncated_horizontallyr   rw   ri   ri   rj   r  u  s   zDataFrameFormatter.is_truncatedc                 C     t | jot| j| jkS r   )r_   r  ro   r   rw   ri   ri   rj   r  y     z,DataFrameFormatter.is_truncated_horizontallyc                 C  r  r   )r_   r  ro   r   rw   ri   ri   rj   r   }  r  z*DataFrameFormatter.is_truncated_verticallyc                 C  s    dt | j dt | jj dS )Nz

[z rows x z	 columns])ro   r   r   rw   ri   ri   rj   dimensions_info  s    z"DataFrameFormatter.dimensions_infoc                 C     t | jjS r   )
_has_namesr   r   rw   ri   ri   rj   has_index_names     z"DataFrameFormatter.has_index_namesc                 C  r  r   )r  r   r   rw   ri   ri   rj   has_column_names  r!  z#DataFrameFormatter.has_column_namesc                 C     t | j| j| jfS r   )allr   r   r  rw   ri   ri   rj   show_row_idx_names     z%DataFrameFormatter.show_row_idx_namesc                 C  r#  r   )r$  r"  r  r   rw   ri   ri   rj   show_col_idx_names  r&  z%DataFrameFormatter.show_col_idx_namesr   c                 C  s   t | jpt| jt| jS r   )r   r   ro   r   rw   ri   ri   rj   max_rows_displayed  s   z%DataFrameFormatter.max_rows_displayedc                 C     |d u rt dS |S )Nzdisplay.multi_sparser   )rh   r  ri   ri   rj   r    s   z'DataFrameFormatter._initialize_sparsifyr)   c                 C  sP   |d u ri S t | jjt |kst|tr|S tdt | dt | jj d)NzFormatters length(+) should match DataFrame number of columns(r   )ro   r   r   r   dict
ValueError)rh   r   ri   ri   rj   r    s   
z)DataFrameFormatter._initialize_formattersc                 C  r)  )Nzdisplay.colheader_justifyr*  )rh   rR   ri   ri   rj   r    s   z&DataFrameFormatter._initialize_justifyrB   c                 C  s.   |d urt tt|}| j| | _|S | jjS r   )rE   r   r#   r   r   )rh   r   colsri   ri   rj   r	    s
   z&DataFrameFormatter._initialize_columnsr%   c                   s    d u ri }|S t  ttfr#d i}| fdd| jjD  |S t  trD  D ]}|| jjvr?|dkr?td| q, }|S t	| jjt	 kr_tdt	  dt	| jj dt
t| jj }|S )Nrf   c                   s   i | ]}| qS ri   ri   )r{   columnr   ri   rj   
<dictcomp>      z;DataFrameFormatter._initialize_colspace.<locals>.<dictcomp>z,Col_space is defined for an unknown column: zCol_space length(r+  r   )r   r   ra   updater   r   r   keysr-  ro   r,  zip)rh   r   r   r/  ri   r0  rj   r
    s0   

z'DataFrameFormatter._initialize_colspacec                 C  s,   |   s| jS t \}}| |r|S | jS )z%Number of columns fitting the screen.)_is_in_terminalr   r	   _is_screen_narrow)rh   r   r   ri   ri   rj   r    s   

z(DataFrameFormatter._calc_max_cols_fittedc                 C  sP   |   r t \}}| jdkr||   S | |r|}n| j}n| j}| |S )z,Number of rows with data fitting the screen.r   )r6  r	   r   _get_number_of_auxillary_rows_is_screen_short_adjust_max_rows)rh   r   r   r   ri   ri   rj   r    s   



z(DataFrameFormatter._calc_max_rows_fittedc                 C  s(   |rt | j|kr| jrt| j|}|S )zAdjust max_rows using display logic.

        See description here:
        https://pandas.pydata.org/docs/dev/user_guide/options.html#frequently-used-options

        GH #37359
        )ro   r   r   r   )rh   r   ri   ri   rj   r:    s   z#DataFrameFormatter._adjust_max_rowsc                 C  s   t | jdkp
| jdkS )z/Check if the output is to be shown in terminal.r   )r_   r   r   rw   ri   ri   rj   r6    s   z"DataFrameFormatter._is_in_terminalc                 C  s   t | jdkot| jj|kS Nr   )r_   r   ro   r   r   )rh   	max_widthri   ri   rj   r7    s   z$DataFrameFormatter._is_screen_narrowc                 C  s   t | jdkot| j|kS r;  )r_   r   ro   r   )rh   
max_heightri   ri   rj   r9    s   z#DataFrameFormatter._is_screen_shortc                 C  s:   d}d}|| }| j r|t| j 7 }| jr|d7 }|S )z?Get number of rows occupied by prompt, dots and dimension info.r   )r   ro   r  
splitlinesr   )rh   Zdot_rowZ
prompt_rowZnum_rowsri   ri   rj   r8  	  s   z0DataFrameFormatter._get_number_of_auxillary_rowsc                 C  s$   | j r|   | jr|   dS dS )zY
        Check whether the frame should be truncated. If so, slice the frame up.
        N)r  _truncate_horizontallyr   _truncate_verticallyrw   ri   ri   rj   r     s
   zDataFrameFormatter.truncatec                 C  s   | j dusJ | j d }|dkrM| jjddd|f }| jjdd| df }t||fdd| _t| jttfrLg | jd| | j| d | _ntt	| j
}| jjddd|f | _|| _dS )zRemove columns, which are not to be displayed and adjust formatters.

        Attributes affected:
            - tr_frame
            - formatters
            - tr_col_num
        Nr   r   Zaxis)r  r  r   rH   r   r   listtupler   r   r   Z
tr_col_num)rh   Zcol_numrO   rP   ri   ri   rj   r?  !  s    

z)DataFrameFormatter._truncate_horizontallyc                 C  s   | j dusJ | j d }|dkr1| jjd|ddf }| jj| dddf }t||f| _ntt| j}| jjd|ddf | _|| _dS )zRemove rows, which are not to be displayed.

        Attributes affected:
            - tr_frame
            - tr_row_num
        Nr   r   )r  r  r   rH   r   r   r   r   )rh   r   headtailri   ri   rj   r@  ;  s   

z'DataFrameFormatter._truncate_verticallyc           
   	     s  g }t  js1 js1t jD ]\}} |}t| jt j	|d j
d}|| q|S t  jrcttt  j _t jt jkrZtdt j dt j ddd  jD }n  j} jrv|D ]}|d qnt jD ]L\}}|| }tt j	|dg fd	d
|D R  } |}t| j| j
d}tt fdd
|D |}	 j
j||	 jd}|||  q{|S )Nr   )stringsrR   minimumr   zWriting z cols but got z aliasesc                 S  s   g | ]}|gqS ri   ri   )r{   labelri   ri   rj   r}   d  r2  zADataFrameFormatter._get_strcols_without_index.<locals>.<listcomp>rf   c                 3      | ]	} j |V  qd S r   r   ro   r   rw   ri   rj   r   o      z@DataFrameFormatter._get_strcols_without_index.<locals>.<genexpr>)rG  r   c                 3  rI  r   rJ  r   rw   ri   rj   r   v  rK  r   )r5   r   	enumerater  
format_col_make_fixed_widthrR   r   r   r   r   r   r   r   ra   ro   r   r-  _get_formatted_column_labelsr%  max)
rh   r  r|   r   r   str_columnsr   ZcheaderZheader_colwidthr   ri   rw   rj   r  M  sP   


z-DataFrameFormatter._get_strcols_without_indexr|   rs   c              	   C  sL   | j }| |}t|jd d |f j|| j| j| j|j	| | j
| jdS )N)rt   r`   r   r  r   )r  _get_formatterru   r   r   rt   r`   r   r   r   r  r   )rh   r|   r   	formatterri   ri   rj   rM  |  s   
zDataFrameFormatter.format_col	str | intCallable | Nonec                 C  sZ   t | jttfrt|rtt|}| j| S d S t|r&|| jvr&| j| }| j|d S r   )	r   r   rB  rC  r3   r   r   r   r   )rh   r|   ri   ri   rj   rR    s   


z!DataFrameFormatter._get_formatterc                   s   ddl m} |j}t|tr_|jddd}tt| }jj	j
}tdd |jD tt|tt|  fddttfd	d|D  }jrTt|rT||}d
d t| D }|S | }jj	}tt|tt|  fddt|D }|S )Nr   )sparsify_labelsF)r  r   c                 s  s    | ]}|j V  qd S r   )Zis_floating)r{   levelri   ri   rj   r     s    zBDataFrameFormatter._get_formatted_column_labels.<locals>.<genexpr>c                   s"   |j vr |  rsd| S |S )N )r   )r   y)need_leadsprestrict_formattingrh   ri   rj   space_format  s   zEDataFrameFormatter._get_formatted_column_labels.<locals>.space_formatc                 3  s$    | ]  fd d D V  qdS )c                   s   g | ]} |qS ri   ri   )r{   rY  )r\  r   ri   rj   r}         zMDataFrameFormatter._get_formatted_column_labels.<locals>.<genexpr>.<listcomp>Nri   )r{   )r\  r   rj   r     s   " c                 S     g | ]}t |qS ri   )rB  r   ri   ri   rj   r}     r~   zCDataFrameFormatter._get_formatted_column_labels.<locals>.<listcomp>c                   s0   g | ]\}} |s | rd | n|gqS )rX  )rR  )r{   r|   r   )rZ  rh   ri   rj   r}     s    )Zpandas.core.indexes.multirV  r   r   rC   r   rB  r5  r   dtypesr   r   levelsr,  mapr6   r  ro   rL  )rh   r   rV  r   Zfmt_columnsr`  rQ  ri   )rZ  r[  rh   r\  rj   rO    s.   

	
z/DataFrameFormatter._get_formatted_column_labelsc                   s   dd j  D  |j}|j}d}t|tr&|jjdj	|d}n	|jj	|dg} fdd|D }j
jd	g|R  d
}jrSdd  D }ndg|j }jr`|| S |S )Nc                 S  s   i | ]
\}}|t t|qS ri   )r   r   )r{   kvri   ri   rj   r1    s    z;DataFrameFormatter._get_formatted_index.<locals>.<dictcomp>	__index__F)r  r   r   rS  )r   rS  c              
     s.   g | ]}t tt|d  ddjdqS )rO   rf   r   )rR   rG  r   )rC  rN  rB  r   r   r   r   rh   ri   rj   r}     s    z;DataFrameFormatter._get_formatted_index.<locals>.<listcomp>r   rn   c                 S  r_  ri   ra   r   ri   ri   rj   r}     r~   rf   )r   itemsr   r   rR  r   rC   r   r  r%  r   r   splitr'  _get_column_name_listZnlevelsr   )rh   r   r   r   fmtr   ZadjoinedZ
col_headerri   rf  rj   r     s,   

	z'DataFrameFormatter._get_formatted_indexlist[Hashable]c                 C  sN   g }| j j}t|tr|dd |jD  |S ||jd u r!dn|j |S )Nc                 s  s     | ]}|d u r
dn|V  qd S re   ri   r   ri   ri   rj   r     s    z;DataFrameFormatter._get_column_name_list.<locals>.<genexpr>rf   )r   r   r   rC   extendr   r   r   )rh   r   r   ri   ri   rj   rj    s   
z(DataFrameFormatter._get_column_name_list)NNTTrZ   NNNNTNNNFr   FT)&r   rL   r   r   r   r   r   r   r   r_   r`   ra   r   r  rR   r   rt   r  r  r  r  r_   r   r   r   r   r   r   r   r   r  ra   r  r_   r  r_   rc   rd   )rc   r  rc   r_   r   )rc   r   )r  r  rc   r_   )r   r  rc   r)   )rR   r   rc   ra   )r   r   rc   rB   )r   r   rc   r%   )rc   r   )r   r   rc   r   r   )r|   r   rc   rs   )r|   rT  rc   rU  )r   rL   rc   r  )r   rL   rc   rs   )rc   rl  )(r   r   r   __doc__rN   rX   rk   r  propertyr  r  r  r   r  r   r"  r%  r'  r(  r  r  r  r	  r
  r  r  r:  r6  r7  r9  r8  r   r?  r@  r  rM  rR  rO  r   rj  ri   ri   ri   rj   r   .  s    
/


















/


)(r   c                   @  s   e Zd ZdZdDddZ													dEdFddZ									dGdHd!d"Z			dIdJd%d&Z			'			(	)		*				+		,	dKdLdBdCZdS )MDataFrameRendereraJ  Class for creating dataframe output in multiple formats.

    Called in pandas.core.generic.NDFrame:
        - to_csv
        - to_latex

    Called in pandas.core.frame.DataFrame:
        - to_html
        - to_string

    Parameters
    ----------
    fmt : DataFrameFormatter
        Formatter with the formatting options.
    rk  r   rc   rd   c                 C  s
   || _ d S r   )rk  )rh   rk  ri   ri   rj   rk     s   
zDataFrameRenderer.__init__NFr\   "FilePath | WriteBuffer[str] | Nonecolumn_formatr   	longtabler_   r   multicolumnmulticolumn_formatmultirowcaptionstr | tuple[str, str] | NonerH  positionc                 C  s>   ddl m} || j|||||||	|
d	}| }t|||dS )zU
        Render a DataFrame to a LaTeX tabular/longtable environment output.
        r   )LatexFormatter)rt  rs  ru  rv  rw  rx  rH  rz  r\   r   )Zpandas.io.formats.latexr{  rk  r   save_to_buffer)rh   r\   rs  rt  r   ru  rv  rw  rx  rH  rz  r{  Zlatex_formatterstringri   ri   rj   to_latex  s   zDataFrameRenderer.to_latexclassesstr | list | tuple | Nonenotebookborderint | bool | Nonetable_idrender_linksc                 C  sF   ddl m}m}	 |r|	n|}
|
| j||||d}| }t|||dS )uJ  
        Render a DataFrame to a html table.

        Parameters
        ----------
        buf : str, path object, file-like object, or None, default None
            String, path object (implementing ``os.PathLike[str]``), or file-like
            object implementing a string ``write()`` function. If None, the result is
            returned as a string.
        encoding : str, default “utf-8”
            Set character encoding.
        classes : str or list-like
            classes to include in the `class` attribute of the opening
            ``<table>`` tag, in addition to the default "dataframe".
        notebook : {True, False}, optional, default False
            Whether the generated HTML is for IPython Notebook.
        border : int
            A ``border=border`` attribute is included in the opening
            ``<table>`` tag. Default ``pd.options.display.html.border``.
        table_id : str, optional
            A css id is included in the opening `<table>` tag if specified.
        render_links : bool, default False
            Convert URLs to HTML links.
        r   )HTMLFormatterNotebookFormatter)r  r  r  r  r|  )Zpandas.io.formats.htmlr  r  rk  r   r}  )rh   r\   r   r  r  r  r  r  r  r  ZKlassZhtml_formatterr~  ri   ri   rj   to_html$  s   "zDataFrameRenderer.to_htmlr   r   c                 C  s0   ddl m} || j|d}| }t|||dS )u%  
        Render a DataFrame to a console-friendly tabular output.

        Parameters
        ----------
        buf : str, path object, file-like object, or None, default None
            String, path object (implementing ``os.PathLike[str]``), or file-like
            object implementing a string ``write()`` function. If None, the result is
            returned as a string.
        encoding: str, default “utf-8”
            Set character encoding.
        line_width : int, optional
            Width to wrap a line in characters.
        r   )StringFormatter)r   r|  )Zpandas.io.formats.stringr  rk  r   r}  )rh   r\   r   r   r  Zstring_formatterr~  ri   ri   rj   r   W  s   zDataFrameRenderer.to_string,winfer"Tstrictpath_or_buf7FilePath | WriteBuffer[bytes] | WriteBuffer[str] | Nonesepra   r   r   index_labelIndexLabel | Noner   compressionr&   rg   	quotecharlineterminator	chunksizedate_formatdoublequote
escapecharerrorsstorage_optionsr+   c                 C  s   ddl m} |du rd}t }nd}|di d|d|
d|d	|d
|d|d|d|d|d|d|d|	d|d|d|d|d| j}|  |rbt|tsXJ | }|  |S dS )z;
        Render dataframe as comma-separated file.
        r   )CSVFormatterNTFr  r  r  r   r  r  rg   r.  r  r   r  r  r  r  r  r  rS  ri   )Zpandas.io.formats.csvsr  r   rk  saver   getvalueclose)rh   r  r   r  r   r  r   r  rg   r  r  r  r  r  r  r  r  r  Zcreated_bufferZcsv_formattercontentri   ri   rj   to_csvq  s^   	
zDataFrameRenderer.to_csv)rk  r   rc   rd   )
NNFNFNFNNN)r\   rr  rs  r   rt  r_   r   r   ru  r_   rv  r   rw  r_   rx  ry  rH  r   rz  r   rc   r   )NNNFNNF)r\   rr  r   r   r  r  r  r_   r  r  r  r   r  r_   rc   r   )NNN)r\   rr  r   r   r   r   rc   r   )NNr  NNr  r  Nr  NNNTNr  N)"r  r  r   r   r  ra   r   r   r  r  r   ra   r  r&   rg   r   r  ra   r  r   r  r   r  r   r  r_   r  r   r  ra   r  r+   rc   r   )	r   r   r   ro  rk   r  r  r   r  ri   ri   ri   rj   rq    sV    
"5rq  r~  ra   r\   rr  r   r   c                 C  s\   t ||d}||  |du r| W  d   S 	 W d   dS 1 s'w   Y  dS )zQ
    Perform serialization. Write to buf or return as string if buf is None.
    )r   N)
get_bufferwriter  )r~  r\   r   fri   ri   rj   r}    s   
$r}  IGenerator[WriteBuffer[str], None, None] | Generator[StringIO, None, None]c                 c  s    | dur
t | } nt } |du rd}n	t| tstdt| dr'| V  dS t| trPtt|  t| d|dd}|V  W d   dS 1 sIw   Y  dS td)	z
    Context manager to open, yield and close buffer for filenames or Path-like
    objects, otherwise yield buf unchanged.
    Nzutf-8z1buf is not a file name and encoding is specified.r  r  rf   )r   newlinez1buf is not a file name and it has no write method)	rJ   r   r   ra   r-  hasattrrI   open	TypeError)r\   r   r  ri   ri   rj   r    s    




"r  rZ   rP   r   Tr   r   rS  rU  rt   r  r`   digitsr   r   str | int | NonerR   r  r   r  rg   fallback_formatterrs   c                 C  s   t | jrt}n0t| jtrt}n't| jrt}nt| jr!t	}nt
| js+t| jr.t}n
t| jr6t}nt}|du r>d}|du rFtd}|du rNtd}|| |||||||||	|
d}| S )a{  
    Format an array for printing.

    Parameters
    ----------
    values
    formatter
    float_format
    na_rep
    digits
    space
    justify
    decimal
    leading_space : bool, optional, default True
        Whether the array should be formatted with a leading space.
        When an array as a column of a Series or DataFrame, we do want
        the leading space to pad between columns.

        When formatting an Index subclass
        (e.g. IntervalIndex._format_native_types), we don't want the
        leading space since it should be left-aligned.
    fallback_formatter

    Returns
    -------
    List[str]
    N   r   display.precision)
r  r`   rt   rS  r   rR   r  r   rg   r  )r/   r   Datetime64Formatterr   r9   Datetime64TZFormatterr8   Timedelta64Formatterr0   ExtensionArrayFormatterr2   r.   FloatArrayFormatterr4   IntArrayFormatterGenericArrayFormatterr   
get_result)r   rS  rt   r`   r  r   rR   r  r   rg   r  Z	fmt_klassZfmt_objri   ri   rj   ru     sB   
)


ru   c                   @  sB   e Zd Z											d&d'dd Zd(d"d#Zd(d$d%ZdS ))r     NrZ   r  rP   r   Tr   r   r  r   rS  rU  r`   ra   r   rT  rt   r  rR   r  rg   r   fixed_widthr_   r   r  r  rc   rd   c                 C  sL   || _ || _|| _|| _|| _|| _|| _|| _|	| _|
| _	|| _
|| _d S r   )r   r  r`   r   rS  rt   rR   r  rg   r  r   r  )rh   r   r  rS  r`   r   rt   rR   r  rg   r  r   r  ri   ri   rj   rk   6  s   
zGenericArrayFormatter.__init__rs   c                 C  s   |   }t|| jS r   )_format_stringsrN  rR   )rh   r   ri   ri   rj   r  R  s   z GenericArrayFormatter.get_resultc                   s|  j d u rtd}|d u rtdfdd}nj }jd ur$j njd ur-j njd uo6jtk}ttjd|d  fdd}t	j
d	d
}t|tjsWtdt|t}|tjt|ttdt|jd@ }j}|d u r{| }g }t|D ]:\}	}
||	 s|sjd ur|d||
  q||	 r|||
 q|du rd}nd}||j||
d q|S )Nr   r  c                   s   t | d ddS )Nz .dr  )_trim_zeros_single_floatr^  )	precisionri   rj   <lambda>[  s    z7GenericArrayFormatter._format_strings.<locals>.<lambda>r   )r   quote_stringsc              	     s   j d ur;t| r;t| r;z | d u rW dS | tu rttW S | tu s't| r*W dS W j S  tt	fy:   Y j S w t
| trDt| S t
| trMt| S t | S )Nrd   r   )r`   r7   r:   r   ra   r   npZisnatr  r-  r   r@   r?   reprr^  )rS  rh   ri   rj   _formatm  s&   


z6GenericArrayFormatter._format_strings.<locals>._formatTZextract_numpyz<ExtensionArray formatting should use ExtensionArrayFormatterr   rA  rX  Fz{v}z {v}rd  )rt   r   rS  r  rg   r   r   rK   r   rA   r   r   r  ndarrayr  r   Z	map_inferr1   r$  r;   rC  rangero   shaper   r   rL  r   r   )rh   rt   r  r  valsinferredZis_float_typer   r   r|   rd  Ztplri   )rS  r  rh   rj   r  V  sT   


 z%GenericArrayFormatter._format_strings)r  NrZ   r  NrP   r   NTTN)r   r   r  r   rS  rU  r`   ra   r   rT  rt   r  rR   ra   r  ra   rg   r   r  r_   r   r  r  rU  rc   rd   r   )r   r   r   rk   r  r  ri   ri   ri   rj   r  5  s    
r  c                      sB   e Zd Zd fddZ		ddddZdddZdddZ  ZS )r  rc   rd   c                   sT   t  j|i | | jd ur$| jd u r&d| _t| jr(| j| _d | _d S d S d S d S NF)r   rk   rt   rS  r  callable)rh   argsr   r   ri   rj   rk     s   

zFloatArrayFormatter.__init__Nrt   r  	thresholdfloat | Noner   c                   sl   du rj rfdd nfdd jdkr$ fddn du r,S fdd	}|S )
z;Returns a function to be applied on each value to format itNc                   s$    d usJ t | r | dS jS )Nvalue)r;   r`   r  )rt   rh   ri   rj   base_formatter  s   
z<FloatArrayFormatter._value_formatter.<locals>.base_formatterc                   s   t | rt| S  jS r   )r;   ra   r`   r  rw   ri   rj   r    r   r   c                   s    |  djdS )Nr   r   )replacer  r  )r  rh   ri   rj   decimal_formatter  s   z?FloatArrayFormatter._value_formatter.<locals>.decimal_formatterc                   s*   t | rt| kr | S  dS jS )Ng        )r;   absr`   r  )r  rh   r  ri   rj   rS    s
   z7FloatArrayFormatter._value_formatter.<locals>.formatter)rt   r  )rh   rt   r  rS  ri   )r  r  rt   rh   r  rj   _value_formatter  s   
	z$FloatArrayFormatter._value_formatter
np.ndarrayc           
        st  ddd j d	ur jj jS jrtd
nd	 fdd}jd	u rDjr@jdu r5d}nd}t|jj	d}n
j}nfdd}||}jsS|S t
|dkrjtdd |D }|j	d k}nd}tjdd" tj}|dk }|dj	  k |dk@  }	W d	   n1 sw   Y  |	s|r|rjdu rd}nd}t|jj	d}||}|S )z
        Returns the float values converted into strings using
        the parameters given at initialisation, as a numpy array
        r   r"   rS  r   r`   ra   c                   s<   t | }t fddt|  | D | j}|S )Nc                   s    g | ]\}}|s |nqS ri   ri   )r{   valmrS  r`   ri   rj   r}     s    zWFloatArrayFormatter.get_result_as_array.<locals>.format_with_na_rep.<locals>.<listcomp>)r:   r  arrayr5  ravelZreshaper  )r   rS  r`   mask	formattedri   r  rj   format_with_na_rep  s   zCFloatArrayFormatter.get_result_as_array.<locals>.format_with_na_repNzdisplay.chop_thresholdc                   sx    | }jdkrdj }nj}j}t|} |||}jr:|r-t|j}nt|j}t	j
|ddS |S )NrO   rX  objectr   )r  rR   r`   r   r.   r  _trim_zeros_complexr  _trim_zeros_floatr  asarray)rt   rS  r`   r   Z
is_complexr   r  rh   r  ri   rj   format_values_with  s   
zCFloatArrayFormatter.get_result_as_array.<locals>.format_values_withTz{value: .{digits:d}f}z{value:.{digits:d}f})r  c                   s
    j |  S r   )rt   r  rw   ri   rj   r  (  s   
 z9FloatArrayFormatter.get_result_as_array.<locals>.<lambda>r   c                 s  s    | ]}t |V  qd S r   r   r   ri   ri   rj   r   4  s    z:FloatArrayFormatter.get_result_as_array.<locals>.<genexpr>   Fignoreinvalidg    .A
   z{value: .{digits:d}e}z{value:.{digits:d}e})r   r"   rS  r   r`   ra   )rS  r   r`   r  r   rt   r   r   r   r  ro   rP  r  errstater  r   )
rh   r  Zfmt_strrt   Zformatted_valuesmaxlenZtoo_longZabs_valsZhas_large_valuesZhas_small_valuesri   r  rj   get_result_as_array  sH   







z'FloatArrayFormatter.get_result_as_arrayrs   c                 C  s   t |  S r   )rB  r  rw   ri   ri   rj   r  M  s   z#FloatArrayFormatter._format_stringsr   NN)rt   r  r  r  rc   r   )rc   r  r   )r   r   r   rk   r  r  r  r   ri   ri   r   rj   r    s    
8cr  c                   @     e Zd ZdddZdS )r  rc   rs   c                   s>   | j du r
dd }ndd }| jp|  fdd| jD }|S )NFc                 S     | dj | dS )Nr  r^  r   r^  ri   ri   rj   r  T      z3IntArrayFormatter._format_strings.<locals>.<lambda>c                 S  r  )Nz dr^  r  r^  ri   ri   rj   r  V  r  c                      g | ]} |qS ri   ri   r   rS  ri   rj   r}   X  r~   z5IntArrayFormatter._format_strings.<locals>.<listcomp>)r   rS  r   )rh   Zformatter_strr   ri   r  rj   r  R  s   


z!IntArrayFormatter._format_stringsNr   r   r   r   r  ri   ri   ri   rj   r  Q      r  c                      s.   e Zd Z		dd fd
dZdddZ  ZS )r  r   Nr   3np.ndarray | Series | DatetimeIndex | DatetimeArraynat_repra   r  rd   rc   c                   $   t  j|fi | || _|| _d S r   )r   rk   r  r  )rh   r   r  r  r   r   ri   rj   rk   ]     
zDatetime64Formatter.__init__rs   c                   sZ    j }t|tst|} jdurt jr fdd|D S |jj j jd}|	 S )z&we by definition have DO NOT have a TZNc                   s   g | ]}  |qS ri   r  r   rw   ri   rj   r}   p  r]  z7Datetime64Formatter._format_strings.<locals>.<listcomp>)r`   r  )
r   r   rF   rS  r  _dataZ_format_native_typesr  r  tolist)rh   r   r   ri   rw   rj   r  h  s   
z#Datetime64Formatter._format_stringsr   N)r   r  r  ra   r  rd   rc   rd   r   r   r   r   rk   r  r   ri   ri   r   rj   r  \  
    r  c                   @  r  )r  rc   rs   c                 C  s|   t | jdd}| j}d }|d u r|jdd}t|tr | }nt|}t	||| j
| j| j| j| j| j| j| j|d}|S )NTr  )Zboxed)	rt   r`   r  r   rR   r  r   rg   r  )rA   r   rS  
_formatterr   r<   rv   r  r  ru   rt   r`   r  r   rR   r  r   rg   )rh   r   rS  r  r  r   ri   ri   rj   r  y  s,   


z'ExtensionArrayFormatter._format_stringsNr   r  ri   ri   ri   rj   r  x  r  r  percentilesnp.ndarray | Sequence[float]c                 C  sl  t | } t jdd t| rt | dkrt | dks"tdW d   n1 s,w   Y  d|  } |  t}t 	|| }t |rS|t
}dd	 |D S t | }|d dkrb|d nd}|d
 dk rpd|d
  nd}t t t t j|||dt }td|}t j| td}| |  tt
||< | |  |t
|| < dd	 |D S )a  
    Outputs rounded and formatted percentiles.

    Parameters
    ----------
    percentiles : list-like, containing floats from interval [0,1]

    Returns
    -------
    formatted : list of strings

    Notes
    -----
    Rounding precision is chosen so that: (1) if any two elements of
    ``percentiles`` differ, they remain different after rounding
    (2) no entry is *rounded* to 0% or 100%.
    Any non-integer is always rounded to at least 1 decimal place.

    Examples
    --------
    Keeps all entries different after rounding:

    >>> format_percentiles([0.01999, 0.02001, 0.5, 0.666666, 0.9999])
    ['1.999%', '2.001%', '50%', '66.667%', '99.99%']

    No element is rounded to 0% or 100% (unless already equal to it).
    Duplicates are allowed:

    >>> format_percentiles([0, 0.5, 0.02001, 0.5, 0.666666, 0.9999])
    ['0%', '50%', '2.0%', '50%', '66.67%', '99.99%']
    r  r  r   r   z/percentiles should all be in the interval [0,1]Nd   c                 S     g | ]}|d  qS %ri   rz   ri   ri   rj   r}     r~   z&format_percentiles.<locals>.<listcomp>)to_beginto_endr  c                 S  r  r  ri   rz   ri   ri   rj   r}     r~   )r  r  r  r6   r$  r-  roundastyper   isclosera   uniquefloorlog10r   Zediff1drP  Z
empty_liker  )r  Zpercentiles_round_typeZint_idxoutZunique_pctsr	  r
  precri   ri   rj   format_percentiles  s<   
"



r  2np.ndarray | DatetimeArray | Index | DatetimeIndexr_   c                 C  s~   t | ts	|  } t | ttfst| } | jd urdS | j}|tk}t| j	}t
|}t||| dk dk}|r=dS dS )NFr   T)r   rB   r  r=   rF   tzZasi8r   r   r   r    r  logical_andr   )r   
values_intconsider_valuesZresoZppd	even_daysri   ri   rj   is_dates_only  s   


r  r   r   NaTType | Timestampr  c                 C  s   | t u r|S t| S r   )r   ra   )r   r  ri   ri   rj   _format_datetime64  s   r  r  c                 C  s"   t | tr|S |r| |S | jS r   )r   r!   strftimeZ
_date_repr)r   r  r  ri   ri   rj   _format_datetime64_dateonly  s
   

r  is_dates_only_r   c                   s   | r	 fddS fddS )z]Return a formatter callable taking a datetime64 as input and providing
    a string as outputc                   s   t |  dS )N)r  r  )r  r^  r  r  ri   rj   r    s    z'get_format_datetime64.<locals>.<lambda>c                   s   t |  dS )Nr  )r  r^  r!  ri   rj   r    s    ri   )r  r  r  ri   r   rj   get_format_datetime64  s   r"  *np.ndarray | DatetimeArray | DatetimeIndexc                 C  s6   t | tjr| jdkr|  } t| }|r|pdS |S )z6given values and a date_format, return a string formatr   z%Y-%m-%d)r   r  r  ndimr  r  )r   r  idori   ri   rj   !get_format_datetime64_from_values  s   r&  c                   @  r  )r  rc   rs   c                   s>   | j t}t|}| jpt|| jd  fdd|D }|S )zwe by definition have a TZ)r  c                   r  ri   ri   r   r  ri   rj   r}   4  r~   z9Datetime64TZFormatter._format_strings.<locals>.<listcomp>)r   r  r  r  rS  r"  r  )rh   r   r%  r   ri   r  rj   r  -  s   z%Datetime64TZFormatter._format_stringsNr   r  ri   ri   ri   rj   r  ,  r  r  c                      s.   e Zd Z		dd fddZdddZ  ZS )r  r   Fr   np.ndarray | TimedeltaIndexr  ra   boxr_   rc   rd   c                   r  r   )r   rk   r  r(  )rh   r   r  r(  r   r   ri   rj   rk   :  r  zTimedelta64Formatter.__init__rs   c                   s.   | j pt| j| j| jd  fdd| jD S )N)r  r(  c                   r  ri   ri   r   r  ri   rj   r}   I  r~   z8Timedelta64Formatter._format_strings.<locals>.<listcomp>)rS  get_format_timedelta64r   r  r(  rw   ri   r  rj   r  E  s   z$Timedelta64Formatter._format_stringsr   F)r   r'  r  ra   r(  r_   rc   rd   r   r  ri   ri   r   rj   r  9  r   r  F,np.ndarray | TimedeltaIndex | TimedeltaArraystr | floatr(  c           
        s^   |  tj}|tk}d}|| dk}t||}| dk}|r#dnd fdd}	|	S )z
    Return a formatter function for a range of timedeltas.
    These will all have the same format argument

    If box, then show the return in quotes
    l     "R: r   Nlongc                   sN   | d u st | rt| rS t| tst| } | jd} r%d| d}|S )Nr  ')r7   r:   r   r   Z
_repr_base)r   r   r(  r   r  ri   rj   r  k  s   
z*get_format_timedelta64.<locals>._formatter)viewr  Zint64r   r  r   )
r   r  r(  r  r  Zone_day_nanosZnot_midnightZbothr  r  ri   r/  rj   r)  L  s   r)  rF  rG  r   TextAdjustment | Nonec                   s   t | dks
|dkr| S |d u rt  n| t fdd| D |d ur*t|tdd ur8kr8d fd	d
fdd| D }  j| |d}|S )Nr   r$  c                 3  s    | ]}  |V  qd S r   r   r   )
adjustmentri   rj   r         z$_make_fixed_width.<locals>.<genexpr>r   r   ra   rc   c                   s6   d urdk  | k@ r| d d  d } | S )Nr   r   r   r^  )r2  conf_maxr   ri   rj   just  s   z_make_fixed_width.<locals>.justc                   r  ri   ri   r   )r5  ri   rj   r}     r~   z%_make_fixed_width.<locals>.<listcomp>r   )r   ra   rc   ra   )ro   r   rP  r   rR   )rF  rR   rG  r   r   ri   )r2  r4  r5  r   rj   rN  {  s   
rN  str_complexesr  c                   sD    fdd| D }dd |D }t |fddt||D }|S )z
    Separates the real and imaginary parts from the complex number, and
    executes the _trim_zeros_float method on each of those.
    c              	     s$   g | ]}d  ttd| qS )rf   z([j+-]))r   r  reri  r   )r  ri   rj   r}     s    z'_trim_zeros_complex.<locals>.<listcomp>c                 S  r_  ri   r   )r{   sri   ri   rj   r}     r~   c                   s   g | ]B\}}|d |d d d    | d d  ||d d d  |d d    ||d d  d   | d d  |d  qS )Nr   r   0r  ri   )r{   r8  rc  )
max_lengthri   rj   r}     s    ")rP  r5  )r6  r  trimmedlengthspaddedri   )r  r:  rj   r    s   

	r  	str_floatc                 C  s    |  d} | dr| d7 } | S )zX
    Trims trailing zeros after a decimal point,
    leaving just one if necessary.
    r9  r   )rstripendswith)r>  ri   ri   rj   r    s   

r  
str_floatsnp.ndarray | list[str]c                   sl   | }t d  ddfdddfd	d
}||r*fdd|D }||s fdd|D }|S )z
    Trims the maximum number of trailing zeros equally from
    all numbers containing decimals, leaving just one if
    necessary.
    z^\s*[\+-]?[0-9]+\z[0-9]*$rc   r_   c                   s   t  | d uS r   )r7  matchr^  )number_regexri   rj   is_number_with_decimal  r   z1_trim_zeros_float.<locals>.is_number_with_decimalr   rB  c                   s0    fdd| D }t |dkotdd |D S )z
        Determine if an array of strings should be trimmed.

        Returns True if all numbers containing decimals (defined by the
        above regular expression) within the array end in a zero, otherwise
        returns False.
        c                   s   g | ]} |r|qS ri   ri   r   rE  ri   rj   r}     s    z:_trim_zeros_float.<locals>.should_trim.<locals>.<listcomp>r   c                 s  s    | ]}| d V  qdS )r9  Nr@  r   ri   ri   rj   r     r3  z9_trim_zeros_float.<locals>.should_trim.<locals>.<genexpr>)ro   r$  )r   numbersrF  ri   rj   should_trim  s   z&_trim_zeros_float.<locals>.should_trimc                   s$   g | ]} |r|d d n|qS )Nr  ri   r   rF  ri   rj   r}     s   $ z%_trim_zeros_float.<locals>.<listcomp>c                   s*   g | ]}|r|  r|d  n|qS )r9  rG  r   )r  rE  ri   rj   r}     s    Nrn  )r   rB  rc   r_   )r7  compile)rA  r  r;  rI  r   ri   )r  rE  rD  rj   r    s   r  r   rB   c                 C  s    t | trtj| j S | jd uS r   )r   rC   comZany_not_noner   r   )r   ri   ri   rj   r    s   

r  c                   @  s   e Zd ZdZi dddddddd	d
dddddddddddddddddddddd d!d"d#Z	%d3d4d,d-Zd5d1d2Zd$S )6EngFormatterzl
    Formats float values according to engineering format.

    Based on matplotlib.ticker.EngFormatter
    irY  iziair  ipiniur  r   rf   r   rc  r  M	   Gr  T   P   E   Z   YNFaccuracyr   use_eng_prefixr_   rc   rd   c                 C  s   || _ || _d S r   r_  r`  )rh   r_  r`  ri   ri   rj   rk   	  s   
zEngFormatter.__init__numfloatra   c           
      C  s  t t|}t |rdS t |rdS d}|dk rd}| }|dkr3t tt| d d }nt d}|t	| j
 }|	t| j
 }t|}| jrX| j
| }n|dk rdd| d}nd	|d}|| d
|  }| jdu rzd}nd| jdd}|j||d}	|	S )a  
        Formats a number in engineering notation, appending a letter
        representing the power of 1000 of the original number. Some examples:
        >>> format_eng = EngFormatter(accuracy=0, use_eng_prefix=True)
        >>> format_eng(0)
        ' 0'
        >>> format_eng = EngFormatter(accuracy=1, use_eng_prefix=True)
        >>> format_eng(1_000_000)
        ' 1.0M'
        >>> format_eng = EngFormatter(accuracy=2, use_eng_prefix=False)
        >>> format_eng("-1e-6")
        '-1.00E-06'

        @param num: the value to represent
        @type num: either a numeric value or a string that can be converted to
                   a numeric value (as per decimal.Decimal constructor)

        @return: engineering formatted string
        rZ   infr   r   r  r   zE-02dzE+r  Nz{mant: g}{prefix}z{mant: .r  z
f}{prefix})mantprefix)r   ra   is_nanis_infiniter   mathr  r  r   rP  ENG_PREFIXESr4  r`  r_  r   )
rh   rb  ZdnumsignZpow10Z	int_pow10rg  rf  
format_strr  ri   ri   rj   __call__  s4   

 
zEngFormatter.__call__r  )r_  r   r`  r_   rc   rd   )rb  rc  rc   ra   )r   r   r   ro  rk  rk   rn  ri   ri   ri   rj   rL    sP    	
rL  r   r_  r   r`  rd   c                 C  s   t dt| | dS )a  
    Format float representation in DataFrame with SI notation.

    Parameters
    ----------
    accuracy : int, default 3
        Number of decimal digits after the floating point.
    use_eng_prefix : bool, default False
        Whether to represent a value with SI prefixes.

    Returns
    -------
    None

    Examples
    --------
    >>> df = pd.DataFrame([1e-9, 1e-3, 1, 1e3, 1e6])
    >>> df
                  0
    0  1.000000e-09
    1  1.000000e-03
    2  1.000000e+00
    3  1.000000e+03
    4  1.000000e+06

    >>> pd.set_eng_float_format(accuracy=1)
    >>> df
             0
    0  1.0E-09
    1  1.0E-03
    2  1.0E+00
    3  1.0E+03
    4  1.0E+06

    >>> pd.set_eng_float_format(use_eng_prefix=True)
    >>> df
            0
    0  1.000n
    1  1.000m
    2   1.000
    3  1.000k
    4  1.000M

    >>> pd.set_eng_float_format(accuracy=1, use_eng_prefix=True)
    >>> df
          0
    0  1.0n
    1  1.0m
    2   1.0
    3  1.0k
    4  1.0M

    >>> pd.set_option("display.float_format", None)  # unset option
    r   N)r   rL  ra  ri   ri   rj   set_eng_float_formatN  s   7ro  rf   ra  sentinelbool | object | strlist[dict[int, int]]c           	      C  s   t | dkrg S dgt | d  }g }| D ]1}d}i }t|D ]\}}|| r,||kr,qd||< || ||< |}qt || ||< || q|S )a  
    For each index in each level the function returns lengths of indexes.

    Parameters
    ----------
    levels : list of lists
        List of values on for level.
    sentinel : string, optional
        Value which states that no new index starts on there.

    Returns
    -------
    Returns list of maps. For each level returns map of indexes (key is index
    in row and value is length of index).
    r   TF)ro   rL  r   )	ra  rp  controlr   rW  
last_indexr<  r|   keyri   ri   rj   get_level_lengths  s    rv  WriteBuffer[str]linesc                 C  s4   t dd |D rdd |D }| d| dS )z
    Appends lines to a buffer.

    Parameters
    ----------
    buf
        The buffer to write to
    lines
        The lines to append.
    c                 s  s    | ]}t |tV  qd S r   )r   ra   r   ri   ri   rj   r     r3  z#buffer_put_lines.<locals>.<genexpr>c                 S  r_  ri   rg  r   ri   ri   rj   r}     r~   z$buffer_put_lines.<locals>.<listcomp>rn   N)r   r  r   )r\   rx  ri   ri   rj   buffer_put_lines  s   ry  )rc   r   )rc   r   r  )r~  ra   r\   rr  r   r   rc   r   r   )r\   rr  r   r   rc   r  )	NrZ   NNrP   r   TNN)r   r   rS  rU  rt   r  r`   ra   r  r   r   r  rR   ra   r  ra   r   r  rg   r   r  rU  rc   rs   )r  r  rc   rs   )r   r  rc   r_   )r   )r   r  r  ra   rc   ra   r  )r   r  r  ra   r  r   rc   ra   )r  r_   r  ra   r  r   rc   r   )r   r#  r  r   rc   r   r*  )r   r+  r  r,  r(  r_   rc   r   )rP   NN)
rF  rs   rR   ra   rG  r   r   r1  rc   rs   )r   )r6  r  r  ra   rc   rs   )r>  ra   rc   ra   )rA  rB  r  ra   rc   rs   )r   rB   rc   r_   )r   F)r_  r   r`  r_   rc   rd   )rf   )ra  r   rp  rq  rc   rr  )r\   rw  rx  rs   rc   rd   )ro  
__future__r   
contextlibr   csvr   r   r  r   	functoolsr   ior   rj  r7  shutilr	   typingr
   r   r   r   r   r   r   r   r   r   r   r   unicodedatar   numpyr  Zpandas._config.configr   r   Zpandas._libsr   Zpandas._libs.missingr   Zpandas._libs.tslibsr   r   r   r   r   r    Zpandas._libs.tslibs.nattyper!   Zpandas._typingr"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   Zpandas.core.dtypes.commonr-   r.   r/   r0   r1   r2   r3   r4   r5   r6   r7   r8   Zpandas.core.dtypes.dtypesr9   Zpandas.core.dtypes.missingr:   r;   Zpandas.core.arraysr<   r=   r>   Zpandas.core.arrays.string_r?   Zpandas.core.baser@   Zpandas.core.commoncorecommonrK  Zpandas.core.constructionrA   Zpandas.core.indexes.apirB   rC   rD   rE   Zpandas.core.indexes.datetimesrF   Zpandas.core.indexes.timedeltasrG   Zpandas.core.reshape.concatrH   Zpandas.io.commonrI   rJ   r   rK   ZpandasrL   rM   rN   __annotations__Z_VALID_JUSTIFY_PARAMETERSrX   rY   r   r   r   r   r   r   r   rq  r}  r  ru   r  r  r  r  r  r  r  r  r  r"  r&  r  r  r)  rN  r  r  r  r  rL  ro  rv  ry  ri   ri   ri   rj   <module>   s    8 48;	@ !
'

!&   E @(Rp -

F
1"

$a;+