o
    ôT·jf•  ã                
   @  sÄ  d Z ddlmZ ddlmZmZ ddlZddlmZm	Z	m
Z
 ddlZddlmZ ddlmZmZ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!m"Z"m#Z#m$Z$m%Z% ddl&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0 ddl1m2Z2m3Z3 ddl4m5Z5 ddl6m7Z7 ddl8m9Z9 ddl:m;Z;m<Z< ddl=m>  m?Z@ ddlAmBZB ddlCm>  mD  mEZF ddlCmGZGmHZHmIZImJZJ ddlKmLZLmMZM ddlNmOZOmPZP ddlQmRZR ddlSmTZTmUZU er÷ddlVmWZW ddlXmYZYmZZZm[Z[m\Z\m]Z] e^eFj_ƒZ_e_ `ddde ad¡dœ¡ dd „ Zbd!d"„ Zcd#d$„ Zdee<d% dd&e_d' d(d)e ad*¡d+œ ƒePd,d-ge;d.d/�ePg d0¢e;ƒePd1d2ge;d.d3�G d4d„ deOƒƒƒƒƒZed@d7d8„Zfd@d9d:„Zg						;dAdBd>d?„ZhdS )Cz define the IntervalIndex é    )Úannotations)ÚleÚltN)ÚTYPE_CHECKINGÚAnyÚLiteral)Úlib)ÚIntervalÚIntervalMixinÚIntervalTree)Ú
BaseOffsetÚPeriodÚ	TimedeltaÚ	TimestampÚ	to_offset)ÚInvalidIndexError)ÚAppenderÚcache_readonly)Úrewrite_exception)Úfind_common_typeÚinfer_dtype_from_scalarÚmaybe_box_datetimelikeÚmaybe_downcast_numericÚmaybe_upcast_numeric_to_64bit)
Úensure_platform_intÚis_float_dtypeÚ
is_integerÚis_integer_dtypeÚis_list_likeÚ	is_numberÚis_object_dtypeÚ	is_scalarÚis_string_dtypeÚpandas_dtype)ÚDatetimeTZDtypeÚIntervalDtype)Úis_valid_na_for_dtype©Úunique)Úvalidate_periods)ÚIntervalArrayÚ_interval_shared_docs)Úis_valid_positional_slice)ÚIndexÚ_index_shared_docsÚensure_indexÚmaybe_extract_name)ÚDatetimeIndexÚ
date_range)ÚExtensionIndexÚinherit_names)Ú
MultiIndex)ÚTimedeltaIndexÚtimedelta_range)ÚHashable)ÚDtypeÚDtypeObjÚIntervalClosedTypeÚSelfÚnptÚIntervalIndexz"IntervalIndex or list of IntervalszX         name : object, optional
              Name to be stored in the index.
         )ÚklassÚqualnameÚtarget_klassÚnamec                 C  sŒ   t | dt| ƒƒ}t| ttfƒrd}t|ƒ}t |d¡s t|tƒr(| t	 
dd¡ S t|ƒr0| d S t|ƒr;t	 | t	j¡S tdtt| ƒƒ› �ƒ‚©NÚdtypezdatetime64[ns]ÚmMé   Únsz%cannot determine next label for type ©ÚgetattrÚtypeÚ
isinstancer   r   r#   r   Úis_np_dtyper$   ÚnpÚtimedelta64r   r   Ú	nextafterÚinfÚ	TypeErrorÚrepr©ÚlabelrD   © rU   ú_/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/core/indexes/interval.pyÚ_get_next_labelv   s   rW   c                 C  sŽ   t | dt| ƒƒ}t| ttfƒrd}t|ƒ}t |d¡s t|tƒr(| t	 
dd¡ S t|ƒr0| d S t|ƒr<t	 | t	j ¡S tdtt| ƒƒ› �ƒ‚rC   rH   rS   rU   rU   rV   Ú_get_prev_label‡   s   rX   c                 C  s   | j di |¤ŽS )zv
    This is called upon unpickling, rather than the default which doesn't have
    arguments and breaks __new__.
    NrU   )Úfrom_arrays)ÚclsÚdrU   rU   rV   Ú_new_IntervalIndex˜   s   r\   Úclassz>Immutable index of intervals that are closed on the same side.rB   zis_overlapping
