§
    Ð! h‘9  ã                  óT  — d Z 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 ddlmZ dd	lmZ dd
lmZ dZ G d„ d¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        Z G d„ de¦  «        ZdS )zLIndexer objects for computing start/end window bounds for rolling operationsé    )Úannotations)Ú	timedeltaN)Ú
BaseOffset)Ú calculate_variable_window_bounds)ÚAppender)Úensure_platform_int)ÚDatetimeIndex)ÚNanoa¾  
Computes the bounds of a window.

Parameters
----------
num_values : int, default 0
    number of values that will be aggregated over
window_size : int, default 0
    the number of rows in a window
min_periods : int, default None
    min_periods passed from the top level rolling API
center : bool, default None
    center passed from the top level rolling API
closed : str, default None
    closed passed from the top level rolling API
step : int, default None
    step passed from the top level rolling API
    .. versionadded:: 1.5
win_type : str, default None
    win_type passed from the top level rolling API

Returns
-------
A tuple of ndarray[int64]s, indicating the boundaries of each
window
c                  óT   — e Zd ZdZ	 ddd
„Z ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚBaseIndexeraà  
    Base class for window bounds calculations.

    Examples
    --------
    >>> from pandas.api.indexers import BaseIndexer
    >>> class CustomIndexer(BaseIndexer):
    ...     def get_window_bounds(self, num_values, min_periods, center, closed, step):
    ...         start = np.empty(num_values, dtype=np.int64)
    ...         end = np.empty(num_values, dtype=np.int64)
    ...         for i in range(num_values):
    ...             start[i] = i
    ...             end[i] = i + self.window_size
    ...         return start, end
    >>> df = pd.DataFrame({"values": range(5)})
    >>> indexer = CustomIndexer(window_size=2)
    >>> df.rolling(indexer).sum()
        values
    0	1.0
    1	3.0
    2	5.0
    3	7.0
    4	4.0
    Nr   Úindex_arrayúnp.ndarray | NoneÚwindow_sizeÚintÚreturnÚNonec                óx   — || _         || _        |                     ¦   «         D ]\  }}t          | ||¦  «         Œd S ©N)r   r   ÚitemsÚsetattr)Úselfr   r   ÚkwargsÚkeyÚvalues         úVc:\xampp_lite_8_4\www\timesheet\venv\Lib\site-packages\pandas/core/indexers/objects.pyÚ__init__zBaseIndexer.__init__H   sN   € ð 'ˆÔØ&ˆÔà Ÿ,š,™.œ.ð 	&ð 	&‰JˆC�Ý�D˜#˜uÑ%Ô%Ð%Ð%ð	&ð 	&ó    Ú
num_valuesÚmin_periodsú
int | NoneÚcenterúbool | NoneÚclosedú
str | NoneÚstepútuple[np.ndarray, np.ndarray]c                ó   — t           ‚r   )ÚNotImplementedError©r   r   r   r!   r#   r%   s         r   Úget_window_boundszBaseIndexer.get_window_boundsQ   s
   € õ "Ð!r   )Nr   )r   r   r   r   r   r   ©r   NNNN©r   r   r   r    r!   r"   r#   r$   r%   r    r   r&   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   Úget_window_bounds_docr*   © r   r   r   r   .   s€   € € € € € ðð ð4 IJð&ð &ð &ð &ð &ð €XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øð"ð "ð "ð "ñ %Ô$ð"ð "ð "r   r   c                  óH   — e Zd ZdZ ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚFixedWindowIndexerz3Creates window boundaries that are of fixed length.r   Nr   r   r   r    r!   r"   r#   r$   r%   r   r&   c                ó  — |s| j         dk    r| j         dz
  dz  }nd}t          j        d|z   |dz   |z   |d¬¦  «        }|| j         z
  }|dv r|dz  }|dv r|dz  }t          j        |d|¦  «        }t          j        |d|¦  «        }||fS )Nr   é   é   Úint64©Údtype©ÚleftÚboth)r<   Úneither)r   ÚnpÚarangeÚclip)	r   r   r   r!   r#   r%   ÚoffsetÚendÚstarts	            r   r*   z$FixedWindowIndexer.get_window_bounds`   s¾   € ð ð 	�TÔ%¨Ò*Ð*ØÔ&¨Ñ*¨qÑ0ˆFˆFàˆFåŒi˜˜F™
 J°¡N°VÑ$;¸TÈÐQÑQÔQˆØ�dÔ&Ñ&ˆØÐ%Ð%Ð%Ø�Q‰JˆEØÐ(Ð(Ð(Ø�1‰HˆCåŒg�c˜1˜jÑ)Ô)ˆÝ”˜˜q *Ñ-Ô-ˆà�cˆzÐr   r+   r,   ©r-   r.   r/   r0   r   r1   r*   r2   r   r   r4   r4   ]   sZ   € € € € € Ø=Ð=à€XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øðð ð ð ñ %Ô$ðð ð r   r4   c                  óH   — e Zd ZdZ ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚVariableWindowIndexerzNCreates window boundaries that are of variable length, namely for time series.r   Nr   r   r   r    r!   r"   r#   r$   r%   r   r&   c                ó>   — t          || j        |||| j        ¦  «        S r   )r   r   r   r)   s         r   r*   z'VariableWindowIndexer.get_window_bounds~   s.   € õ 0ØØÔØØØØÔñ
