U
    ùQ”i0  ã                   @  s$  d dl mZ d dlmZmZ d dlZd dlmZ d dlm	Z	m
Z
mZmZmZ d dlmZ d dlmZmZ d dlmZ d d	lmZ e	r˜d d
lmZmZmZ edƒZedƒZedƒZedƒZddeeeeddœZ edƒZ!edƒZ"dddee!e"ddœZ#edƒZ$ddddœdd„Z%ddddœd d!„Z&dGd"d#d$œd%d&„Z'G d'd(„ d(eƒZ(G d)d*„ d*e(ƒZ)G d+d,„ d,e(ƒZ*G d-d.„ d.ƒZ+G d/d0„ d0e+ƒZ,G d1d2„ d2e+ƒZ-G d3d4„ d4eƒZ.G d5d6„ d6e.ƒZ/G d7d8„ d8e/ƒZ0G d9d:„ d:e.ƒZ1G d;d<„ d<e/e1ƒZ2G d=d>„ d>e.ƒZ3G d?d@„ d@e3ƒZ4G dAdB„ dBe3e1ƒZ5ddCdDœdEdF„Z6dS )Hé    )Úannotations)ÚABCÚabstractmethodN)Údedent)ÚTYPE_CHECKINGÚIterableÚIteratorÚMappingÚSequence©Ú
get_option)ÚDtypeÚWriteBuffer)Úformat)Úpprint_thing)Ú	DataFrameÚIndexÚSeriesa      max_cols : int, optional
        When to switch from the verbose to the truncated output. If the
        DataFrame has more than `max_cols` columns, the truncated output
        is used. By default, the setting in
        ``pandas.options.display.max_info_columns`` is used.aR      show_counts : bool, optional
        Whether to show the non-null counts. By default, this is shown
        only if the DataFrame is smaller than
        ``pandas.options.display.max_info_rows`` and
        ``pandas.options.display.max_info_columns``. A value of True always
        shows the counts, and False never shows the counts.a�      >>> int_values = [1, 2, 3, 4, 5]
    >>> text_values = ['alpha', 'beta', 'gamma', 'delta', 'epsilon']
    >>> float_values = [0.0, 0.25, 0.5, 0.75, 1.0]
    >>> df = pd.DataFrame({"int_col": int_values, "text_col": text_values,
    ...                   "float_col": float_values})
    >>> df
        int_col text_col  float_col
    0        1    alpha       0.00
    1        2     beta       0.25
    2        3    gamma       0.50
    3        4    delta       0.75
    4        5  epsilon       1.00

    Prints information of all columns:

    >>> df.info(verbose=True)
    <class 'pandas.core.frame.DataFrame'>
    RangeIndex: 5 entries, 0 to 4
    Data columns (total 3 columns):
     #   Column     Non-Null Count  Dtype
    ---  ------     --------------  -----
     0   int_col    5 non-null      int64
     1   text_col   5 non-null      object
     2   float_col  5 non-null      float64
    dtypes: float64(1), int64(1), object(1)
    memory usage: 248.0+ bytes

    Prints a summary of columns count and its dtypes but not per column
    information:

    >>> df.info(verbose=False)
    <class 'pandas.core.frame.DataFrame'>
    RangeIndex: 5 entries, 0 to 4
    Columns: 3 entries, int_col to float_col
    dtypes: float64(1), int64(1), object(1)
    memory usage: 248.0+ bytes

    Pipe output of DataFrame.info to buffer instead of sys.stdout, get
    buffer content and writes to a text file:

    >>> import io
    >>> buffer = io.StringIO()
    >>> df.info(buf=buffer)
    >>> s = buffer.getvalue()
    >>> with open("df_info.txt", "w",
    ...           encoding="utf-8") as f:  # doctest: +SKIP
    ...     f.write(s)
    260

    The `memory_usage` parameter allows deep introspection mode, specially
    useful for big DataFrames and fine-tune memory optimization:

    >>> random_strings_array = np.random.choice(['a', 'b', 'c'], 10 ** 6)
    >>> df = pd.DataFrame({
    ...     'column_1': np.random.choice(['a', 'b', 'c'], 10 ** 6),
    ...     'column_2': np.random.choice(['a', 'b', 'c'], 10 ** 6),
    ...     'column_3': np.random.choice(['a', 'b', 'c'], 10 ** 6)
    ... })
    >>> df.info()
    <class 'pandas.core.frame.DataFrame'>
    RangeIndex: 1000000 entries, 0 to 999999
    Data columns (total 3 columns):
     #   Column    Non-Null Count    Dtype
    ---  ------    --------------    -----
     0   column_1  1000000 non-null  object
     1   column_2  1000000 non-null  object
     2   column_3  1000000 non-null  object
    dtypes: object(3)
    memory usage: 22.9+ MB

    >>> df.info(memory_usage='deep')
    <class 'pandas.core.frame.DataFrame'>
    RangeIndex: 1000000 entries, 0 to 999999
    Data columns (total 3 columns):
     #   Column    Non-Null Count    Dtype
    ---  ------    --------------    -----
     0   column_1  1000000 non-null  object
     1   column_2  1000000 non-null  object
     2   column_3  1000000 non-null  object
    dtypes: object(3)
    memory usage: 165.9 MBz”    DataFrame.describe: Generate descriptive statistics of DataFrame
        columns.
    DataFrame.memory_usage: Memory usage of DataFrame columns.r   z and columnsÚ )ÚklassZtype_subZmax_cols_subÚshow_counts_subZexamples_subZsee_also_subZversion_added_subaî      >>> int_values = [1, 2, 3, 4, 5]
    >>> text_values = ['alpha', 'beta', 'gamma', 'delta', 'epsilon']
    >>> s = pd.Series(text_values, index=int_values)
    >>> s.info()
    <class 'pandas.core.series.Series'>
    Index: 5 entries, 1 to 5
    Series name: None
    Non-Null Count  Dtype
    --------------  -----
    5 non-null      object
    dtypes: object(1)
    memory usage: 80.0+ bytes

    Prints a summary excluding information about its values:

    >>> s.info(verbose=False)
    <class 'pandas.core.series.Series'>
    Index: 5 entries, 1 to 5
    dtypes: object(1)
    memory usage: 80.0+ bytes

    Pipe output of Series.info to buffer instead of sys.stdout, get
    buffer content and writes to a text file:

    >>> import io
    >>> buffer = io.StringIO()
    >>> s.info(buf=buffer)
    >>> s = buffer.getvalue()
    >>> with open("df_info.txt", "w",
    ...           encoding="utf-8") as f:  # doctest: +SKIP
    ...     f.write(s)
    260

    The `memory_usage` parameter allows deep introspection mode, specially
    useful for big Series and fine-tune memory optimization:

    >>> random_strings_array = np.random.choice(['a', 'b', 'c'], 10 ** 6)
    >>> s = pd.Series(np.random.choice(['a', 'b', 'c'], 10 ** 6))
    >>> s.info()
    <class 'pandas.core.series.Series'>
    RangeIndex: 1000000 entries, 0 to 999999
    Series name: None
    Non-Null Count    Dtype
    --------------    -----
    1000000 non-null  object
    dtypes: object(1)
    memory usage: 7.6+ MB

    >>> s.info(memory_usage='deep')
    <class 'pandas.core.series.Series'>
    RangeIndex: 1000000 entries, 0 to 999999
    Series name: None
    Non-Null Count    Dtype
    --------------    -----
    1000000 non-null  object
    dtypes: object(1)
    memory usage: 55.3 MBzp    Series.describe: Generate descriptive statistics of Series.
    Series.memory_usage: Memory usage of Series.r   z