values
Ú a7      Examples
    --------
    A new ``IntervalIndex`` is typically constructed using
    :func:`interval_range`:

    >>> pd.interval_range(start=0, end=5)
    IntervalIndex([(0, 1], (1, 2], (2, 3], (3, 4], (4, 5]],
                  dtype='interval[int64, right]')

    It may also be constructed using one of the constructor
    methods: :meth:`IntervalIndex.from_arrays`,
    :meth:`IntervalIndex.from_breaks`, and :meth:`IntervalIndex.from_tuples`.

    See further examples in the doc strings of ``interval_range`` and the
    mentioned constructor methods.
    )r?   ÚsummaryrB   Úextra_attributesÚextra_methodsÚexamplesÚ
set_closedÚ	to_tuplesT)Úwrap)Ú	__array__ÚoverlapsÚcontainsÚclosed_leftÚclosed_rightÚ	open_leftÚ
open_rightÚis_emptyÚis_non_overlapping_monotonicÚclosed)Úcachec                      sÀ  e Zd ZU dZded< ded< ded< ded< ded< ded	< d
ed< d
ed< dZeZ					dˆd‰dd„Ze	e
ed d e d¡e d¡dœ ƒ				dŠd‹d d!„ƒƒZe	e
ed" d e d¡e d#¡dœ ƒ				dŠdŒd$d%„ƒƒZe	e
ed& d e d¡e d'¡dœ ƒ				dŠdŒd(d)„ƒƒZed�d+d,„ƒZdŽd/d0„Zd�d3d4„Zed�d6d7„ƒZd8d9„ Zed‘d;d<„ƒZe
ejjƒd’d“d?d@„ƒZed”dAdB„ƒZed”dCdD„ƒZed”dEdF„ƒZd”dGdH„ZdIdJ„ Z d•d–dNdO„Z!d—dQdR„Z"			d˜d™d\d]„Z#e
e$d^ e% ƒdšd`da„ƒZ&d›dbdc„Z'dšddde„Z(ed”dfdg„ƒZ)dhZ*dœ‡ fdkdl„Z+ed”dmdn„ƒZ,d�dodp„Z-dždrds„Z.edŸdtdu„ƒZ/edŸdvdw„ƒZ0edŸdxdy„ƒZ1edŸdzd{„ƒZ2d|d}„ Z3d dd€„Z4d d�d‚„Z5d¡d„d…„Z6d†d‡„ Z7‡  Z8S )¢r>   Úintervalindexr;   ro   Úboolrn   ri   rj   rk   rl   r*   Ú_dataÚ_valuesFNTúIntervalClosedType | NonerD   úDtype | NoneÚcopyrB   úHashable | NoneÚverify_integrityÚreturnr<   c                 C  sV   t ||| ƒ}td| jƒ� t|||||d�}W d   ƒ n1 s w   Y  |  ||¡S )Nr*   )ro   rw   rD   ry   )r0   r   Ú__name__r*   Ú_simple_new)rZ   Údataro   rD   rw   rB   ry   ÚarrayrU   rU   rV   Ú__new__ß   s   	ûÿ	zIntervalIndex.__new__Úfrom_breakszY
             name : str, optional
                  Name of the resulting IntervalIndex.zÇ        Examples
        --------
        >>> pd.IntervalIndex.from_breaks([0, 1, 2, 3])
        IntervalIndex([(0, 1], (1, 2], (2, 3]],
                      dtype='interval[int64, right]')
        )r?   rB   rb   Úrightc                 C  óL   t d| jƒ� tj||||d�}W d   ƒ n1 sw   Y  | j||d�S ©Nr*   )ro   rw   rD   ©rB   )r   r{   r*   r€   r|   )rZ   Úbreaksro   rB   rw   rD   r~   rU   rU   rV   r€   õ   s   ÿÿzIntervalIndex.from_breaksrY   zÏ        Examples
        --------
        >>> pd.IntervalIndex.from_arrays([0, 1, 2], [1, 2, 3])
        IntervalIndex([(0, 1], (1, 2], (2, 3]],
                      dtype='interval[int64, right]')
        c                 C  sN   t d| jƒ� tj|||||d�}W d   ƒ n1 sw   Y  | j||d�S )Nr*   )rw   rD   r„   )r   r{   r*   rY   r|   )rZ   Úleftr�   ro   rB   rw   rD   r~   rU   rU   rV   rY     s   
ÿÿzIntervalIndex.from_arraysÚfrom_tupleszÄ        Examples
        --------
        >>> pd.IntervalIndex.from_tuples([(0, 1), (1, 2)])
        IntervalIndex([(0, 1], (1, 2]],
                       dtype='interval[int64, right]')
        c                 C  r‚   rƒ   )r   r{   r*   r‡   r|   )rZ   r}   ro   rB   rw   rD   ÚarrrU   rU   rV   r‡   <  s   ÿzIntervalIndex.from_tuplesr   c                 C  s8   |   | j¡}t|ƒ}|   | j¡}t|ƒ}t||| jd�S )N©ro   )Ú_maybe_convert_i8r†   r   r�   r   ro   )Úselfr†   r�   rU   rU   rV   Ú_engine`  s
   zIntervalIndex._engineÚkeyr   c                 C  sN   t |ƒ t|tƒst|| jƒr| jS dS z|  |¡ W dS  ty&   Y dS w )zÏ
        return a boolean if this key is IN the index
        We *only* accept an Interval

        Parameters
        ----------
        key : Interval

        Returns
        -------
        bool
        FT)ÚhashrK   r	   r&   rD   ÚhasnansÚget_locÚKeyError)r‹   r�   rU   rU   rV   Ú__contains__i  s   

