U
    +if                     @  s  U d Z ddlmZ ddlZddlmZmZmZmZm	Z	m
Z
mZmZmZ ddlZddlm  mZ ddlmZmZmZmZmZmZ ddlmZ ddlmZ ddlm Z  dd	l!m"Z"m#Z# dd
l$m%Z%m&Z&m'Z'm(Z(m)Z) ddl*m+Z+m,Z,m-Z- ddl.m/Z/m0Z0 ddl1m2Z2m3Z3m4Z4 ddl5m6Z6 ddl7m8Z8m9Z9m:Z: ddl;m<Z< ddl=m>Z> ddl?m@Z@mAZAmBZB ervddlmCZCmDZD ddlEmFZF i ZGdeHd< dddddZIe
dddZJG dd de6ZKG dd dZLG d d! d!eMZNG d"d# d#eMZOG d$d% d%ee ZPG d&d de<ZQdS )'z.
Base and utility classes for pandas objects.
    )annotationsN)	TYPE_CHECKINGAnyGenericHashableLiteralTypeVarcastfinaloverload)	ArrayLikeDtypeObj
IndexLabelNDFrameTShapenpt)PYPY)functionAbstractMethodError)cache_readonlydoc)is_categorical_dtypeis_dict_likeis_extension_array_dtypeis_object_dtype	is_scalar)ABCDataFrameABCIndex	ABCSeries)isnaremove_na_arraylike)
algorithmsnanopsops)DirNamesMixin)
duplicatedunique1dvalue_counts)OpsMixin)ExtensionArray)!create_series_with_explicit_dtypeensure_wrapped_if_datetimelikeextract_array)NumpySorterNumpyValueArrayLike)Categoricalzdict[str, str]_shared_docsIndexOpsMixin )klassZinplaceuniquer&   _T)boundc                      s\   e Zd ZU dZded< edd Zdddd	ZddddddZdd fddZ	  Z
S )PandasObjectz/
    Baseclass for various pandas objects.
    zdict[str, Any]_cachec                 C  s   t | S )zJ
        Class constructor (for this class it's just `__class__`.
        )typeself r=   C/home/mars/bis/venv/lib/python3.8/site-packages/pandas/core/base.py_constructorh   s    zPandasObject._constructorstrreturnc                 C  s
   t | S )zI
        Return a string representation for a particular object.
        )object__repr__r;   r=   r=   r>   rD   o   s    zPandasObject.__repr__Nz
str | NoneNone)keyrB   c                 C  s4   t | dsdS |dkr"| j  n| j|d dS )zV
        Reset cached properties. If ``key`` is passed, only clears that key.
        r9   N)hasattrr9   clearpop)r<   rF   r=   r=   r>   _reset_cachev   s
    
zPandasObject._reset_cacheintc                   s<   t | dd}|r2|dd}tt|r(|n| S t  S )zx
        Generates the total memory usage for an object that returns
        either a value or Series of values
        memory_usageNTdeep)getattrrK   r   sumsuper
__sizeof__)r<   rL   Zmem	__class__r=   r>   rR      s
    