.. versionadded:: 1.4.0
aÅ  
    Print a concise summary of a {klass}.

    This method prints information about a {klass} including
    the index dtype{type_sub}, non-null values and memory usage.
    {version_added_sub}
    Parameters
    ----------
    verbose : bool, optional
        Whether to print the full summary. By default, the setting in
        ``pandas.options.display.max_info_columns`` is followed.
    buf : writable buffer, defaults to sys.stdout
        Where to send the output. By default, the output is printed to
        sys.stdout. Pass a writable buffer if you need to further process
        the output.
    {max_cols_sub}
    memory_usage : bool, str, optional
        Specifies whether total memory usage of the {klass}
        elements (including the index) should be displayed. By default,
        this follows the ``pandas.options.display.memory_usage`` setting.

        True always show memory usage. False never shows memory usage.
        A value of 'deep' is equivalent to "True with deep introspection".
        Memory usage is shown in human-readable units (base-2
        representation). Without deep introspection a memory estimation is
        made based in column dtype and number of rows assuming values
        consume the same memory amount for corresponding dtypes. With deep
        memory introspection, a real memory usage calculation is performed
        at the cost of computational resources. See the
        :ref:`Frequently Asked Questions <df-memory-usage>` for more
        details.
    {show_counts_sub}

    Returns
    -------
    None
        This method prints a summary of a {klass} and returns None.

    See Also
    --------
    {see_also_sub}

    Examples
    --------
    {examples_sub}
    zstr | DtypeÚintÚstr)ÚsÚspaceÚreturnc                 C  s   t | ƒd|…  |¡S )a»  
    Make string of specified length, padding to the right if necessary.

    Parameters
    ----------
    s : Union[str, Dtype]
        String to be formatted.
    space : int
        Length to force string to be of.

    Returns
    -------
    str
        String coerced to given length.

    Examples
    --------
    >>> pd.io.formats.info._put_str("panda", 6)
    'panda '
    >>> pd.io.formats.info._put_str("panda", 4)
    'pand'
    N)r   Úljust)r   r   © r   ú:/tmp/pip-unpacked-wheel-58ppya8e/pandas/io/formats/info.pyÚ_put_str$  s    r   Úfloat)ÚnumÚsize_qualifierr   c                 C  sB   dD ],}| dk r(| d›|› d|› �  S | d } q| d›|› d�S )a{  
    Return size in human readable format.

    Parameters
    ----------
    num : int
        Size in bytes.
    size_qualifier : str
        Either empty, or '+' (if lower bound).

    Returns
    -------
    str
        Size in human readable format.

    Examples
    --------
    >>> _sizeof_fmt(23028, '')
    '22.5 KB'

    >>> _sizeof_fmt(23028, '+')
    '22.5+ KB'
    )ÚbytesZKBZMBZGBZTBg      �@z3.1fú z PBr   )r!   r"   Úxr   r   r   Ú_sizeof_fmt>  s
    