ÿzIntervalIndex.__contains__ÚslobjÚslicec                 C  s   | j | }t| ƒj|| jd�S )zH
        Fastpath for __getitem__ when we know we have a slice.
        r„   )rs   rJ   r|   Ú_name)r‹   r“   ÚresrU   rU   rV   Ú_getitem_slice‚  s   
zIntervalIndex._getitem_slicer5   c                 C  s   t j| j| jgddgd�S )Nr†   r�   )Únames)r5   rY   r†   r�   ©r‹   rU   rU   rV   Ú_multiindex‰  s   zIntervalIndex._multiindexc                 C  s(   | j | j| j| jdœ}tt| ƒ|fd fS )N)r†   r�   ro   rB   )r†   r�   ro   rB   r\   rJ   )r‹   r[   rU   rU   rV   Ú
__reduce__�  s   üzIntervalIndex.__reduce__Ústrc                 C  s   dS )z4Return a string of the type inferred from the valuesÚintervalrU   r™   rU   rU   rV   Úinferred_type–  s   zIntervalIndex.inferred_typeÚdeepÚintc                 C  s   | j j|d�| jj|d� S )N)rŸ   )r†   Úmemory_usager�   )r‹   rŸ   rU   rU   rV   r¡   œ  s   zIntervalIndex.memory_usagec                 C  s   | ddd… j S )z€
        Return True if the IntervalIndex is monotonic decreasing (only equal or
        decreasing values), else False
        Néÿÿÿÿ)Úis_monotonic_increasingr™   rU   rU   rV   Úis_monotonic_decreasing¤  s   z%IntervalIndex.is_monotonic_decreasingc                 C  s‚   | j }| j}|  ¡  ¡ dkrdS |js|jrdS tƒ }t |jdd�¡d }|D ]}|| || f}||v r9 dS | 	|¡ q(dS )zX
        Return True if the IntervalIndex contains unique elements, else False.
        rF   FT)Úkeepr   )
r†   r�   ÚisnaÚsumÚ	is_uniqueÚsetrM   ÚwhereÚ
duplicatedÚadd)r‹   r†   r�   Ú
seen_pairsÚ	check_idxÚidxÚpairrU   rU   rV   r¨   ¬  s   zIntervalIndex.is_uniquec                 C  s   | j jS )a�  
        Return True if the IntervalIndex has overlapping intervals, else False.

        Two intervals overlap if they share a common point, including closed
        endpoints. Intervals that only have an open endpoint in common do not
        overlap.

        Returns
        -------
        bool
            Boolean indicating if the IntervalIndex has overlapping intervals.

        See Also
        --------
        Interval.overlaps : Check whether two Interval objects overlap.
        IntervalIndex.overlaps : Check an IntervalIndex elementwise for
            overlaps.

        Examples
        --------
        >>> index = pd.IntervalIndex.from_tuples([(0, 2), (1, 3), (4, 5)])
        >>> index
        IntervalIndex([(0, 2], (1, 3], (4, 5]],
              dtype='interval[int64, right]')
        >>> index.is_overlapping
        True

        Intervals that share closed endpoints overlap:

        >>> index = pd.interval_range(0, 3, closed='both')
        >>> index
        IntervalIndex([[0, 1], [1, 2], [2, 3]],
              dtype='interval[int64, both]')
        >>> index.is_overlapping
        True

        Intervals that only have an open endpoint in common do not overlap:

        >>> index = pd.interval_range(0, 3, closed='left')
        >>> index
        IntervalIndex([[0, 1), [1, 2), [2, 3)],
              dtype='interval[int64, left]')
        >>> index.is_overlapping
        False
        )rŒ   Úis_overlappingr™   rU   rU   rV   r±   Ä  s   0zIntervalIndex.is_overlappingc                 C  sB   t |ddƒ}t|tƒst|tƒr|  |j¡S tttt	f}t||ƒS )a	  
        Check if a given key needs i8 conversion. Conversion is necessary for
        Timestamp, Timedelta, DatetimeIndex, and TimedeltaIndex keys. An
        Interval-like requires conversion if its endpoints are one of the
        aforementioned types.

        Assumes that any list-like data has already been cast to an Index.

        Parameters
        ----------
        key : scalar or Index-like
            The key that should be checked for i8 conversion

        Returns
        -------
        bool
        rD   N)