zPandasObject.__sizeof__)N)__name__
__module____qualname____doc____annotations__propertyr?   rD   rJ   rR   __classcell__r=   r=   rS   r>   r8   `   s   

r8   c                   @  s&   e Zd ZdZdd ZddddZdS )	NoNewAttributesMixina  
    Mixin which prevents adding new attributes.

    Prevents additional attributes via xxx.attribute = "something" after a
    call to `self.__freeze()`. Mainly used to prevent the user from using
    wrong attributes on an accessor (`Series.cat/.str/.dt`).

    If you really want to add a new attribute at a later time, you need to use
    `object.__setattr__(self, key, value)`.
    c                 C  s   t | dd dS )z9
        Prevents setting additional attributes.
        __frozenTN)rC   __setattr__r;   r=   r=   r>   _freeze   s    zNoNewAttributesMixin._freezer@   )rF   c                 C  sT   t | ddrB|dksB|t| jksBt | |d d k	sBtd| dt| || d S )Nr]   Fr9   z"You cannot add any new attribute '')rO   r:   __dict__AttributeErrorrC   r^   )r<   rF   valuer=   r=   r>   r^      s    z NoNewAttributesMixin.__setattr__N)rU   rV   rW   rX   r_   r^   r=   r=   r=   r>   r\      s   r\   c                   @  s   e Zd ZdS )	DataErrorNrU   rV   rW   r=   r=   r=   r>   rd      s   rd   c                   @  s   e Zd ZdS )SpecificationErrorNre   r=   r=   r=   r>   rf      s   rf   c                   @  s   e Zd ZU dZded< dZded< ded< d	d
gZeeZe	e
dd Zedd Ze	eddddZe	edd Zdd ZdddddZdd ZeZdS )SelectionMixinz
    mixin implementing the selection & aggregation interface on a group-like
    object sub-classes need to define: obj, exclusions
    r   objNzIndexLabel | None
_selectionzfrozenset[Hashable]
exclusionsr9   __setstate__c                 C  s&   t | jtttttjfs | jgS | jS N)
isinstanceri   listtupler   r   npndarrayr;   r=   r=   r>   _selection_list   s     zSelectionMixin._selection_listc                 C  s,   | j d kst| jtr| jS | j| j  S d S rl   )ri   rm   rh   r   r;   r=   r=   r>   _selected_obj   s    zSelectionMixin._selected_objrK   rA   c                 C  s   | j jS rl   )rs   ndimr;   r=   r=   r>   rt      s    zSelectionMixin.ndimc                 C  sP   | j d k	r"t| jtr"| j| j S t| jdkrF| jj| jddddS | jS d S )Nr      FT)axisZconsolidateZ
only_slice)ri   rm   rh   r   rr   lenrj   Z
_drop_axisr;   r=   r=   r>   _obj_with_exclusions   s       z#SelectionMixin._obj_with_exclusionsc                 C  s  | j d k	rtd| j  dt|tttttjfrt	| j
j|t	t|krtt|| j
j}tdt|dd  | jt|ddS t| dd	s|| j
jkrtd
| | j|ddS || j
krtd
| | j
| }|j}| j|||dS d S )Nz
Column(s) z already selectedzColumns not found: ru      rt   Zas_indexFzColumn not found: )rt   subset)ri   
IndexErrorrm   rn   ro   r   r   rp   rq   rw   rh   columnsintersectionset
differenceKeyErrorr@   _gotitemrO   rt   )r<   rF   bad_keysr|   rt   r=   r=   r>   __getitem__   s     


zSelectionMixin.__getitem__r{   c                 C  s   t | dS )a  
        sub-classes to define
        return a sliced object

        Parameters
        ----------
        key : str / list of selections
        ndim : {1, 2}
            requested ndim of result
        subset : object, default None
            subset to act on
        Nr   )r<   rF   rt   r|   r=   r=   r>   r      s    zSelectionMixin._gotitemc                 O  s   t | d S rl   r   )r<   funcargskwargsr=   r=   r>   	aggregate  s    zSelectionMixin.aggregate)N)rU   rV   rW   rX   rY   ri   Z_internal_namesr   Z_internal_names_setr
   rZ   rr   r   rs   rt   rx   r   r   r   Zaggr=   r=   r=   r>   rg      s*   

rg   c                   @  s  e Zd ZU dZdZedgZded< edddd	Z	ed
dddZ
dddddZeeddZeddddZddddZeddddZdd ZeddddZedddd Zed!dd"d#Zd$d%ejfd&d'd(d)d*d+Zed'dd,d-Zdd'd/d0d1Zed2d3d4d5dd'dd6d7d8Zdd'd/d9d:Zeed3d2d;d5dddd<d=Zd>d? ZeZd@dA Ze d'ddBdCZ!dDdE Z"dFd.d$d$dGdHdIdJdKZ#e$ddLdMZ%dd'd'd'd'dNdOdPZ&dQdR Z'dd'ddSdTdUZ(ed'ddVdWZ)ed'ddXdYZ*ed'ddZd[Z+ed'dd\d]Z,dd'dd^d_d`Z-ee.j/dadadae01dbdcdd'dedfdgdhZ/die2dj< e3ddldmdndodpdqdrZ4e3ddsdmdndtdpdudrZ4ee2dj dvdwddydmdndzdpd{drZ4dd}d~Z5e$ddddddZ6dd Z7dd Z8d$S )r2   zS
    Common ops mixin to support a unified interface / docs for Series / Index
    i  tolistzfrozenset[str]_hidden_attrsr   rA   c                 C  s   t | d S rl   r   r;   r=   r=   r>   dtype  s    zIndexOpsMixin.dtypezExtensionArray | np.ndarrayc                 C  s   t | d S rl   r   r;   r=   r=   r>   _values$  s    zIndexOpsMixin._valuesr6   )r<   rB   c                 O  s   t || | S )zw
        Return the transpose, which is by definition self.

        Returns
        -------
        %(klass)s
        )nvZvalidate_transpose)r<   r   r   r=   r=   r>   	transpose)  s    zIndexOpsMixin.transposezD
        Return the transpose, which is by definition self.
        )r   r   c                 C  s   | j jS )zE
        Return a tuple of the shape of the underlying data.
        )r   shaper;   r=   r=   r>   r   ;  s    zIndexOpsMixin.shaperK   c                 C  s   t | d S rl   r   r;   r=   r=   r>   __len__B  s    zIndexOpsMixin.__len__c                 C  s   dS )zO
        Number of dimensions of the underlying data, by definition 1.
        ru   r=   r;   r=   r=   r>   rt   F  s    zIndexOpsMixin.ndimc                 C  s$   t | dkrtt| S tddS )a  
        Return the first element of the underlying data as a Python scalar.

        Returns
        -------
        scalar
            The first element of %(klass)s.

        Raises
        ------
        ValueError
            If the data is not length-1.
        ru   z6can only convert an array of size 1 to a Python scalarN)rw   nextiter