r&   úbool | str | Noneú
bool | str)Úmemory_usager   c                 C  s   | dkrt dƒ} | S )z5Get memory usage based on inputs and display options.Nzdisplay.memory_usager   )r)   r   r   r   Ú_initialize_memory_usage]  s    r*   c                   @  s¸   e Zd ZU dZded< ded< eeddœdd	„ƒƒZeed
dœdd„ƒƒZeeddœdd„ƒƒZ	eeddœdd„ƒƒZ
eddœdd„ƒZeddœdd„ƒZeddddddœdd„ƒZdS ) ÚBaseInfoaj  
    Base class for DataFrameInfo and SeriesInfo.

    Parameters
    ----------
    data : DataFrame or Series
        Either dataframe or series.
    memory_usage : bool or str, optional
        If "deep", introspect the data deeply by interrogating object dtypes
        for system-level memory consumption, and include it in the returned
        values.
    úDataFrame | SeriesÚdatar(   r)   úIterable[Dtype]©r   c                 C  s   dS )z¡
        Dtypes.

        Returns
        -------
        dtypes : sequence
            Dtype of each of the DataFrame's columns (or one series column).
        Nr   ©Úselfr   r   r   Údtypesw  s    zBaseInfo.dtypesúMapping[str, int]c                 C  s   dS )ú!Mapping dtype - number of counts.Nr   r0   r   r   r   Údtype_countsƒ  s    zBaseInfo.dtype_countsúSequence[int]c                 C  s   dS )úBSequence of non-null counts for all columns or column (if series).Nr   r0   r   r   r   Únon_null_countsˆ  s    zBaseInfo.non_null_countsr   c                 C  s   dS )zœ
        Memory usage in bytes.

        Returns
        -------
        memory_usage_bytes : int
            Object's total memory usage in bytes.
        Nr   r0   r   r   r   Úmemory_usage_bytes�  s    zBaseInfo.memory_usage_bytesr   c                 C  s   t | j| jƒ› d�S )z0Memory usage in a form of human readable string.Ú
)r&   r9   r"   r0   r   r   r   Úmemory_usage_string™  s    zBaseInfo.memory_usage_stringc                 C  s2   d}| j r.| j dkr.d| jks*| jj ¡ r.d}|S )Nr   ÚdeepÚobjectú+)r)   r5   r-   ÚindexZ_is_memory_usage_qualified)r1   r"   r   r   r   r"   ž  s    
ÿ
þzBaseInfo.size_qualifierúWriteBuffer[str] | Noneú
int | Noneúbool | NoneÚNone©ÚbufÚmax_colsÚverboseÚshow_countsr   c                C  s   d S ©Nr   )r1   rE   rF   rG   rH   r   r   r   Úrender­  s    	zBaseInfo.renderN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Ú__annotations__Úpropertyr   r2   r5   r8   r9   r;   r"   rJ   r   r   r   r   r+   f  s*   


r+   c                   @  s¦   e Zd ZdZd!ddddœdd„Zed	d
œdd„ƒZedd
œdd„ƒZedd
œdd„ƒZedd
œdd„ƒZ	edd
œdd„ƒZ
edd
œdd„ƒZddddddœdd „ZdS )"ÚDataFrameInfoz0
    Class storing dataframe-specific info.
    Nr   r'   rC   ©r-   r)   r   c                 C  s   || _ t|ƒ| _d S rI   ©r-   r*   r)   ©r1   r-   r)   r   r   r   Ú__init__¾  s    zDataFrameInfo.__init__r3   r/   c                 C  s
   t | jƒS rI   )Ú_get_dataframe_dtype_countsr-   r0   r   r   r   r5   Æ  s    zDataFrameInfo.dtype_countsr.   c                 C  s   | j jS )z
        Dtypes.

        Returns
        -------
        dtypes
            Dtype of each of the DataFrame's columns.
        ©r-   r2   r0   r   r   r   r2   Ê  s    
zDataFrameInfo.dtypesr   c                 C  s   | j jS )zz
        Column names.

        Returns
        -------
        ids : Index
            DataFrame's column names.
        )r-   Úcolumnsr0   r   r   r   ÚidsÖ  s    
zDataFrameInfo.idsr   c                 C  s
   t | jƒS ©z#Number of columns to be summarized.)ÚlenrY   r0   r   r   r   Ú	col_countâ  s    zDataFrameInfo.col_countr6   c                 C  s
   | j  ¡ S )r7   ©r-   Úcountr0   r   r   r   r8   ç  s    zDataFrameInfo.non_null_countsc                 C  s   | j dk}| jj d|d� ¡ S )Nr<   T©r?   r<   )r)   r-   Úsum©r1   r<   r   r   r   r9   ì  s    