rI   rK   r%   r	   Ú_needs_i8_conversionr†   r   r   r1   r6   )r‹   r�   Ú	key_dtypeÚi8_typesrU   rU   rV   r²   ö  s
   
z"IntervalIndex._needs_i8_conversionc           	      C  s$  t |ƒrt|ƒ}t|ƒ}|  |¡s|S t|ƒ}t|ddƒ}t|tƒs't|tƒrB|  	|j
¡}|  	|j¡}|r7tntj}|||| jd�S |rkt|ƒ\}}t|tƒrS|j}n+t|tƒr\|j}n"t|tjtjfƒrj| d¡}n|jt|jƒ}}|jr~| |j ¡}| jj}||kr�td|› d|› �ƒ‚|S )a  
        Maybe convert a given key to its equivalent i8 value(s). Used as a
        preprocessing step prior to IntervalTree queries (self._engine), which
        expects numeric data.

        Parameters
        ----------
        key : scalar or list-like
            The key that should maybe be converted to i8.

        Returns
        -------
        scalar or list-like
            The original key if no conversion occurred, int if converted scalar,
            Index with an int64 dtype if converted list-like.
        rD   Nr‰   Úi8z)Cannot index an IntervalIndex of subtype z with values of dtype ) r   r/   r   r²   r!   rI   rK   r%   r	   rŠ   r†   r�   r>   rY   ro   r   r   Úordinalr   Ú_valuerM   Ú
datetime64rN   ÚviewrD   r-   Úasi8r�   rª   Ú_isnanÚsubtypeÚ
ValueError)	r‹   r�   Úscalarr³   r†   r�   ÚconstructorÚkey_i8r¼   rU   rU   rV   rŠ     sD   
ÿ


€ÿÿzIntervalIndex._maybe_convert_i8r†   ÚsideúLiteral['left', 'right']c                 C  sz   | j stdƒ‚t|ttfƒrtdƒ‚|dkr| jjs"|dkr-| jjs-| j}| j	r,t
|ƒ}n
| j}| jr7t|ƒ}| ||¡S )Nzrcan only get slices from an IntervalIndex if bounds are non-overlapping and all monotonic increasing or decreasingz,Interval objects are not currently supportedr†   r�   )rn   r‘   rK   r
   r>   ÚNotImplementedErrorr†   r£   r�   rl   rW   rk   rX   Ú_searchsorted_monotonic)r‹   rT   rÁ   Úsub_idxrU   rU   rV   rÄ   Q  s    ÿ€z%IntervalIndex._searchsorted_monotonicúint | slice | np.ndarrayc              
   C  s  |   |¡ t|tƒr!| j|jkrt|ƒ‚| j|jk| j|jk@ }n9t|| jƒr,|  	¡ }n.| j
r1tnt}| jr8tnt}z|| j|ƒ||| jƒ@ }W n tyY } zt|ƒ|‚d}~ww | ¡ }|dkrft|ƒ‚|dkrn| ¡ S t | d¡¡}t|tƒrŠ|jdu rŠt|jt| ƒ|jƒ}|S )a£  
        Get integer location, slice or boolean mask for requested label.

        Parameters
        ----------
        key : label

        Returns
        -------
        int if unique index, slice if monotonic index, else mask

        Examples
        --------
        >>> i1, i2 = pd.Interval(0, 1), pd.Interval(1, 2)
        >>> index = pd.IntervalIndex([i1, i2])
        >>> index.get_loc(1)
        0

        You can also supply a point inside an interval.

        >>> index.get_loc(1.5)
        1

        If a label is in several intervals, you get the locations of all the
        relevant intervals.

        >>> i3 = pd.Interval(0, 2)
        >>> overlapping_index = pd.IntervalIndex([i1, i2, i3])
        >>> overlapping_index.get_loc(0.5)
        array([ True, False,  True])

        Only exact matches will be returned if an interval is provided.

        >>> index.get_loc(pd.Interval(0, 1))
        0
        Nr   rF   Úu1)Ú_check_indexing_errorrK   r	   ro   r‘   r†   r�   r&   rD   r¦   ri   r   r   rj   rQ   r§   Úargmaxr   Úmaybe_booleans_to_slicer¹   r”   ÚstopÚstartÚlenÚstep)r‹   r�   ÚmaskÚop_leftÚop_rightÚerrÚmatchesr–   rU   rU   rV   r�   n  s0   