ValueErrorr;   r=   r=   r>   itemM  s    zIndexOpsMixin.itemc                 C  s   | j jS )zD
        Return the number of bytes in the underlying data.
        )r   nbytesr;   r=   r=   r>   r   _  s    zIndexOpsMixin.nbytesc                 C  s
   t | jS )zG
        Return the number of elements in the underlying data.
        )rw   r   r;   r=   r=   r>   sizef  s    zIndexOpsMixin.sizer*   c                 C  s   t | dS )aM  
        The ExtensionArray of the data backing this Series or Index.

        Returns
        -------
        ExtensionArray
            An ExtensionArray of the values stored within. For extension
            types, this is the actual array. For NumPy native types, this
            is a thin (no copy) wrapper around :class:`numpy.ndarray`.

            ``.array`` differs ``.values`` which may require converting the
            data to a different form.

        See Also
        --------
        Index.to_numpy : Similar method that always returns a NumPy array.
        Series.to_numpy : Similar method that always returns a NumPy array.

        Notes
        -----
        This table lays out the different array types for each extension
        dtype within pandas.

        ================== =============================
        dtype              array type
        ================== =============================
        category           Categorical
        period             PeriodArray
        interval           IntervalArray
        IntegerNA          IntegerArray
        string             StringArray
        boolean            BooleanArray
        datetime64[ns, tz] DatetimeArray
        ================== =============================

        For any 3rd-party extension types, the array type will be an
        ExtensionArray.

        For all remaining dtypes ``.array`` will be a
        :class:`arrays.NumpyExtensionArray` wrapping the actual ndarray
        stored within. If you absolutely need a NumPy array (possibly with
        copying / coercing data), then use :meth:`Series.to_numpy` instead.

        Examples
        --------
        For regular NumPy types like int, and float, a PandasArray
        is returned.

        >>> pd.Series([1, 2, 3]).array
        <PandasArray>
        [1, 2, 3]
        Length: 3, dtype: int64

        For extension types, like Categorical, the actual ExtensionArray
        is returned

        >>> ser = pd.Series(pd.Categorical(['a', 'b', 'a']))
        >>> ser.array
        ['a', 'b', 'a']
        Categories (2, object): ['a', 'b']
        Nr   r;   r=   r=   r>   arraym  s    ?zIndexOpsMixin.arrayNFznpt.DTypeLike | Noneboolz