z DataFrameInfo.memory_usage_bytesr@   rA   rB   rD   c                C  s   t | |||d�}| |¡ d S )N)ÚinforF   rG   rH   )ÚDataFrameInfoPrinterÚ	to_buffer©r1   rE   rF   rG   rH   Úprinterr   r   r   rJ   ñ  s    üzDataFrameInfo.render)N)rK   rL   rM   rN   rU   rP   r5   r2   rY   r\   r8   r9   rJ   r   r   r   r   rQ   ¹  s     ýrQ   c                   @  sŽ   e Zd ZdZdddddœdd„Zddddd	œd
dddddœdd„Zeddœdd„ƒZeddœdd„ƒZeddœdd„ƒZ	eddœdd„ƒZ
dS )Ú
SeriesInfoz-
    Class storing series-specific info.
    Nr   r'   rC   rR   c                 C  s   || _ t|ƒ| _d S rI   rS   rT   r   r   r   rU     s    zSeriesInfo.__init__)rE   rF   rG   rH   r@   rA   rB   rD   c                C  s,   |d k	rt dƒ‚t| ||d�}| |¡ d S )NzIArgument `max_cols` can only be passed in DataFrame.info, not Series.info)rb   rG   rH   )Ú
ValueErrorÚSeriesInfoPrinterrd   re   r   r   r   rJ     s    ÿýzSeriesInfo.renderr6   r/   c                 C  s   | j  ¡ gS rI   r]   r0   r   r   r   r8   #  s    zSeriesInfo.non_null_countsr.   c                 C  s
   | j jgS rI   rW   r0   r   r   r   r2   '  s    zSeriesInfo.dtypesr3   c                 C  s   ddl m} t|| jƒƒS )Nr   )r   )Zpandas.core.framer   rV   r-   )r1   r   r   r   r   r5   +  s    zSeriesInfo.dtype_countsr   c                 C  s   | j dk}| jj d|d�S )z“Memory usage in bytes.

        Returns
        -------
        memory_usage_bytes : int
            Object's total memory usage in bytes.
        r<   Tr_   )r)   r-   ra   r   r   r   r9   1  s    	
zSeriesInfo.memory_usage_bytes)N)rK   rL   rM   rN   rU   rJ   rP   r8   r2   r5   r9   r   r   r   r   rg     s     ýúrg   c                   @  s4   e Zd ZdZddddœdd„Zedd	œd
d„ƒZdS )ÚInfoPrinterAbstractz6
    Class for printing dataframe or series info.
    Nr@   rC   )rE   r   c                 C  s.   |   ¡ }| ¡ }|dkrtj}t ||¡ dS )z Save dataframe info into buffer.N)Ú_create_table_builderÚ	get_linesÚsysÚstdoutÚfmtZbuffer_put_lines)r1   rE   Ztable_builderÚlinesr   r   r   rd   C  s
    zInfoPrinterAbstract.to_bufferÚTableBuilderAbstractr/   c                 C  s   dS )z!Create instance of table builder.Nr   r0   r   r   r   rk   K  s    z)InfoPrinterAbstract._create_table_builder)N)rK   rL   rM   rN   rd   r   rk   r   r   r   r   rj   >  s   rj   c                   @  sž   e Zd ZdZdddddddœdd	„Zed