%


€þzIntervalIndex.get_locÚtargetr-   Úmethodú
str | NoneÚlimitú
int | NoneÚ	toleranceú
Any | Noneúnpt.NDArray[np.intp]c                 C  s^   t |tƒr|  |¡}t|ƒS t|jƒs(t|jƒs(|  |¡}| j |j	¡}t|ƒS |  
|¡d S )Nr   )rK   r>   Ú_get_indexer_unique_sidesr    rD   r"   rŠ   rŒ   Úget_indexerÚvaluesÚ_get_indexer_pointwiser   )r‹   rÔ   rÕ   r×   rÙ   ÚindexerrU   rU   rV   Ú_get_indexer±  s   

ö
þzIntervalIndex._get_indexerÚget_indexer_non_uniqueú1tuple[npt.NDArray[np.intp], npt.NDArray[np.intp]]c                 C  s¶   t |ƒ}|  |¡s|  |¡s| j|d dd�S t|tƒr6| jjr1| jjr1|  	|¡}|dk 
¡ d }n"|  |¡S t|jƒs@|  |¡sE|  |¡S |  |¡}| j |j¡\}}t|ƒt|ƒfS )NFr'   r¢   r   )r/   Ú_should_compareÚ_should_partial_indexÚ_get_indexer_non_comparablerK   r>   r†   r¨   r�   rÜ   Únonzerorß   r    rD   rŠ   rŒ   râ   rÞ   r   )r‹   rÔ   rà   ÚmissingrU   rU   rV   râ   Ë  s   




z$IntervalIndex.get_indexer_non_uniquec                 C  s2   | j  |j ¡}| j |j¡}t ||k|d¡}|S )zZ
        _get_indexer specialized to the case where both of our sides are unique.
        r¢   )r†   rÝ   r�   rM   rª   )r‹   rÔ   Úleft_indexerÚright_indexerrà   rU   rU   rV   rÜ   ê  s   z'IntervalIndex._get_indexer_unique_sidesc              	   C  sê   g g }}t |ƒD ]^\}}z-|  |¡}t|tƒr%tj|j|j|jdd�}nt	 
|¡r2tj|dd�}nt |¡d }W n' tyN   | |¡ t dg¡}Y n tya   | |¡ t dg¡}Y nw | |¡ q	t |¡}t|ƒt|ƒfS )zV
        pointwise implementation for get_indexer and get_indexer_non_unique.
        Úintp©rD   rF   )Úndminr   r¢   )Ú	enumerater�   rK   r”   rM   ÚarangerÌ   rË   rÎ   r   r   r~   rª   r‘   Úappendr   Úconcatenater   )r‹   rÔ   rà   rè   Úir�   ÚlocsrU   rU   rV   rß   ö  s(   



€

ü
z$IntervalIndex._get_indexer_pointwisec                 C  s   | j  o	| jjdk S )Né   )r±   rŒ   Ú	_na_countr™   rU   rU   rV   Ú_index_as_unique  s   zIntervalIndex._index_as_uniquezKcannot handle overlapping indices; use IntervalIndex.get_indexer_non_uniqueÚkindúLiteral['loc', 'getitem']c                   sN   |j d u s |j dks d}|dkrt|ƒ‚|dkr t|ƒs t|ƒ‚tƒ  ||¡S )NrF   zClabel-based slicing with step!=1 is not supported for IntervalIndexÚlocÚgetitem)rÎ   r½   r,   ÚsuperÚ_convert_slice_indexer)r‹   r�   r÷   Úmsg©Ú	__class__rU   rV   rü     s   z$IntervalIndex._convert_slice_indexerc                 C  s   | j jjdv S )NrE   )rD   r¼   r÷   r™   rU   rU   rV   Ú_should_fallback_to_positional,  s   z,IntervalIndex._should_fallback_to_positionalc                 C  s   t | |ƒ ||¡S ©N)rI   Ú_maybe_cast_slice_bound)r‹   rT   rÁ   rU   rU   rV   r  4  s   z%IntervalIndex._maybe_cast_slice_boundr:   c                 C  s&   t |tƒsdS t| j|gƒ}t|ƒ S )NF)rK   r%   r   rD   r    )r‹   rD   Úcommon_subtyperU   rU   rV   Ú_is_comparable_dtype7  s   