np.ndarray)r   copyrB   c                 K  s   t | jr$| jj|f||d|S |rHt| d }td| dtj| j	|d}|sf|t
jk	r| }|t
jk	r|||  < |S )a  
        A NumPy ndarray representing the values in this Series or Index.

        Parameters
        ----------
        dtype : str or numpy.dtype, optional
            The dtype to pass to :meth:`numpy.asarray`.
        copy : bool, default False
            Whether to ensure that the returned value is not a view on
            another array. Note that ``copy=False`` does not *ensure* that
            ``to_numpy()`` is no-copy. Rather, ``copy=True`` ensure that
            a copy is made, even if not strictly necessary.
        na_value : Any, optional
            The value to use for missing values. The default value depends
            on `dtype` and the type of the array.

            .. versionadded:: 1.0.0

        **kwargs
            Additional keywords passed through to the ``to_numpy`` method
            of the underlying array (for extension arrays).

            .. versionadded:: 1.0.0

        Returns
        -------
        numpy.ndarray

        See Also
        --------
        Series.array : Get the actual data stored within.
        Index.array : Get the actual data stored within.
        DataFrame.to_numpy : Similar method for DataFrame.

        Notes
        -----
        The returned array will be the same up to equality (values equal
        in `self` will be equal in the returned array; likewise for values
        that are not equal). When `self` contains an ExtensionArray, the
        dtype may be different. For example, for a category-dtype Series,
        ``to_numpy()`` will return a NumPy array and the categorical dtype
        will be lost.

        For NumPy dtypes, this will be a reference to the actual data stored
        in this Series or Index (assuming ``copy=False``). Modifying the result
        in place will modify the data stored in the Series or Index (not that
        we recommend doing that).

        For extension types, ``to_numpy()`` *may* require copying data and
        coercing the result to a NumPy type (possibly object), which may be
        expensive. When you need a no-copy reference to the underlying data,
        :attr:`Series.array` should be used instead.

        This table lays out the different dtypes and default return types of
        ``to_numpy()`` for various dtypes within pandas.

        ================== ================================
        dtype              array type
        ================== ================================
        category[T]        ndarray[T] (same dtype as input)
        period             ndarray[object] (Periods)
        interval           ndarray[object] (Intervals)
        IntegerNA          ndarray[object]
        datetime64[ns]     datetime64[ns]
        datetime64[ns, tz] ndarray[object] (Timestamps)
        ================== ================================

        Examples
        --------
        >>> ser = pd.Series(pd.Categorical(['a', 'b', 'a']))
        >>> ser.to_numpy()
        array(['a', 'b', 'a'], dtype=object)

        Specify the `dtype` to control how datetime-aware data is represented.
        Use ``dtype=object`` to return an ndarray of pandas :class:`Timestamp`
        objects, each with the correct ``tz``.

        >>> ser = pd.Series(pd.date_range('2000', periods=2, tz="CET"))
        >>> ser.to_numpy(dtype=object)
        array([Timestamp('2000-01-01 00:00:00+0100', tz='CET'),
               Timestamp('2000-01-02 00:00:00+0100', tz='CET')],
              dtype=object)

        Or ``dtype='datetime64[ns]'`` to return an ndarray of native
        datetime64 values. The values are converted to UTC and the timezone
        info is dropped.

        >>> ser.to_numpy(dtype="datetime64[ns]")
        ... # doctest: +ELLIPSIS
        array(['1999-12-31T23:00:00.000000000', '2000-01-01T23:00:00...'],
              dtype='datetime64[ns]')
        )r   na_valuer   z/to_numpy() got an unexpected keyword argument 'r`   )r   )r   r   r   to_numpyrn   keys	TypeErrorrp   asarrayr   lib
no_defaultr   r    )r<   r   r   r   r   r   resultr=   r=   r>   r     s    c


zIndexOpsMixin.to_numpyc                 C  s   | j  S rl   )r   r;   r=   r=   r>   empty!  s    zIndexOpsMixin.emptyTskipnac                 O  s&   t | t || tj| j|dS )a  
        Return the maximum value of the Index.

        Parameters
        ----------
        axis : int, optional
            For compatibility with NumPy. Only 0 or None are allowed.
        skipna : bool, default True
            Exclude NA/null values when showing the result.
        *args, **kwargs
            Additional arguments and keywords for compatibility with NumPy.

        Returns
        -------
        scalar
            Maximum value.

        See Also
        --------
        Index.min : Return the minimum value in an Index.
        Series.max : Return the maximum value in a Series.
        DataFrame.max : Return the maximum values in a DataFrame.

        Examples
        --------
        >>> idx = pd.Index([3, 2, 1])
        >>> idx.max()
        3

        >>> idx = pd.Index(['c', 'b', 'a'])
        >>> idx.max()
        'c'

        For a MultiIndex, the maximum is determined lexicographically.

        >>> idx = pd.MultiIndex.from_product([('a', 'b'), (2, 1)])
        >>> idx.max()
        ('b', 2)
        r   )r   validate_minmax_axisZvalidate_maxr#   Znanmaxr   r<   rv   r   r   r   r=   r=   r>   max%  s    (