dœdd„ƒZeddœdd„ƒZeddœdd„ƒZed
dœdd„ƒZ	dd
dœdd„Z
dddœdd„Zddœdd„ZdS )rc   a{  
    Class for printing dataframe info.

    Parameters
    ----------
    info : DataFrameInfo
        Instance of DataFrameInfo.
    max_cols : int, optional
        When to switch from the verbose to the truncated output.
    verbose : bool, optional
        Whether to print the full summary.
    show_counts : bool, optional
        Whether to show the non-null counts.
    NrQ   rA   rB   rC   )rb   rF   rG   rH   r   c                 C  s0   || _ |j| _|| _|  |¡| _|  |¡| _d S rI   )rb   r-   rG   Ú_initialize_max_colsrF   Ú_initialize_show_countsrH   )r1   rb   rF   rG   rH   r   r   r   rU   `  s
    zDataFrameInfoPrinter.__init__r   r/   c                 C  s   t dt| jƒd ƒS )z"Maximum info rows to be displayed.zdisplay.max_info_rowsé   )r   r[   r-   r0   r   r   r   Úmax_rowsm  s    zDataFrameInfoPrinter.max_rowsÚboolc                 C  s   t | j| jkƒS )zDCheck if number of columns to be summarized does not exceed maximum.)rv   r\   rF   r0   r   r   r   Úexceeds_info_colsr  s    z&DataFrameInfoPrinter.exceeds_info_colsc                 C  s   t t| jƒ| jkƒS )zACheck if number of rows to be summarized does not exceed maximum.)rv   r[   r-   ru   r0   r   r   r   Úexceeds_info_rowsw  s    z&DataFrameInfoPrinter.exceeds_info_rowsc                 C  s   | j jS rZ   ©rb   r\   r0   r   r   r   r\   |  s    zDataFrameInfoPrinter.col_count)rF   r   c                 C  s   |d krt d| jd ƒS |S )Nzdisplay.max_info_columnsrt   )r   r\   )r1   rF   r   r   r   rr   �  s    z)DataFrameInfoPrinter._initialize_max_cols©rH   r   c                 C  s$   |d krt | j o| j ƒS |S d S rI   )rv   rw   rx   ©r1   rH   r   r   r   rs   †  s    z,DataFrameInfoPrinter._initialize_show_countsÚDataFrameTableBuilderc                 C  sR   | j rt| j| jd�S | j dkr,t| jd�S | jr>t| jd�S t| j| jd�S dS )z[
        Create instance of table builder based on verbosity and display settings.
        ©rb   Úwith_countsF©rb   N)rG   ÚDataFrameTableBuilderVerboserb   rH   ÚDataFrameTableBuilderNonVerboserw   r0   r   r   r   rk   Œ  s    þ
þz*DataFrameInfoPrinter._create_table_builder)NNN)rK   rL   rM   rN   rU   rP   ru   rw   rx   r\   rr   rs   rk   r   r   r   r   rc   P  s       ûrc   c                   @  sD   e Zd ZdZddddddœdd„Zd	d
œdd„Zdddœdd„ZdS )ri   a  Class for printing series info.

    Parameters
    ----------
    info : SeriesInfo
        Instance of SeriesInfo.
    verbose : bool, optional
        Whether to print the full summary.
    show_counts : bool, optional
        Whether to show the non-null counts.
    Nrg   rB   rC   )rb   rG   rH   r   c                 C  s$   || _ |j| _|| _|  |¡| _d S rI   )rb   r-   rG   rs   rH   )r1   rb   rG   rH   r   r   r   rU   ®  s    zSeriesInfoPrinter.__init__ÚSeriesTableBuilderr/   c                 C  s0   | j s| j dkr t| j| jd�S t| jd�S dS )zF
        Create instance of table builder based on verbosity.
        Nr}   r   )rG   ÚSeriesTableBuilderVerboserb   rH   ÚSeriesTableBuilderNonVerboser0   r   r   r   rk   ¹  s    þz'SeriesInfoPrinter._create_table_builderrv   rz   c                 C  s   |d krdS |S d S )NTr   r{   r   r   r   rs   Å  s    z)SeriesInfoPrinter._initialize_show_counts)NN)rK   rL   rM   rN   rU   rk   rs   r   r   r   r   ri   ¡  s     üri   c                   @  sÊ   e Zd ZU dZded< ded< eddœdd„ƒZed	dœd
d„ƒZeddœdd„ƒZ	eddœdd„ƒZ
eddœdd„ƒZeddœdd„ƒZeddœdd„ƒZddœdd„Zddœdd„Zddœd d!„Zd"S )#rq   z*
    Abstract builder for info table.
    ú	list[str]Ú_linesr+   rb   r/   c                 C  s   dS )z-Product in a form of list of lines (strings).Nr   r0   r   r   r   rl   Ô  s    zTableBuilderAbstract.get_linesr,   c                 C  s   | j jS rI   ©rb   r-   r0   r   r   r   r-   Ø  s    zTableBuilderAbstract.datar.   c                 C  s   | j jS )z*Dtypes of each of the DataFrame's columns.)rb   r2   r0   r   r   r   r2   Ü  s    zTableBuilderAbstract.dtypesr3   c                 C  s   | j jS )r4   )rb   r5   r0   r   r   r   r5   á  s    z!TableBuilderAbstract.dtype_countsrv   c                 C  s   t | jjƒS )z Whether to display memory usage.)rv   rb   r)   r0   r   r   r   Údisplay_memory_usageæ  s    z)TableBuilderAbstract.display_memory_usager   c                 C  s   | j jS )z/Memory usage string with proper size qualifier.)rb   r;   r0   r   r   r   r;   ë  s    z(TableBuilderAbstract.memory_usage_stringr6   c                 C  s   | j jS rI   )rb   r8   r0   r   r   r   r8   ð  s    z$TableBuilderAbstract.non_null_countsrC   c                 C  s   | j  tt| jƒƒ¡ dS )z>Add line with string representation of dataframe to the table.N)r†   Úappendr   Útyper-   r0   r   r   r   Úadd_object_type_lineô  s    z)TableBuilderAbstract.add_object_type_linec                 C  s   | j  | jj ¡ ¡ dS )z,Add line with range of indices to the table.N)r†   r‰   r-   r?   Ú_summaryr0   r   r   r   Úadd_index_range_lineø  s    z)TableBuilderAbstract.add_index_range_linec                 C  s4   dd„ t | j ¡ ƒD ƒ}| j dd |¡› �¡ dS )z2Add summary line with dtypes present in dataframe.c                 S  s"   g | ]\}}|› d |d›d�‘qS )ú(Údú)r   )Ú.0ÚkeyÚvalr   r   r   Ú
<listcomp>þ  s    z8TableBuilderAbstract.add_dtypes_line.<locals>.<listcomp>zdtypes: z, N)Úsortedr5   Úitemsr†   r‰   Újoin)r1   Zcollected_dtypesr   r   r   Úadd_dtypes_lineü  s    ÿz$TableBuilderAbstract.add_dtypes_lineN)rK   rL   rM   rN   rO   r   rl   rP   r-   r2   r5   rˆ   r;   r8   r‹   r�   r˜   r   r   r   r   rq   Ì  s(   
rq   c                   @  s’   e Zd ZdZdddœdd„Zddœd	d
„Zddœdd„Zeddœdd„ƒZe	ddœdd„ƒZ
e	ddœdd„ƒZe	ddœdd„ƒZddœdd„ZdS )r|   z�
    Abstract builder for dataframe info table.

    Parameters
    ----------
    info : DataFrameInfo.
        Instance of DataFrameInfo.
    rQ   rC   ©rb   r   c                C  s
   || _ d S rI   r   ©r1   rb   r   r   r   rU     s    zDataFrameTableBuilder.__init__r…   r/   c                 C  s(   g | _ | jdkr|  ¡  n|  ¡  | j S )Nr   )r†   r\   Ú_fill_empty_infoÚ_fill_non_empty_infor0   r   r   r   rl     s
    

zDataFrameTableBuilder.get_linesc                 C  s0   |   ¡  |  ¡  | j dt| jƒj› d�¡ dS )z;Add lines to the info table, pertaining to empty dataframe.zEmpty r:   N)r‹   r�   r†   r‰   rŠ   r-   rK   r0   r   r   r   r›     s    z&DataFrameTableBuilder._fill_empty_infoc                 C  s   dS ©z?Add lines to the info table, pertaining to non-empty dataframe.Nr   r0   r   r   r   rœ     s    z*DataFrameTableBuilder._fill_non_empty_infor   c                 C  s   | j jS )z
DataFrame.r‡   r0   r   r   r   r-   #  s    zDataFrameTableBuilder.datar   c                 C  s   | j jS )zDataframe columns.)rb   rY   r0   r   r   r   rY   (  s    zDataFrameTableBuilder.idsr   c                 C  s   | j jS )z-Number of dataframe columns to be summarized.ry   r0   r   r   r   r\   -  s    zDataFrameTableBuilder.col_countc                 C  s   | j  d| j› �¡ dS ©z!Add line containing memory usage.zmemory usage: N©r†   r‰   r;   r0   r   r   r   Úadd_memory_usage_line2  s    z+DataFrameTableBuilder.add_memory_usage_lineN)rK   rL   rM   rN   rU   rl   r›   r   rœ   rP   r-   rY   r\   r    r   r   r   r   r|     s   	r|   c                   @  s,   e Zd ZdZddœdd„Zddœdd„ZdS )	r�   z>
    Dataframe info table builder for non-verbose output.
    rC   r/   c                 C  s2   |   ¡  |  ¡  |  ¡  |  ¡  | jr.|  ¡  dS r�   )r‹   r�   Úadd_columns_summary_liner˜   rˆ   r    r0   r   r   r   rœ   <  s    z4DataFrameTableBuilderNonVerbose._fill_non_empty_infoc                 C  s   | j  | jjdd�¡ d S )NZColumns©Úname)r†   r‰   rY   rŒ   r0   r   r   r   r¡   E  s    z8DataFrameTableBuilderNonVerbose.add_columns_summary_lineN)rK   rL   rM   rN   rœ   r¡   r   r   r   r   r�   7  s   	r�   c                   @  sò   e Zd ZU dZdZded< ded< ded< d	ed
< eeddœdd„ƒƒZeddœdd„ƒZ	ddœdd„Z
ddœdd„Zddœdd„Zeddœdd„ƒZeddœdd„ƒZddœdd„Zddœdd „Zddœd!d"„Zd#dœd$d%„Zd#dœd&d'„Zd(S ))ÚTableBuilderVerboseMixinz(
    Mixin for verbose info output.
    z  r   ÚSPACINGzSequence[Sequence[str]]Ústrrowsr6   Úgross_column_widthsrv   r~   úSequence[str]r/   c                 C  s   dS )ú.Headers names of the columns in verbose table.Nr   r0   r   r   r   ÚheadersS  s    z TableBuilderVerboseMixin.headersc                 C  s   dd„ | j D ƒS )z'Widths of header columns (only titles).c                 S  s   g | ]}t |ƒ‘qS r   ©r[   ©r‘   Úcolr   r   r   r”   [  s     zATableBuilderVerboseMixin.header_column_widths.<locals>.<listcomp>)rª   r0   r   r   r   Úheader_column_widthsX  s    z-TableBuilderVerboseMixin.header_column_widthsc                 C  s   |   ¡ }dd„ t| j|ƒD ƒS )zAGet widths of columns containing both headers and actual content.c                 S  s   g | ]}t |Ž ‘qS r   ©Úmax)r‘   Zwidthsr   r   r   r”   `  s   ÿzETableBuilderVerboseMixin._get_gross_column_widths.<locals>.<listcomp>)Ú_get_body_column_widthsÚzipr®   )r1   Zbody_column_widthsr   r   r   Ú_get_gross_column_widths]  s    
