U
|
¬ý°dª ã @ s¦ d Z ddlmZ ddlZddlZddlmZ ejdddddd ddd dd
|
d d d Z ejdddddd ddddddZ
|
ejdddddddd ddddZ dS )z
|
Numba 1D sum kernels that can be shared by
|
* Dataframe / Series
|
* groupby
|
* rolling / expanding
|
|
Mirrors pandas/_libs/window/aggregation.pyx
|
é )Ú annotationsN)Úis_monotonic_increasingTF)ZnopythonZnogilÚparallelÚfloatÚintz$tuple[int, float, float, int, float])ÚvalÚnobsÚsum_xÚ compensationÚnum_consecutive_same_valueÚ
|
prev_valueÚreturnc C sZ t | ¡sL|d7 }| | }|| }|| | }|}| |krD|d7 }nd}| }|||||fS ©Né ©ÚnpÚisnan)r r r r
|
r r ÚyÚt© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/_numba/kernels/sum_.pyÚadd_sum s
|
|
r ztuple[int, float, float])r r r r
|
r c C s> t | ¡s4|d8 }| | }|| }|| | }|}|||fS r r )r r r r
|
r r r r r Ú
|
remove_sum* s
|
|
r z
|
np.ndarray)ÚvaluesÚstartÚendÚ min_periodsr c C s~ t |}d}d}d}d}t|o&t|} tj|tjd}
|
t|D ]6} || } || } | dksb| s¤| | }d}t| | D ](}| | }t||||||\}}}}}qxnpt|| d | D ] }| | }t||||\}}}q¶t|| d | D ](}| | }t||||||\}}}}}qê|d kr,|kr6n nd}n*||krZ||krT|| }n|}ntj}||
|
| < | s@d}d}d}q@|
|
S )Nr g )Zdtyper ) Úlenr r ÚemptyÚfloat64Úranger r Únan)r r r r ÚNr r Zcompensation_addZcompensation_removeZ is_monotonic_increasing_boundsÚoutputÚiÚsÚer r Újr Úresultr r r Ú sliding_sum7 s ÿþ úúû ÿ úúû
|
|
|
r) ) Ú__doc__Ú
|
__future__r ZnumbaÚnumpyr Z!pandas.core._numba.kernels.sharedr Zjitr r r) r r r r Ú<module> s
|
