
    OwgF                       U d dl mZ d dlmZ d dlZd dlmZ d dlm	Z	 d dl
mZmZ  G d de      Z G d	 d
e      ZdZe G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Ze G d de             Z ej0                  ej2                         e        ej0                  ej4                         e        ej0                  ej6                         e        ej0                  ej8                         e        ej0                  ej:                         e        ej0                  ej<                         e        ej0                  ej>                         e        ej0                  ej@                         e       iZ!de"d<   y)    )annotations)ClassVarN)register_extension_dtype)is_integer_dtype)NumericArrayNumericDtypec                      e Zd ZdZ ej
                  ej                        ZeZ	e
dd       Ze
dd       Ze
dd       Zy)	IntegerDtypea'  
    An ExtensionDtype to hold a single size & kind of integer dtype.

    These specific implementations are subclasses of the non-public
    IntegerDtype. For example, we have Int8Dtype to represent signed int 8s.

    The attributes name & type are set when these subclasses are created.
    c                    t         S )zq
        Return the array type associated with this dtype.

        Returns
        -------
        type
        )IntegerArrayclss    Q/var/www/horilla/myenv/lib/python3.12/site-packages/pandas/core/arrays/integer.pyconstruct_array_typez!IntegerDtype.construct_array_type   s
         c                    t         S )N)NUMPY_INT_TO_DTYPEr   s    r   _get_dtype_mappingzIntegerDtype._get_dtype_mapping(   s    !!r   c           	        	 |j                  |d|      S # t        $ ra}|j                  ||      }||k(  j                         r|cY d}~S t        d|j                   dt	        j                  |             |d}~ww xY w)z
        Safely cast the values to the given dtype.

        "safe" in this context means the casting is lossless. e.g. if 'values'
        has a floating dtype, each value must be an integer.
        safe)castingcopy)r   Nz"cannot safely cast non-equivalent z to )astype	TypeErroralldtypenp)r   valuesr   r   errcasteds         r   
_safe_castzIntegerDtype._safe_cast,   s    		==T=BB 	]]5t]4F& %%'4V\\N$rxxPUFWX	s     	B 'A;B /A;;B N)returnztype[IntegerArray])r"   dict[np.dtype, IntegerDtype])r   
np.ndarrayr   znp.dtyper   boolr"   r$   )__name__
__module____qualname____doc__r   r   int64_default_np_dtyper   _checkerclassmethodr   r   r!    r   r   r
   r
      s]     !*H  " "  r   r
   c                       e Zd ZdZeZdZdZdZy)r   a  
    Array of integer (optional missing) values.

    Uses :attr:`pandas.NA` as the missing value.

    .. warning::

       IntegerArray is currently experimental, and its API or internal
       implementation may change without warning.

    We represent an IntegerArray with 2 numpy arrays:

    - data: contains a numpy integer array of the appropriate dtype
    - mask: a boolean array holding a mask on the data, True is missing

    To construct an IntegerArray from generic array-like input, use
    :func:`pandas.array` with one of the integer dtypes (see examples).

    See :ref:`integer_na` for more.

    Parameters
    ----------
    values : numpy.ndarray
        A 1-d integer-dtype array.
    mask : numpy.ndarray
        A 1-d boolean-dtype array indicating missing values.
    copy : bool, default False
        Whether to copy the `values` and `mask`.

    Attributes
    ----------
    None

    Methods
    -------
    None

    Returns
    -------
    IntegerArray

    Examples
    --------
    Create an IntegerArray with :func:`pandas.array`.

    >>> int_array = pd.array([1, None, 3], dtype=pd.Int32Dtype())
    >>> int_array
    <IntegerArray>
    [1, <NA>, 3]
    Length: 3, dtype: Int32

    String aliases for the dtypes are also available. They are capitalized.

    >>> pd.array([1, None, 3], dtype='Int32')
    <IntegerArray>
    [1, <NA>, 3]
    Length: 3, dtype: Int32

    >>> pd.array([1, None, 3], dtype='UInt16')
    <IntegerArray>
    [1, <NA>, 3]
    Length: 3, dtype: UInt16
       r   N)	r&   r'   r(   r)   r
   