þz1TableBuilderVerboseMixin._get_gross_column_widthsc                 C  s   t t| jŽ ƒ}dd„ |D ƒS )z$Get widths of table content columns.c                 S  s   g | ]}t d d„ |D ƒƒ‘qS )c                 s  s   | ]}t |ƒV  qd S rI   r«   )r‘   r%   r   r   r   Ú	<genexpr>h  s     zNTableBuilderVerboseMixin._get_body_column_widths.<locals>.<listcomp>.<genexpr>r¯   r¬   r   r   r   r”   h  s     zDTableBuilderVerboseMixin._get_body_column_widths.<locals>.<listcomp>)Úlistr²   r¦   )r1   Zstrcolsr   r   r   r±   e  s    z0TableBuilderVerboseMixin._get_body_column_widthsúIterator[Sequence[str]]c                 C  s   | j r|  ¡ S |  ¡ S dS )z„
        Generator function yielding rows content.

        Each element represents a row comprising a sequence of strings.
        N)r~   Ú_gen_rows_with_countsÚ_gen_rows_without_countsr0   r   r   r   Ú	_gen_rowsj  s    z"TableBuilderVerboseMixin._gen_rowsc                 C  s   dS ©z=Iterator with string representation of body data with counts.Nr   r0   r   r   r   r·   u  s    z.TableBuilderVerboseMixin._gen_rows_with_countsc                 C  s   dS ©z@Iterator with string representation of body data without counts.Nr   r0   r   r   r   r¸   y  s    z1TableBuilderVerboseMixin._gen_rows_without_countsrC   c                 C  s0   | j  dd„ t| j| jƒD ƒ¡}| j |¡ d S )Nc                 S  s   g | ]\}}t ||ƒ‘qS r   ©r   )r‘   ÚheaderZ	col_widthr   r   r   r”     s   ÿz<TableBuilderVerboseMixin.add_header_line.<locals>.<listcomp>)r¥   r—   r²   rª   r§   r†   r‰   )r1   Zheader_liner   r   r   Úadd_header_line}  s    þÿz(TableBuilderVerboseMixin.add_header_linec                 C  s0   | j  dd„ t| j| jƒD ƒ¡}| j |¡ d S )Nc                 S  s   g | ]\}}t d | |ƒ‘qS )ú-r¼   )r‘   Zheader_colwidthÚgross_colwidthr   r   r   r”   ˆ  s   ÿz?TableBuilderVerboseMixin.add_separator_line.<locals>.<listcomp>)r¥   r—   r²   r®   r§   r†   r‰   )r1   Zseparator_liner   r   r   Úadd_separator_line†  s     ÿþÿz+TableBuilderVerboseMixin.add_separator_linec                 C  s:   | j D ].}| j dd„ t|| jƒD ƒ¡}| j |¡ qd S )Nc                 S  s   g | ]\}}t ||ƒ‘qS r   r¼   )r‘   r­   rÀ   r   r   r   r”   ”  s   ÿz;TableBuilderVerboseMixin.add_body_lines.<locals>.<listcomp>)r¦   r¥   r—   r²   r§   r†   r‰   )r1   ÚrowZ	body_liner   r   r   Úadd_body_lines‘  s    