z"IntervalIndex._is_comparable_dtypec                 C  ó   t | jjdd�S ©NF)rw   )r-   rs   r†   r™   rU   rU   rV   r†   ?  ó   zIntervalIndex.leftc                 C  r  r  )r-   rs   r�   r™   rU   rU   rV   r�   C  r  zIntervalIndex.rightc                 C  r  r  )r-   rs   Úmidr™   rU   rU   rV   r  G  r  zIntervalIndex.midc                 C  r  r  )r-   rs   Úlengthr™   rU   rU   rV   r	  K  r  zIntervalIndex.lengthc                 C  sf   | j jr| jjr|  |¡}n|j jr$|jjr$|  ¡  ¡ dkr$| | ¡}n|  |¡}|du r1| ¡ }|S )zL
        intersection specialized to the case with matching dtypes.
        rF   N)r†   r¨   r�   Ú_intersection_uniquer¦   r§   Ú_intersection_non_uniqueÚsort_values)r‹   ÚotherÚsortÚtakenrU   rU   rV   Ú_intersectionR  s    
zIntervalIndex._intersectionr  c                 C  sP   | j  |j ¡}| j |j¡}||k|dk@ }| | ¡ d ¡}t|ƒ}|  |¡S )a'  
        Used when the IntervalIndex does not have any common endpoint,
        no matter left or right.
        Return the intersection with another IntervalIndex.
        Parameters
        ----------
        other : IntervalIndex
        Returns
        -------
        IntervalIndex
        r¢   r   )r†   rÝ   r�   Útakerç   r(   )r‹   r  ÚlindexerÚrindexerÚmatchrà   rU   rU   rV   r
  f  s   
z"IntervalIndex._intersection_uniquec                 C  s†   t jt| ƒtd�}| jr |jr t  t| ƒ¡|  ¡  d }d||< tt|j	|j
ƒƒ}tt| j	| j
ƒƒD ]\}}||v r>d||< q2| | S )a  
        Used when the IntervalIndex does have some common endpoints,
        on either sides.
        Return the intersection with another IntervalIndex.

        Parameters
        ----------
        other : IntervalIndex

        Returns
        -------
        IntervalIndex
        rì   r   T)rM   ÚzerosrÍ   rr   r�   rï   r¦   r©   Úzipr†   r�   rî   )r‹   r  rÏ   Úfirst_nan_locÚ