zIndexOpsMixin.maxr   minlargest)opZopposerc   )r   rB   c                 O  sX   | j }t| t|||}t|trF|s<|  r<dS | S nt	j
||dS dS )aS  
        Return int position of the {value} value in the Series.

        If the {op}imum is achieved in multiple locations,
        the first row position is returned.

        Parameters
        ----------
        axis : {{None}}
            Dummy argument for consistency with Series.
        skipna : bool, default True
            Exclude NA/null values when showing the result.
        *args, **kwargs
            Additional arguments and keywords for compatibility with NumPy.

        Returns
        -------
        int
            Row position of the {op}imum value.

        See Also
        --------
        Series.arg{op} : Return position of the {op}imum value.
        Series.arg{oppose} : Return position of the {oppose}imum value.
        numpy.ndarray.arg{op} : Equivalent method for numpy arrays.
        Series.idxmax : Return index label of the maximum values.
        Series.idxmin : Return index label of the minimum values.

        Examples
        --------
        Consider dataset containing cereal calories

        >>> s = pd.Series({{'Corn Flakes': 100.0, 'Almond Delight': 110.0,
        ...                'Cinnamon Toast Crunch': 120.0, 'Cocoa Puff': 110.0}})
        >>> s
        Corn Flakes              100.0
        Almond Delight           110.0
        Cinnamon Toast Crunch    120.0
        Cocoa Puff               110.0
        dtype: float64

        >>> s.argmax()
        2
        >>> s.argmin()
        0

        The maximum cereal calories is the third element and
        the minimum cereal calories is the first element,
        since series is zero-indexed.
        ry   r   N)r   r   r   Zvalidate_argmax_with_skipnarm   r*   r    anyargmaxr#   Z	nanargmaxr<   rv   r   r   r   Zdelegater=   r=   r>   r   Q  s    4


 zIndexOpsMixin.argmaxc                 O  s&   t | t || tj| j|dS )a  
        Return the minimum value of the Index.

        Parameters
        ----------
        axis : {None}
            Dummy argument for consistency with Series.
        skipna : bool, default True
            Exclude NA/null values when showing the result.
        *args, **kwargs
            Additional arguments and keywords for compatibility with NumPy.

        Returns
        -------
        scalar
            Minimum value.

        See Also
        --------
        Index.max : Return the maximum value of the object.
        Series.min : Return the minimum value in a Series.
        DataFrame.min : Return the minimum values in a DataFrame.

        Examples
        --------
        >>> idx = pd.Index([3, 2, 1])
        >>> idx.min()
        1

        >>> idx = pd.Index(['c', 'b', 'a'])
        >>> idx.min()
        'a'

        For a MultiIndex, the minimum is determined lexicographically.

        >>> idx = pd.MultiIndex.from_product([('a', 'b'), (2, 1)])
        >>> idx.min()
        ('a', 1)
        r   )r   r   Zvalidate_minr#   Znanminr   r   r=   r=   r>   r     s    (
zIndexOpsMixin.minsmallestc                 O  sX   | j }t| t|||}t|trF|s<|  r<dS | S nt	j
||dS d S )Nry   r   )r   r   r   Zvalidate_argmin_with_skipnarm   r*   r    r   argminr#   Z	nanargminr   r=   r=   r>   r     s    


 zIndexOpsMixin.argminc                 C  s
   | j  S )a  
        Return a list of the values.

        These are each a scalar type, which is a Python scalar
        (for str, int, float) or a pandas scalar
        (for Timestamp/Timedelta/Interval/Period)

        Returns
        -------
        list

        See Also
        --------
        numpy.ndarray.tolist : Return the array as an a.ndim-levels deep
            nested list of Python scalars.
        )r   r   r;   r=   r=   r>   r     s    zIndexOpsMixin.tolistc                 C  s2   t | jtjst| jS t| jjt| jjS dS )a  
        Return an iterator of the values.

        These are each a scalar type, which is a Python scalar
        (for str, int, float) or a pandas scalar
        (for Timestamp/Timedelta/Interval/Period)

        Returns
        -------
        iterator
        N)	rm   r   rp   rq   r   mapr   ranger   r;   r=   r=   r>   __iter__  s    
zIndexOpsMixin.__iter__c                 C  s   t t|  S )zc
        Return True if there are any NaNs.

        Enables various performance speedups.
        )r   r    r   r;   r=   r=   r>   hasnans  s    zIndexOpsMixin.hasnansc                 C  s
   t | jS rl   )r    r   r;   r=   r=   r>   r      s    zIndexOpsMixin.isnar   )rv   r   numeric_onlyfilter_typer@   namec          	      K  s>   t | |d}|dkr,tt| j d| |f d|i|S )zA
        Perform the reduction type operation if we can.
        Nz cannot perform the operation r   )rO   r   r:   rU   )	r<   r   r   rv   r   r   r   kwdsr   r=   r=   r>   _reduce  s    zIndexOpsMixin._reducec           	        s  t |r<t|tr.t|dr.|  fdd}nt|tjd}t|trt| j	rft