ô 
ð 	
r   r+   r,   rE   r2   r   r   rG   rG   {   sZ   € € € € € ØXÐXà€XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øð
ð 
ð 
ð 
ñ %Ô$ð
ð 
ð 
r   rG   c                  ód   ‡ — e Zd ZdZ	 	 	 	 ddˆ fd„Z ee¦  «        	 	 	 	 	 ddd„¦   «         Zˆ xZS )ÚVariableOffsetWindowIndexeraP  
    Calculate window boundaries based on a non-fixed offset such as a BusinessDay.

    Examples
    --------
    >>> from pandas.api.indexers import VariableOffsetWindowIndexer
    >>> df = pd.DataFrame(range(10), index=pd.date_range("2020", periods=10))
    >>> offset = pd.offsets.BDay(1)
    >>> indexer = VariableOffsetWindowIndexer(index=df.index, offset=offset)
    >>> df
                0
    2020-01-01  0
    2020-01-02  1
    2020-01-03  2
    2020-01-04  3
    2020-01-05  4
    2020-01-06  5
    2020-01-07  6
    2020-01-08  7
    2020-01-09  8
    2020-01-10  9
    >>> df.rolling(indexer).sum()
                   0
    2020-01-01   0.0
    2020-01-02   1.0
    2020-01-03   2.0
    2020-01-04   3.0
    2020-01-05   7.0
    2020-01-06  12.0
    2020-01-07   6.0
    2020-01-08   7.0
    2020-01-09   8.0
    2020-01-10   9.0
    Nr   r   r   r   r   ÚindexúDatetimeIndex | NonerB   úBaseOffset | Noner   r   c                óê   •—  t          ¦   «         j        ||fi |¤Ž t          |t          ¦  «        st	          d¦  «        ‚|| _        t          |t          ¦  «        st	          d¦  «        ‚|| _        d S )Nzindex must be a DatetimeIndex.z(offset must be a DateOffset-like object.)Úsuperr   Ú
isinstancer	   Ú
ValueErrorrK   r   rB   )r   r   r   rK   rB   r   Ú	__class__s         €r   r   z$VariableOffsetWindowIndexer.__init__¹   s{   ø€ ð 	�‰ŒÔ˜ kÐ<Ð<°VÐ<Ð<Ð<Ý˜%¥Ñ/Ô/ð 	?ÝÐ=Ñ>Ô>Ð>ØˆŒ
Ý˜&¥*Ñ-Ô-ð 	IÝÐGÑHÔHÐHØˆŒˆˆr   r   r   r    r!   r"   r#   r$   r%   r&   c                óÂ  — |�t          d¦  «        ‚|dk    r,t          j        dd¬¦  «        t          j        dd¬¦  «        fS |€| j        �dnd}|dv }|dv }| j        |d	z
           | j        d         k     rd
}nd	}|| j        z  }	t          j        |d¬¦  «        }
|
                     d
¦  «         t          j        |d¬¦  «        }|                     d
¦  «         d|
d<   |rd	|d<   nd|d<   t          d¦  «        }t          d	|¦  «        D ]Î}| j        |         }||	z
  }|r|t          d	¦  «        z  }||
|<   t          |
|d	z
           |¦  «        D ]"}| j        |         |z
  |z  }||k    r||
|<    nŒ#| j        ||d	z
                    |z
  |z  }||k    r|s||d	z
           d	z   ||<   n||k    r	|d	z   ||<   n||d	z
           ||<   |s||xx         d	z  cc<   ŒÏ|