þÿz'TableBuilderVerboseMixin.add_body_linesúIterator[str]c                 c  s   | j D ]}|› d�V  qdS )z7Iterator with string representation of non-null counts.z	 non-nullN)r8   )r1   r^   r   r   r   Ú_gen_non_null_counts›  s    
z-TableBuilderVerboseMixin._gen_non_null_countsc                 c  s   | j D ]}t|ƒV  qdS )z5Iterator with string representation of column dtypes.N)r2   r   )r1   Zdtyper   r   r   Ú_gen_dtypes   s    
z$TableBuilderVerboseMixin._gen_dtypesN)rK   rL   rM   rN   r¥   rO   rP   r   rª   r®   r³   r±   r¹   r·   r¸   r¾   rÁ   rÃ   rÅ   rÆ   r   r   r   r   r¤   I  s,   
	
r¤   c                   @  sˆ   e Zd ZdZddddœdd„Zddœd	d
„Zeddœdd„ƒZddœdd„Zddœdd„Z	ddœdd„Z
ddœdd„Zddœdd„ZdS )r€   z:
    Dataframe info table builder for verbose output.
    rQ   rv   rC   ©rb   r~   r   c                C  s(   || _ || _t|  ¡ ƒ| _|  ¡ | _d S rI   ©rb   r~   rµ   r¹   r¦   r³   r§   ©r1   rb   r~   r   r   r   rU   «  s    z%DataFrameTableBuilderVerbose.__init__r/   c                 C  sJ   |   ¡  |  ¡  |  ¡  |  ¡  |  ¡  |  ¡  |  ¡  | jrF|  ¡  dS r�   )	r‹   r�   r¡   r¾   rÁ   rÃ   r˜   rˆ   r    r0   r   r   r   rœ   ¶  s    z1DataFrameTableBuilderVerbose._fill_non_empty_infor¨   c                 C  s   | j rddddgS dddgS )r©   z # ZColumnúNon-Null Countr   ©r~   r0   r   r   r   rª   Â  s    z$DataFrameTableBuilderVerbose.headersc                 C  s   | j  d| j› d�¡ d S )NzData columns (total z
 columns):)r†   r‰   r\   r0   r   r   r   r¡   É  s    z5DataFrameTableBuilderVerbose.add_columns_summary_liner¶   c                 c  s"   t |  ¡ |  ¡ |  ¡ ƒE dH  dS r»   )r²   Ú_gen_line_numbersÚ_gen_columnsrÆ   r0   r   r   r   r¸   Ì  s
    ýz5DataFrameTableBuilderVerbose._gen_rows_without_countsc                 c  s(   t |  ¡ |  ¡ |  ¡ |  ¡ ƒE dH  dS rº   )r²   rÌ   rÍ   rÅ   rÆ   r0   r   r   r   r·   Ô  s    üz2DataFrameTableBuilderVerbose._gen_rows_with_countsrÄ   c                 c  s$   t | jƒD ]\}}d|› �V  q