d| j}||S | j}|j|}t|j|}|S t| j	rt| jdr| j}|dk	rtdd }nB| jt}|d	krd
d }n$|dkrtj}nd| d}t||||}|S )a  
        An internal function that maps values using the input
        correspondence (which can be a dict, Series, or function).

        Parameters
        ----------
        mapper : function, dict, or Series
            The input correspondence object
        na_action : {None, 'ignore'}
            If 'ignore', propagate NA values, without passing them to the
            mapping function

        Returns
        -------
        Union[Index, MultiIndex], inferred
            The output of the mapping function applied to the index.
            If the function returns a tuple with more than one element
            a MultiIndex will be returned.
        __missing__c                   s    |  S rl   r=   )xZdict_with_defaultr=   r>   <lambda>9      z+IndexOpsMixin._map_values.<locals>.<lambda>)Zdtype_if_emptyr0   r   Nc                 S  s
   |  |S rl   )r   valuesfr=   r=   r>   r   _  r   ignorec                 S  s   t | |t| tjS rl   )r   Zmap_infer_maskr    viewrp   Zuint8r   r=   r=   r>   r   c  s     z+na_action must either be 'ignore' or None, z was passed)r   rm   dictrG   r+   rp   Zfloat64r   r   r   r	   r   r   indexZget_indexerr"   Ztake_ndr   NotImplementedErrorZastyperC   r   Z	map_inferr   )	r<   ZmapperZ	na_actioncatr   Zindexer
new_valuesZmap_fmsgr=   r   r>   _map_values  s>     






zIndexOpsMixin._map_values)	normalizesort	ascendingdropnac                 C  s   t | |||||dS )a  
        Return a Series containing counts of unique values.

        The resulting object will be in descending order so that the
        first element is the most frequently-occurring element.
        Excludes NA values by default.

        Parameters
        ----------
        normalize : bool, default False
            If True then the object returned will contain the relative
            frequencies of the unique values.
        sort : bool, default True
            Sort by frequencies.
        ascending : bool, default False
            Sort in ascending order.
        bins : int, optional
            Rather than count values, group them into half-open bins,
            a convenience for ``pd.cut``, only works with numeric data.
        dropna : bool, default True
            Don't include counts of NaN.

        Returns
        -------
        Series

        See Also
        --------
        Series.count: Number of non-NA elements in a Series.
        DataFrame.count: Number of non-NA elements in a DataFrame.
        DataFrame.value_counts: Equivalent method on DataFrames.

        Examples
        --------
        >>> index = pd.Index([3, 1, 2, 3, 4, np.nan])
        >>> index.value_counts()
        3.0    2
        1.0    1
        2.0    1
        4.0    1
        dtype: int64

        With `normalize` set to `True`, returns the relative frequency by
        dividing all values by the sum of values.

        >>> s = pd.Series([3, 1, 2, 3, 4, np.nan])
        >>> s.value_counts(normalize=True)
        3.0    0.4
        1.0    0.2
        2.0    0.2
        4.0    0.2
        dtype: float64

        **bins**

        Bins can be useful for going from a continuous variable to a
        categorical variable; instead of counting unique
        apparitions of values, divide the index in the specified
        number of half-open bins.

        >>> s.value_counts(bins=3)
        (0.996, 2.0]    2
        (2.0, 3.0]      2
        (3.0, 4.0]      1
        dtype: int64

        **dropna**

        With `dropna` set to `False` we can also see NaN index values.

        >>> s.value_counts(dropna=False)
        3.0    2
        1.0    1
        2.0    1
        4.0    1
        NaN    1
        dtype: int64
        )r   r   r   binsr   )r(   )r<   r   r   r   r   r   r=   r=   r>   r(   t  s    VzIndexOpsMixin.value_countsc                 C  sZ   | j }t|tjsN| }| jjdkrVt| trVt| jdd d krVt	|}nt