|fS )Nz/step not implemented for variable offset windowr   r8   r9   Úrightr=   )rT   r=   r;   r6   éÿÿÿÿ)	r(   r?   ÚemptyrK   rB   Úfillr   Úranger
   )r   r   r   r!   r#   r%   Úright_closedÚleft_closedÚindex_growth_signÚoffset_diffrD   rC   ÚzeroÚiÚ	end_boundÚstart_boundÚjÚ
start_diffÚend_diffs                      r   r*   z-VariableOffsetWindowIndexer.get_window_boundsÉ   sp  € ð ÐÝ%Ð&WÑXÔXÐXØ˜Š?ˆ?Ý”8˜A WÐ-Ñ-Ô-­r¬x¸ÀÐ/IÑ/IÔ/IÐIÐIð ˆ>Ø $¤
Ð 6�W�W¸FˆFàÐ!2Ð2ˆØÐ 0Ð0ˆàŒ:�j 1‘nÔ%¨¬
°1¬Ò5Ð5Ø "ÐÐà !ÐØ'¨$¬+Ñ5ˆå”˜¨7Ð3Ñ3Ô3ˆØ�
Š
�2‰ŒˆÝŒh�z¨Ð1Ñ1Ô1ˆØ�Š�‰Œˆàˆˆa‰ð ð 	ØˆC�‰FˆFð ˆC�‰Få˜‰|Œ|ˆõ �q˜*Ñ%Ô%ð 	ð 	ˆAØœ
 1œˆIØ# kÑ1ˆKð ð 'Ø�t A™wœwÑ&�ð ˆE�!‰HÝ˜5  Q¡œ<¨Ñ+Ô+ð ð �Ø"œj¨œm¨kÑ9Ð=NÑN�
Ø Ò$Ð$Ø �E˜!‘HØ�Eð %ð œ
 3 q¨1¡u¤:Ô.°Ñ:Ð>OÑOˆHØ˜4ÒÐ¨ÐØ˜Q ™Uœ a™��A‘�Ø˜TÒ!Ð!Ø˜Q™��A‘�à˜Q ™Uœ��A‘ð  ð Ø�A��”˜!‘��‘øà�cˆzÐr   )Nr   NN)