other_tupsrò   ÚtuprU   rU   rV   r  |  s   €z&IntervalIndex._intersection_non_uniqueú
np.ndarrayc                 C  ó   t dƒ‚)NzRIntervalIndex does not use libjoin fastpaths or pass values to IndexEngine objects©rÃ   r™   rU   rU   rV   Ú_get_engine_target›  s   ÿz IntervalIndex._get_engine_targetc                 C  r  )Nz,IntervalIndex does not use libjoin fastpathsr  )r‹   ÚresultrU   rU   rV   Ú_from_join_target¤  s   zIntervalIndex._from_join_target)NNFNT)ro   ru   rD   rv   rw   rr   rB   rx   ry   rr   rz   r<   )r�   NFN)
ro   ru   rB   rx   rw   rr   rD   rv   rz   r>   )
ro   r;   rB   rx   rw   rr   rD   rv   rz   r>   )rz   r   )r�   r   rz   rr   )r“   r”   rz   r>   )rz   r5   )rz   rœ   )F)rŸ   rr   rz   r    ©rz   rr   )r†   )rÁ   rÂ   )rz   rÆ   )NNN)
rÔ   r-   rÕ   rÖ   r×   rØ   rÙ   rÚ   rz   rÛ   )rÔ   r-   rz   rã   )rÔ   r>   rz   rÛ   )r�   r”   r÷   rø   )rÁ   rœ   )rD   r:   rz   rr   )rz   r-   )r  r>   rz   r>   )rz   r  )9r{   Ú
__module__Ú__qualname__Ú_typÚ__annotations__Ú_can_hold_stringsr*   Ú	_data_clsr   Úclassmethodr   r+   ÚtextwrapÚdedentr€   rY   r‡   r   rŒ   r’   r—   rš   r›   Úpropertyrž   r-   r¡   Ú__doc__r¤   r¨   r±   r²   rŠ   rÄ   r�   rá   r.   Ú_index_doc_kwargsrâ   rÜ   rß   rö   Ú_requires_unique_msgrü   r   r  r  r†   r�   r  r	  r  r
  r  r  r  Ú__classcell__rU   rU   rþ   rV   r>       sö   
 -ùÿÿùÿÿúÿÿùÿÿùÿÿùÿÿú

	

1B
Fû

 ÿ




	rz   rr   c                 C  s$   t t| ƒt| tƒt| tƒ| du gƒS )zJ
    Helper for interval_range to check if start/end are valid types.
    N)Úanyr   rK   r   r   )ÚendpointrU   rU   rV   Ú_is_valid_endpointª  s   üÿr1  c                 C  sL   dd„ }dd„ }t | ƒrt |ƒp%|| ƒo||ƒp%|| ƒo||ƒp%t | |¡S )zK
    Helper for interval_range to check type compat of start/end/freq.
    c                 S  ó   t | ttfƒS r  )rK   r   r   ©ÚxrU   rU   rV   Ú<lambda>¼  ó    z%_is_type_compatible.<locals>.<lambda>c                 S  r2  r  )rK   r   r   r3  rU   rU   rV   r5  ½  r6  )r   ÚcomÚany_none)ÚaÚbÚis_ts_compatÚis_td_compatrU   rU   rV   Ú_is_type_compatible¸  s   ÿþ
ür=  r�   rx   r;   c           	   
   C  s
  t | ƒ} t |ƒ}| dur| n|}|du r#t || |¡r#t|ƒr!dnd}t | |||¡dkr1tdƒ‚t| ƒs<td| › �ƒ‚t|ƒsGtd|› �ƒ‚t|ƒ}|durnt|ƒsnzt|ƒ}W n tym } ztd|› �ƒ|‚d}~ww t	t
