U
|
¬ý°d¢3 ã @ sÚ d Z ddlmZ ddlmZ ddlZddlmZ ddl m
|
Z
|
ddl m Z ddl mZ d ZG d
|
d d ZG d d d eZG dd deZG dd deZG dd deZG dd deZG dd deZG dd deZdS )zLIndexer objects for computing start/end window bounds for rolling operationsé )Ú annotations)Ú timedeltaN)Ú calculate_variable_window_bounds)ÚAppender)Úensure_platform_int)Ú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 @ sF e Zd ZdZddddddd Zeeddd
|
d d d
|
d dddZdS )Ú BaseIndexerz*Base class for window bounds calculations.Nr únp.ndarray | NoneÚintÚNone©Ú index_arrayÚ window_sizeÚreturnc K s. || _ || _| ¡ D ]\}}t| || qdS )z
|
Parameters
|
----------
|
**kwargs :
|
keyword arguments that will be available when get_window_bounds is called
|
N)r r ÚitemsÚsetattr)Úselfr r ÚkwargsÚkeyÚvalue© r úSd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/indexers/objects.pyÚ__init__. s zBaseIndexer.__init__ú
|
int | Noneú bool | Noneú
|
str | Noneú tuple[np.ndarray, np.ndarray]©Ú
|
num_valuesÚ min_periodsÚcenterÚclosedÚstepr c C s t d S ©N)ÚNotImplementedError©r r r r r! r" r r r Úget_window_bounds= s z BaseIndexer.get_window_bounds)Nr )r NNNN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r Úget_window_bounds_docr& r r r r r + s ÿ úr c @ s2 e Zd ZdZeed ddddddd d
|
d ZdS ) ÚFixedWindowIndexerz3Creates window boundaries that are of fixed length.r Nr
|
r r r r r c C s |r| j d d }nd}tjd| |d | |dd}|| j }|dkrP|d8 }|dkr`|d8 }t |d|¡}t |d|¡}||fS )Né é r Úint64©Zdtype©ÚleftÚboth)r2 Zneither)r ÚnpÚarangeÚclip) r r r r r! r" ÚoffsetÚendÚstartr r r r& L s
|
z$FixedWindowIndexer.get_window_bounds)r NNNN©r' r( r) r* r r+ r& r r r r r, I s úr, c @ s2 e Zd ZdZeed ddddddd d
|
d ZdS ) ÚVariableWindowIndexerzNCreates window boundaries that are of variable length, namely for time series.r Nr
|
r r r r r c C s t || j|||| jS r# )r r r r% r r r r& j s úz'VariableWindowIndexer.get_window_bounds)r NNNNr: r r r r r; g s úr; c sN e Zd ZdZddddd fdd Zeeddd
|
d d d
|
d dddZ ZS )ÚVariableOffsetWindowIndexerzNCalculate window boundaries based on a non-fixed offset such as a BusinessDay.Nr r r
|
r r c s" t j||f| || _|| _d S r# )Úsuperr Úindexr7 )r r r r> r7 r ©Ú __class__r r r s z$VariableOffsetWindowIndexer.__init__r r r r r c C s¸ |d k rt d|dkr4tjdddtjdddfS |d krN| jd k rJdnd}|dk}|dk}| j|d | jd k r|d
|
}nd }tj|dd} | d
|
¡ tj|dd}
|
|
|
d
|
¡ d| d< |rÆd |
|
d< nd|
|
d< td |D ]Ö} | j| } | j| || j } |r | td 8 } | | | < t| | d | D ]0}| j| | | tdkr&|| | < qXq&| j|
|
| d | | tdkr| d |
|
| < n|
|
| d |
|
| < |sØ|
|
| d 8 < qØ| |
|
fS ) Nz/step not implemented for variable offset windowr r/ r0 Úrightr3 )rA r3 r1 r- éÿÿÿÿ) r$ r4 Úemptyr> ÚfillÚranger7 r r )r r r r r! r" Z right_closedZ left_closedZindex_growth_signr9 r8 ÚiZ end_boundZ start_boundÚjr r r r& sF
|
|
|
|
|
$z-VariableOffsetWindowIndexer.get_window_bounds)Nr NN)r NNNN) r' r( r) r* r r r+ r& Ú __classcell__r r r? r r< s û úr< c @ s2 e Zd ZdZeed ddddddd d
|
d ZdS ) ÚExpandingIndexerz;Calculate expanding window bounds, mimicking df.expanding()r Nr
|
r r r r r c C s& t j|t jdt jd|d t jdfS )Nr0 r- )r4 Úzerosr/ r5 r% r r r r& Û s
|
þz"ExpandingIndexer.get_window_bounds)r NNNNr: r r r r rI Ø s úrI c @ s2 e Zd ZdZeed ddddddd d
|
d ZdS ) Ú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 c C s` |r t d|d k r t d|d kr(d}tjd||dd}|| j }| jrXt |d|¡}||fS )Nz.Forward-looking windows can't have center=TruezAForward-looking windows don't support setting the closed argumentr- r r/ r0 )Ú
|
ValueErrorr4 r5 r r6 )r r r r r! r" r9 r8 r r r r& s ÿ
|
z+FixedForwardWindowIndexer.get_window_bounds)r NNNNr: r r r r rK ê s úrK c s^ e Zd ZdZdddedfddddddd fd
|
d Zeedd d ddd ddddZ Z S )ÚGroupbyIndexerzMCalculate bounds to compute groupby rolling, mimicking df.groupby().rolling()Nr r zint | BaseIndexerz dict | Noneztype[BaseIndexer]r )r r Úgroupby_indicesÚwindow_indexerÚindexer_kwargsr c sH |pi | _ || _|r | ¡ ni | _t jf || j d|¡d| dS )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 N)rN rO ÚcopyrP r= r Úpop)r r r rN rO rP r r? r r r s
|
þýzGroupbyIndexer.__init__r
|
r r r r r c C sZ g }g }d}| j ¡ D ]ø\} }
|
| jd k r:| j t|
|
¡} n| j} | jf | | jd| j} | t |
|
||||¡\} }|
|
t j ¡} |
|
t j ¡}t | t |ks¤t dt ||t |
|
¡}|t |
|
7 }t ||d d g¡j
|
t j dd}| | t| ¡¡ | | t|¡¡ qt |dkr>t jg t j dt jg t j dfS t |¡} t |¡}| |fS ) Nr rQ z6these should be equal in length from get_window_boundsrB r- F)rR r0 )rN r r Ztaker rO r rP r& ÚlenZastyper4 r/ ÚAssertionErrorr5 ÚappendÚarrayZ concatenate)r r r r r! r" Z start_arraysZ
|
end_arraysZwindow_indices_startr Úindicesr Zindexerr9 r8 Zwindow_indicesr r r r& D sV
|
þý ÿ ÿþ
|
ÿ ÿ
|
|
z GroupbyIndexer.get_window_bounds)r NNNN)
|
r' r( r) r* r r r r+ r& rH r r r? r rM s ú $ úrM c @ s2 e Zd ZdZeed ddddddd d
|
d ZdS ) Ú ExponentialMovingWindowIndexerz/Calculate ewm window bounds (the entire window)r Nr
|
r r r r r c C s$ t jdgt jdt j|gt jdfS )Nr r0 )r4 rW r/ r% r r r r& } s z0ExponentialMovingWindowIndexer.get_window_bounds)r NNNNr: r r r r rY z s úrY )r* Ú
|
__future__r Údatetimer Únumpyr4 Z pandas._libs.window.indexersr Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr Zpandas.tseries.offsetsr r+ r r, r; r< rI rK rM rY r r r r Ú<module> s W3]
|