r   r   r   r   rK   rL   rB   rM   r   r   r+   r,   )	r-   r.   r/   r0   r   r   r1   r*   Ú__classcell__©rR   s   @r   rJ   rJ   •   s¦   ø€ € € € € ð!ð !ðJ *.ØØ&*Ø$(ðð ð ð ð ð ð ð  €XÐ#Ñ$Ô$ð Ø"&Ø"Ø!ØðJð Jð Jð Jñ %Ô$ðJð Jð Jð Jð Jr   rJ   c                  óH   — e Zd ZdZ ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚExpandingIndexerz;Calculate expanding window bounds, mimicking df.expanding()r   Nr   r   r   r    r!   r"   r#   r$   r%   r   r&   c                óŠ   — t          j        |t           j        ¬¦  «        t          j        d|dz   t           j        ¬¦  «        fS )Nr9   r6   )r?   Úzerosr8   r@   r)   s         r   r*   z"ExpandingIndexer.get_window_bounds  s<   € õ ŒH�Z¥r¤xÐ0Ñ0Ô0ÝŒI�a˜ a™­r¬xÐ8Ñ8Ô8ð
ð 	
r   r+   r,   rE   r2   r   r   rg   rg     sZ   € € € € € ØEÐEà€XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øð
ð 
ð 
ð 
ñ %Ô$ð
ð 
ð 
r   rg   c                  óH   — e Zd ZdZ ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚFixedForwardWindowIndexera¿  
    Creates window boundaries for fixed-length windows that include the current row.

    Examples
    --------
    >>> df = pd.DataFrame({'B': [0, 1, 2, np.nan, 4]})
    >>> df
         B
    0  0.0
    1  1.0
    2  2.0
    3  NaN
    4  4.0

    >>> indexer = pd.api.indexers.FixedForwardWindowIndexer(window_size=2)
    >>> df.rolling(window=indexer, min_periods=1).sum()
         B
    0  1.0
    1  3.0
    2  2.0
    3  4.0
    4  4.0
    r   Nr   r   r   r    r!   r"   r#   r$   r%   r   r&   c                óÔ   — |rt          d¦  «        ‚|�t          d¦  «        ‚|€d}t          j        d||d¬¦  «        }|| j        z   }| j        rt          j        |d|¦  «        }||fS )Nz.Forward-looking windows can't have center=TruezAForward-looking windows don't support setting the closed argumentr6   r   r8   r9   )rQ   r?   r@   r   rA   )r   r   r   r!   r#   r%   rD   rC   s           r   r*   z+FixedForwardWindowIndexer.get_window_boundsB  s�   € ð ð 	OÝÐMÑNÔNÐNØÐÝØSñô ð ð ˆ<ØˆDå”	˜!˜Z¨°WÐ=Ñ=Ô=ˆØ�dÔ&Ñ&ˆØÔð 	.Ý”'˜#˜q *Ñ-Ô-ˆCà�cˆzÐr   r+   r,   rE   r2   r   r   rk   rk   )  s`   € € € € € ðð ð0 €XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øðð ð ð ñ %Ô$ðð ð r   rk   c                  óf   ‡ — e Zd ZdZdddedfdˆ fd„Z ee¦  «        	 	 	 	 	 ddd„¦   «         Zˆ xZ	S )ÚGroupbyIndexerzMCalculate bounds to compute groupby rolling, mimicking df.groupby().rolling()Nr   r   r   r   úint | BaseIndexerÚgroupby_indicesúdict | NoneÚwindow_indexerútype[BaseIndexer]Úindexer_kwargsr   r   c                óÌ   •— |pi | _         || _        |r|                     ¦   «         ni | _         t	          ¦   «         j        d|| j                             d|¦  «        dœ|¤Ž dS )a4  
        Parameters
        ----------
        index_array : np.ndarray or None
            np.ndarray of the index of the original object that we are performing
            a chained groupby operation over. This index has been pre-sorted relative to
            the groups
        window_size : int or BaseIndexer
            window size during the windowing operation
        groupby_indices : dict or None
            dict of {group label: [positional index of rows belonging to the group]}
        window_indexer : BaseIndexer
            BaseIndexer class determining the start and end bounds of each group
        indexer_kwargs : dict or None
            Custom kwargs to be passed to window_indexer
        **kwargs :
            keyword arguments that will be available when get_window_bounds is called
        r   ©r   r   Nr2   )rp   rr   Úcopyrt   rO   r   Úpop)r   r   r   rp   rr   rt   r   rR   s          €r   r   zGroupbyIndexer.__init___  sŠ   ø€ ð6  /Ð4°"ˆÔØ,ˆÔØ7EÐM˜n×1Ò1Ñ3Ô3Ð3È2ˆÔØ�‰ŒÔð 	
Ø#ØÔ+×/Ò/°¸{ÑKÔKð	
ð 	
ð ð	
ð 	
ð 	
ð 	
ð 	
r   r   r   r   r    r!   r"   r#   r$   r%   r&   c                óÎ  — g }g }d}| j                              ¦   «         D �]Æ\  }	}
| j        �(| j                             t	          |
¦  «        ¦  «        }n| j        } | j        d	|| j        dœ| j        ¤Ž}|                     t          |
¦  «        ||||¦  «        \  }}| 
                    t          j        ¦  «        }| 
                    t          j        ¦  «        }t          |¦  «        t          |¦  «        k    s
J d¦   «         ‚t          j        ||t          |
¦  «        z   ¦  «        }|t          |
¦  «        z  }t          j        ||d         dz   g¦  «         
                    t          j        d¬¦  «        }|                     |                     t	          |¦  «        ¦  «        ¦  «         |                     |                     t	          |¦  «        ¦  «        ¦  «         �ŒÈt          |¦  «        dk    r@t          j        g t          j        ¬¦  «        t          j        g t          j        ¬¦  «        fS t          j        |¦  «        }t          j        |¦  «        }||fS )