dS )z6Iterator with string representation of column numbers.r$   N)Ú	enumeraterY   )r1   ÚiÚ_r   r   r   rÌ   Ý  s    z.DataFrameTableBuilderVerbose._gen_line_numbersc                 c  s   | j D ]}t|ƒV  qdS )z4Iterator with string representation of column names.N)rY   r   )r1   r­   r   r   r   rÍ   â  s    
z)DataFrameTableBuilderVerbose._gen_columnsN)rK   rL   rM   rN   rU   rœ   rP   rª   r¡   r¸   r·   rÌ   rÍ   r   r   r   r   r€   ¦  s   	r€   c                   @  s`   e Zd ZdZdddœdd„Zddœd	d
„Zeddœdd„ƒZddœdd„Ze	ddœdd„ƒZ
dS )r‚   z‡
    Abstract builder for series info table.

    Parameters
    ----------
    info : SeriesInfo.
        Instance of SeriesInfo.
    rg   rC   r™   c                C  s
   || _ d S rI   r   rš   r   r   r   rU   ò  s    zSeriesTableBuilder.__init__r…   r/   c                 C  s   g | _ |  ¡  | j S rI   )r†   rœ   r0   r   r   r   rl   õ  s    zSeriesTableBuilder.get_linesr   c                 C  s   | j jS )zSeries.r‡   r0   r   r   r   r-   ú  s    zSeriesTableBuilder.datac                 C  s   | j  d| j› �¡ dS rž   rŸ   r0   r   r   r   r    ÿ  s    z(SeriesTableBuilder.add_memory_usage_linec                 C  s   dS ©z<Add lines to the info table, pertaining to non-empty series.Nr   r0   r   r   r   rœ     s    z'SeriesTableBuilder._fill_non_empty_infoN)rK   rL   rM   rN   rU   rl   rP   r-   r    r   rœ   r   r   r   r   r‚   è  s   	r‚   c                   @  s   e Zd ZdZddœdd„ZdS )r„   z;
    Series info table builder for non-verbose output.
    rC   r/   c                 C  s*   |   ¡  |  ¡  |  ¡  | jr&|  ¡  dS rÑ   )r‹   r�   r˜   rˆ   r    r0   r   r   r   rœ     s
    z1SeriesTableBuilderNonVerbose._fill_non_empty_infoN)rK   rL   rM   rN   rœ   r   r   r   r   r„     s   r„   c                   @  sl   e Zd ZdZddddœdd„Zddœd	d
„Zddœdd„Zeddœdd„ƒZddœdd„Z	ddœdd„Z
dS )rƒ   z7
    Series info table builder for verbose output.
    rg   rv   rC   rÇ   c                C  s(   || _ || _t|  ¡ ƒ| _|  ¡ | _d S rI   rÈ   rÉ   r   r   r   rU     s    z"SeriesTableBuilderVerbose.__init__r/   c                 C  sJ   |   ¡  |  ¡  |  ¡  |  ¡  |  ¡  |  ¡  |  ¡  | jrF|  ¡  dS rÑ   )	r‹   r�   Úadd_series_name_liner¾   rÁ   rÃ   r˜   rˆ   r    r0   r   r   r   rœ   &  s    z.SeriesTableBuilderVerbose._fill_non_empty_infoc                 C  s   | j  d| jj› �¡ d S )NzSeries name: )r†   r‰   r-   r£   r0   r   r   r   rÒ   2  s    z.SeriesTableBuilderVerbose.add_series_name_liner¨   c                 C  s   | j rddgS dgS )r©   rÊ   r   rË   r0   r   r   r   rª   5  s    z!SeriesTableBuilderVerbose.headersr¶   c                 c  s   |   ¡ E dH  dS r»   )rÆ   r0   r   r   r   r¸   <  s    z2SeriesTableBuilderVerbose._gen_rows_without_countsc                 c  s   t |  ¡ |  ¡ ƒE dH  dS rº   )r²   rÅ   rÆ   r0   r   r   r   r·   @  s    þz/SeriesTableBuilderVerbose._gen_rows_with_countsN)rK   rL   rM   rN   rU   rœ   rÒ   rP   rª   r¸   r·   r   r   r   r   rƒ     s   rƒ   r3   )Údfr   c                 C  s   | j  ¡  dd„ ¡ ¡ S )zK
    Create mapping between datatypes and their number of occurrences.
    c                 S  s   | j S rI   r¢   )r%   r   r   r   Ú<lambda>M  ó    z-_get_dataframe_dtype_counts.<locals>.<lambda>)r2   Zvalue_countsÚgroupbyr`   )rÓ   r   r   r   rV   H  s    rV   )N)7Ú
__future__r   Úabcr   r   rm   Útextwrapr   Útypingr   r   r   r	   r
   Zpandas._configr   Zpandas._typingr   r   Zpandas.io.formatsr   ro   Zpandas.io.formats.printingr   Zpandasr   r   r   Zframe_max_cols_subr   Zframe_examples_subZframe_see_also_subZframe_sub_kwargsZseries_examples_subZseries_see_also_subZseries_sub_kwargsZINFO_DOCSTRINGr   r&   r*   r+   rQ   rg   rj   rc   ri   rq   r|   r�   r¤   r€   r‚   r„   rƒ   rV   r   r   r   r   Ú<module>   s„   ÿ
ÿÿVÿ	ùÿ>ÿùÿ3  ÿ	SI<Q+83]B 2