|}|S )N)mMtz)r   rm   rp   rq   r5   r   kindr   rO   r   r'   )r<   r   r   r=   r=   r>   r5     s    zIndexOpsMixin.unique)r   rB   c                 C  s   |   }|rt|}t|S )a  
        Return number of unique elements in the object.

        Excludes NA values by default.

        Parameters
        ----------
        dropna : bool, default True
            Don't include NaN in the count.

        Returns
        -------
        int

        See Also
        --------
        DataFrame.nunique: Method nunique for DataFrame.
        Series.count: Count non-NA/null observations in the Series.

        Examples
        --------
        >>> s = pd.Series([1, 3, 5, 7, 7])
        >>> s
        0    1
        1    3
        2    5
        3    7
        4    7
        dtype: int64

        >>> s.nunique()
        4
        )r5   r!   rw   )r<   r   Zuniqsr=   r=   r>   nunique  s    "zIndexOpsMixin.nuniquec                 C  s   | j ddt| kS )zr
        Return boolean if values in the object are unique.

        Returns
        -------
        bool
        F)r   )r   rw   r;   r=   r=   r>   	is_unique  s    	zIndexOpsMixin.is_uniquec                 C  s   ddl m} || jS )z
        Return boolean if values in the object are
        monotonic_increasing.

        Returns
        -------
        bool
        r   Index)pandasr   is_monotonicr<   r   r=   r=   r>   r     s    
zIndexOpsMixin.is_monotonicc                 C  s   | j S )z)
        Alias for is_monotonic.
        )r   r;   r=   r=   r>   is_monotonic_increasing!  s    z%IndexOpsMixin.is_monotonic_increasingc                 C  s   ddl m} || jS )z
        Return boolean if values in the object are
        monotonic_decreasing.

        Returns
        -------
        bool
        r   r   )r   r   is_monotonic_decreasingr   r=   r=   r>   r   )  s    
z%IndexOpsMixin.is_monotonic_decreasing)rN   rB   c                 C  sR   t | jdr| jj|dS | jj}|rNt| rNtsNttj| j	}|t
|7 }|S )aN  
        Memory usage of the values.

        Parameters
        ----------
        deep : bool, default False
            Introspect the data deeply, interrogate
            `object` dtypes for system-level memory consumption.

        Returns
        -------
        bytes used

        See Also
        --------
        numpy.ndarray.nbytes : Total bytes consumed by the elements of the
            array.

        Notes
        -----
        Memory usage does not include memory consumed by elements that
        are not components of the array if deep=False or if used on PyPy
        rL   rM   )rG   r   rL   r   r   r   r	   rp   rq   r   r   Zmemory_usage_of_objects)r<   rN   vr   r=   r=   r>   _memory_usage7  s    zIndexOpsMixin._memory_usager3   z            sort : bool, default False
                Sort `uniques` and shuffle `codes` to maintain the
                relationship.
            )r   orderZ	size_hintr   ry   z