_dtype_cls_internal_fill_value_truthy_value_falsey_valuer.   r   r   r   r   @   s#    >@ J  MMr   r   a  
An ExtensionDtype for {dtype} integer data.

Uses :attr:`pandas.NA` as its missing value, rather than :attr:`numpy.nan`.

Attributes
----------
None

Methods
-------
None

Examples
--------
For Int8Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.Int8Dtype())
>>> ser.dtype
Int8Dtype()

For Int16Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.Int16Dtype())
>>> ser.dtype
Int16Dtype()

For Int32Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.Int32Dtype())
>>> ser.dtype
Int32Dtype()

For Int64Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.Int64Dtype())
>>> ser.dtype
Int64Dtype()

For UInt8Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.UInt8Dtype())
>>> ser.dtype
UInt8Dtype()

For UInt16Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.UInt16Dtype())
>>> ser.dtype
UInt16Dtype()

For UInt32Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.UInt32Dtype())
>>> ser.dtype
UInt32Dtype()

For UInt64Dtype:

>>> ser = pd.Series([2, pd.NA], dtype=pd.UInt64Dtype())
>>> ser.dtype
UInt64Dtype()
c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)	Int8DtypeInt8ClassVar[str]nameint8r   N)r&   r'   r(   r   r:   typer9   __annotations___dtype_docstringformatr)   r.   r   r   r6   r6      s(    77D D- %%F%3Gr   r6   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)
Int16DtypeInt16r8   r9   int16r;   N)r&   r'   r(   r   rC   r<   r9   r=   r>   r?   r)   r.   r   r   rA   rA      (    88D!D-!%%G%4Gr   rA   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)
Int32DtypeInt32r8   r9   int32r;   N)r&   r'   r(   r   rH   r<   r9   r=   r>   r?   r)   r.   r   r   rF   rF      rD   r   rF   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)
Int64DtypeInt64r8   r9   r*   r;   N)r&   r'   r(   r   r*   r<   r9   r=   r>   r?   r)   r.   r   r   rJ   rJ      rD   r   rJ   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)
UInt8DtypeUInt8r8   r9   uint8r;   N)r&   r'   r(   r   rO   r<   r9   r=   r>   r?   r)   r.   r   r   rM   rM      rD   r   rM   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)UInt16DtypeUInt16r8   r9   uint16r;   N)r&   r'   r(   r   rS   r<   r9   r=   r>   r?   r)   r.   r   r   rQ   rQ      (    99D"D-"%%H%5Gr   rQ   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)UInt32DtypeUInt32r8   r9   uint32r;   N)r&   r'   r(   r   rX   r<   r9   r=   r>   r?   r)   r.   r   r   rV   rV      rT   r   rV   c                  X    e Zd ZU ej                  ZdZded<   ej                  d      Z
y)UInt64DtypeUInt64r8   r9   uint64r;   N)r&   r'   r(   r   r\   r<   r9   r=   r>   r?   r)   r.   r   r   rZ   rZ      rT   r   rZ   r#   r   )#
__future__r   typingr   numpyr   pandas.core.dtypes.baser   pandas.core.dtypes.commonr   pandas.core.arrays.numericr   r   r
   r   r>   r6   rA   rF   rJ   rM   rQ   rV   rZ   r   r:   rC   rH   r*   rO   rS   rX   r\   r   r=   r.   r   r   <module>rc      s   "   < 6-< -`I< IX> F 4 4 4 5 5 5 5 5 5 5 5 5 5 5 5 6, 6 6 6, 6 6 6, 6 6 BHHRWWy{BHHRXX
BHHRXX
BHHRXX
BHHRXX
BHHRYYBHHRYYBHHRYY	4 0 	r   