Nr   rv   z6these should be equal in length from get_window_boundsrU   r6   F)rw   r9   r2   )rp   r   r   Útaker   rr   r   rt   r*   ÚlenÚastyper?   r8   r@   ÚappendÚarrayÚconcatenate)r   r   r   r!   r#   r%   Ústart_arraysÚ
end_arraysÚwindow_indices_startr   Úindicesr   ÚindexerrD   rC   Úwindow_indicess                   r   r*   z GroupbyIndexer.get_window_boundsƒ  sb  € ð ˆØˆ
Ø ÐØ Ô0×6Ò6Ñ8Ô8ð 	Mñ 	M‰LˆC�ð ÔÐ+Ø"Ô.×3Ò3Õ4GÈÑ4PÔ4PÑQÔQ��à"Ô.�Ø)�dÔ)ð Ø'Ø Ô,ðð ð Ô%ðð ˆGð
 !×2Ò2Ý�G‘”˜k¨6°6¸4ñô ‰JˆE�3ð —L’L¥¤Ñ*Ô*ˆEØ—*’*�RœXÑ&Ô&ˆCÝ�u‘:”:¥Øñ"ô "ò ð ð àGñô ð õ
  œYØ$Ð&:½SÀ¹\¼\Ñ&Iñô ˆNð !¥C¨¡L¤LÑ0Ð åœY ~¸ÀrÔ8JÈQÑ8NÐ7OÑPÔP×WÒWÝ”˜uð Xñ ô ˆNð ×Ò × 3Ò 3Õ4GÈÑ4NÔ4NÑ OÔ OÑPÔPÐPØ×Ò˜n×1Ò1Õ2EÀcÑ2JÔ2JÑKÔKÑLÔLÐLÑLÝˆ|ÑÔ Ò!Ð!Ý”8˜B¥b¤hÐ/Ñ/Ô/µ´¸"ÅBÄHÐ1MÑ1MÔ1MÐMÐMÝ”˜|Ñ,Ô,ˆÝŒn˜ZÑ(Ô(ˆØ�cˆzÐr   )r   r   r   ro   rp   rq   rr   rs   rt   rq   r   r   r+   r,   )
r-   r.   r/   r0   r   r   r   r1   r*   rd   re   s   @r   rn   rn   \  sœ   ø€ € € € € ØWÐWð *.Ø)*Ø'+Ø,7Ø&*ð"
ð "
ð "
ð "
ð "
ð "
ð "
ðH €XÐ#Ñ$Ô$ð Ø"&Ø"Ø!Øð2ð 2ð 2ð 2ñ %Ô$ð2ð 2ð 2ð 2ð 2r   rn   c                  óH   — e Zd ZdZ ee¦  «        	 	 	 	 	 ddd„¦   «         ZdS )ÚExponentialMovingWindowIndexerz/Calculate ewm window bounds (the entire window)r   Nr   r   r   r    r!   r"   r#   r$   r%   r   r&   c                ó†   — t          j        dgt           j        ¬¦  «        t          j        |gt           j        ¬¦  «        fS )Nr   r9   )r?   r~   r8   r)   s         r   r*   z0ExponentialMovingWindowIndexer.get_window_bounds¼  s5   € õ Œx˜˜¥2¤8Ð,Ñ,Ô,­b¬h¸
°|Í2Ì8Ð.TÑ.TÔ.TÐTÐTr   r+   r,   rE   r2   r   r   r‡   r‡   ¹  sa   € € € € € Ø9Ð9à€XÐ#Ñ$Ô$ð Ø"&Ø"Ø!ØðUð Uð Uð Uñ %Ô$ðUð Uð Ur   r‡   )r0   Ú
__future__r   Údatetimer   Únumpyr?   Úpandas._libs.tslibsr   Úpandas._libs.window.indexersr   Úpandas.util._decoratorsr   Úpandas.core.dtypes.commonr   Úpandas.core.indexes.datetimesr	   Úpandas.tseries.offsetsr
   r1   r   r4   rG   rJ   rg   rk   rn   r‡   r2   r   r   ú<module>r’      s%  ðØ RÐ RØ "Ð "Ð "Ð "Ð "Ð "à Ð Ð Ð Ð Ð à Ð Ð Ð à *Ð *Ð *Ð *Ð *Ð *Ø IÐ IÐ IÐ IÐ IÐ IØ ,Ð ,Ð ,Ð ,Ð ,Ð ,à 9Ð 9Ð 9Ð 9Ð 9Ð 9à 7Ð 7Ð 7Ð 7Ð 7Ð 7à 'Ð 'Ð 'Ð 'Ð 'Ð 'ðÐ ð8,"ð ,"ð ,"ð ,"ð ,"ñ ,"ô ,"ð ,"ð^ð ð ð ð ˜ñ ô ð ð<
ð 
ð 
ð 
ð 
˜Kñ 
ô 
ð 
ð4ð ð ð ð  +ñ ô ð ðD
ð 
ð 
ð 
ð 
�{ñ 
ô 
ð 
ð$0ð 0ð 0ð 0ð 0 ñ 0ô 0ð 0ðfZð Zð Zð Zð Z�[ñ Zô Zð ZðzUð Uð Uð Uð U [ñ Uô Uð Uð Uð Ur   