int | Noner   na_sentinelc                 C  s   t j| ||dS )Nr   )r"   	factorize)r<   r   r   r=   r=   r>   r   Z  s    zIndexOpsMixin.factorizea  
        Find indices where elements should be inserted to maintain order.

        Find the indices into a sorted {klass} `self` such that, if the
        corresponding elements in `value` were inserted before the indices,
        the order of `self` would be preserved.

        .. note::

            The {klass} *must* be monotonically sorted, otherwise
            wrong locations will likely be returned. Pandas does *not*
            check this for you.

        Parameters
        ----------
        value : array-like or scalar
            Values to insert into `self`.
        side : {{'left', 'right'}}, optional
            If 'left', the index of the first suitable location found is given.
            If 'right', return the last such index.  If there is no suitable
            index, return either 0 or N (where N is the length of `self`).
        sorter : 1-D array-like, optional
            Optional array of integer indices that sort `self` into ascending
            order. They are typically the result of ``np.argsort``.

        Returns
        -------
        int or array of int
            A scalar or array of insertion points with the
            same shape as `value`.

        See Also
        --------
        sort_values : Sort by the values along either axis.
        numpy.searchsorted : Similar method from NumPy.

        Notes
        -----
        Binary search is used to find the required insertion points.

        Examples
        --------
        >>> ser = pd.Series([1, 2, 3])
        >>> ser
        0    1
        1    2
        2    3
        dtype: int64

        >>> ser.searchsorted(4)
        3

        >>> ser.searchsorted([0, 4])
        array([0, 3])

        >>> ser.searchsorted([1, 3], side='left')
        array([0, 2])

        >>> ser.searchsorted([1, 3], side='right')
        array([1, 3])

        >>> ser = pd.Series(pd.to_datetime(['3/11/2000', '3/12/2000', '3/13/2000']))
        >>> ser
        0   2000-03-11
        1   2000-03-12
        2   2000-03-13
        dtype: datetime64[ns]

        >>> ser.searchsorted('3/14/2000')
        3

        >>> ser = pd.Categorical(
        ...     ['apple', 'bread', 'bread', 'cheese', 'milk'], ordered=True
        ... )
        >>> ser
        ['apple', 'bread', 'bread', 'cheese', 'milk']
        Categories (4, object): ['apple' < 'bread' < 'cheese' < 'milk']

        >>> ser.searchsorted('bread')
        1

        >>> ser.searchsorted(['bread'], side='right')
        array([3])

        If the values are not monotonically sorted, wrong locations
        may be returned:

        >>> ser = pd.Series([2, 1, 3])
        >>> ser
        0    2
        1    1
        2    3
        dtype: int64

        >>> ser.searchsorted(1)  # doctest: +SKIP
        0  # wrong result, correct would be 1
        searchsorted.znpt._ScalarLike_cozLiteral[('left', 'right')]r.   znp.intp)rc   sidesorterrB   c                 C  s   d S rl   r=   r<   rc   r   r   r=   r=   r>   r     s    zIndexOpsMixin.searchsortedznpt.ArrayLike | ExtensionArrayznpt.NDArray[np.intp]c                 C  s   d S rl   r=   r   r=   r=   r>   r     s    r   )r4   leftz$NumpyValueArrayLike | ExtensionArrayznpt.NDArray[np.intp] | np.intpc                 C  s4   | j }t|tjs"|j|||dS tj||||dS )N)r   r   )r   rm   rp   rq   r   r"   )r<   rc   r   r   r   r=   r=   r>   r     s    firstc                 C  s   | j |d}| |  S N)keep)_duplicated)r<   r   r&   r=   r=   r>   drop_duplicates  s    zIndexOpsMixin.drop_duplicatesz!Literal[('first', 'last', False)]znpt.NDArray[np.bool_])r   rB   c                 C  s   t | j|dS r   )r&   r   )r<   r   r=   r=   r>   r      s    zIndexOpsMixin._duplicatedc              	   C  sj   t | |}| j}t|ddd}t ||j}t|}tjdd t 	|||}W 5 Q R X | j
||dS )NT)Zextract_numpyZextract_ranger   )allr   )r$   Zget_op_result_namer   r-   Zmaybe_prepare_scalar_for_opr   r,   rp   ZerrstateZarithmetic_op_construct_result)r<   otherr   Zres_nameZlvaluesZrvaluesr   r=   r=   r>   _arith_method  s    zIndexOpsMixin._arith_methodc                 C  s   t | dS )z~
        Construct an appropriately-wrapped result from the ArrayLike result
        of an arithmetic-like operation.
        Nr   )r<   r   r   r=   r=   r>   r     s    zIndexOpsMixin._construct_result)NT)NT)NT)NT)N)FTFNT)T)F)Fry   )..)..)r   N)r   )r   )9rU   rV   rW   rX   Z__array_priority__	frozensetr   rY   rZ   r   r   r   Tr   r   rt   r   r   r   r   r   r   r   r   r   r   r   r   r   r   Zto_listr   r   r   r    r   r
   r   r(   r5   r   r   r   r   r   r   r"   r   textwrapdedentr1   r   r   r   r   r   r   r=   r=   r=   r>   r2     s   
Bs,C,Y     _'
#g      
 )RrX   
__future__r   r   typingr   r   r   r   r   r   r	   r
   r   numpyrp   Zpandas._libs.libZ_libsr   Zpandas._typingr   r   r   r   r   r   Zpandas.compatr   Zpandas.compat.numpyr   r   Zpandas.errorsr   Zpandas.util._decoratorsr   r   Zpandas.core.dtypes.commonr   r   r   r   r   Zpandas.core.dtypes.genericr   r   r   Zpandas.core.dtypes.missingr    r!   Zpandas.corer"   r#   r$   Zpandas.core.accessorr%   Zpandas.core.algorithmsr&   r'   r(   Zpandas.core.arrayliker)   Zpandas.core.arraysr*   Zpandas.core.constructionr+   r,   r-   r.   r/   r   r0   r1   rY   Z_indexops_doc_kwargsr6   r8   r\   	Exceptionrd   rf   rg   r2   r=   r=   r=   r>   <module>   sF   , /"[