| |ƒt
| |ƒt
||ƒgƒs‚td	ƒ‚|durŠ|d7 }t|ƒrçt | ||¡r¡t | ||d
  |¡}n/|du r°t||  | ƒd }n| du r½||d |  } n|du rÉ| |d |  }t | ||¡}t	dd„ t | ||¡D ƒƒræt|t d¡ƒ}nt|tƒrõt| |||d�}nt| |||d�}tj|||d�S )a  
    Return a fixed frequency IntervalIndex.

    Parameters
    ----------
    start : numeric or datetime-like, default None
        Left bound for generating intervals.
    end : numeric or datetime-like, default None
        Right bound for generating intervals.
    periods : int, default None
        Number of periods to generate.
    freq : numeric, str, Timedelta, datetime.timedelta, or DateOffset, default None
        The length of each interval. Must be consistent with the type of start
        and end, e.g. 2 for numeric, or '5H' for datetime-like.  Default is 1
        for numeric and 'D' for datetime-like.
    name : str, default None
        Name of the resulting IntervalIndex.
    closed : {'left', 'right', 'both', 'neither'}, default 'right'
        Whether the intervals are closed on the left-side, right-side, both
        or neither.

    Returns
    -------
    IntervalIndex

    See Also
    --------
    IntervalIndex : An Index of intervals that are all closed on the same side.

    Notes
    -----
    Of the four parameters ``start``, ``end``, ``periods``, and ``freq``,
    exactly three must be specified. If ``freq`` is omitted, the resulting
    ``IntervalIndex`` will have ``periods`` linearly spaced elements between
    ``start`` and ``end``, inclusively.

    To learn more about datetime-like frequency strings, please see `this link
    <https://pandas.pydata.org/pandas-docs/stable/user_guide/timeseries.html#offset-aliases>`__.

    Examples
    --------
    Numeric ``start`` and  ``end`` is supported.

    >>> pd.interval_range(start=0, end=5)
    IntervalIndex([(0, 1], (1, 2], (2, 3], (3, 4], (4, 5]],
                  dtype='interval[int64, right]')

    Additionally, datetime-like input is also supported.

    >>> pd.interval_range(start=pd.Timestamp('2017-01-01'),
    ...                   end=pd.Timestamp('2017-01-04'))
    IntervalIndex([(2017-01-01 00:00:00, 2017-01-02 00:00:00],
                   (2017-01-02 00:00:00, 2017-01-03 00:00:00],
                   (2017-01-03 00:00:00, 2017-01-04 00:00:00]],
                  dtype='interval[datetime64[ns], right]')

    The ``freq`` parameter specifies the frequency between the left and right.
    endpoints of the individual intervals within the ``IntervalIndex``.  For
    numeric ``start`` and ``end``, the frequency must also be numeric.

    >>> pd.interval_range(start=0, periods=4, freq=1.5)
    IntervalIndex([(0.0, 1.5], (1.5, 3.0], (3.0, 4.5], (4.5, 6.0]],
                  dtype='interval[float64, right]')

    Similarly, for datetime-like ``start`` and ``end``, the frequency must be
    convertible to a DateOffset.

    >>> pd.interval_range(start=pd.Timestamp('2017-01-01'),
    ...                   periods=3, freq='MS')
    IntervalIndex([(2017-01-01 00:00:00, 2017-02-01 00:00:00],
                   (2017-02-01 00:00:00, 2017-03-01 00:00:00],
                   (2017-03-01 00:00:00, 2017-04-01 00:00:00]],
                  dtype='interval[datetime64[ns], right]')

    Specify ``start``, ``end``, and ``periods``; the frequency is generated
    automatically (linearly spaced).

    >>> pd.interval_range(start=0, end=6, periods=4)
    IntervalIndex([(0.0, 1.5], (1.5, 3.0], (3.0, 4.5], (4.5, 6.0]],
              dtype='interval[float64, right]')

    The ``closed`` parameter specifies which endpoints of the individual
    intervals within the ``IntervalIndex`` are closed.

    >>> pd.interval_range(end=5, periods=4, closed='both')
    IntervalIndex([[1, 2], [2, 3], [3, 4], [4, 5]],
                  dtype='interval[int64, both]')
    NrF   ÚDé   zVOf the four parameters: start, end, periods, and freq, exactly three must be specifiedz,start must be numeric or datetime-like, got z*end must be numeric or datetime-like, got z7freq must be numeric or convertible to DateOffset, got z+start, end, freq need to be type compatiblegš™™™™™¹?c                 s  s   � | ]}t |ƒV  qd S r  )r   )Ú.0r4  rU   rU   rV   Ú	<genexpr>`  s   € z!interval_range.<locals>.<genexpr>Úint64)rÌ   ÚendÚperiodsÚfreq)rB   ro   )r   r7  r8  r   Úcount_not_noner½   r1  r)   r   Úallr=  rQ   Úall_not_nonerM   rï   r    ÚlinspaceÚnot_noner   rD   rK   r   r2   r7   r>   r€   )	rÌ   rC  rD  rE  rB   ro   r0  rÒ   r…   rU   rU   rV   Úinterval_rangeÆ  sl   `ÿÿþ€ÿýÿþ€
rK  r   )NNNNNr�   )rB   rx   ro   r;   rz   r>   )ir+  Ú
__future__r   Úoperatorr   r   r(  Útypingr   r   r   ÚnumpyrM   Úpandas._libsr   Úpandas._libs.intervalr	   r
   r   Úpandas._libs.tslibsr   r   r   r   r   Úpandas.errorsr   Úpandas.util._decoratorsr   r   Úpandas.util._exceptionsr   Úpandas.core.dtypes.castr   r   r   r   r   Úpandas.core.dtypes.commonr   r   r   r   r   r   r    r!   r"   r#   Úpandas.core.dtypes.dtypesr$   r%   Úpandas.core.dtypes.missingr&   Úpandas.core.algorithmsr(   Úpandas.core.arrays.datetimeliker)   Úpandas.core.arrays.intervalr*   r+   Úpandas.core.commonÚcoreÚcommonr7  Úpandas.core.indexersr,   Úpandas.core.indexes.baseÚindexesÚbaseÚibaser-   r.   r/   r0   Úpandas.core.indexes.datetimesr1   r2   Úpandas.core.indexes.extensionr3   r4   Úpandas.core.indexes.multir5   Úpandas.core.indexes.timedeltasr6   r7   Úcollections.abcr8   Úpandas._typingr9   r:   r;   r<   r=   Údictr,  Úupdater)  rW   rX   r\   r>   r1  r=  rK  rU   rU   rU   rV   Ú<module>   sš    0
ÿüÿÿúÿÿ
õ     
c
ú