U
|
¬ý°d= ã
|
@ sª d Z ddlmZ ddlZddlZddlmZ ejdddddd dd dd dd
|
d d d Z ejdddddd dd dddddZ
|
ejdddddddd ddddZ dS )z
|
Numba 1D mean 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, int, float, int, float])ÚvalÚnobsÚsum_xÚneg_ctÚ compensationÚnum_consecutive_same_valueÚ
|
prev_valueÚreturnc C sl t | ¡s\|d7 }| | }|| }|| | }|}| dk rB|d7 }| |krT|d7 }nd}| }||||||fS ©Né r ©ÚnpÚisnan) r r r r
|
r r r ÚyÚt© r úWd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/_numba/kernels/mean_.pyÚadd_mean s
|
|
|
r z tuple[int, float, int, float])r r r r
|
r r c C sP t | ¡sD|d8 }| | }|| }|| | }|}| dk rD|d8 }||||fS r r )r r r r
|
r r r r r r Ú remove_mean- s
|
|
r z
|
np.ndarray)ÚvaluesÚstartÚendÚ min_periodsr c
|
C s¬ t |}d}d}d}d}d} t|o*t|}
|
tj|tjd} t|D ]`} || } || }| dksf|
|
s¬| | }d}t| |D ],}| | }t|||||||\}}}}}}q|nxt|| d | D ]$}| | }t||||| \}}}} q¾t|| d |D ],}| | }t|||||||\}}}}}}qö||kr|dkr|| }||krP|}n2|dkrj|dk rjd}n||kr|dkrd}ntj}|| | < |
|
sDd}d}d}d} qD| S )Nr g )Zdtyper ) Úlenr r ÚemptyÚfloat64Úranger r Únan)r r r r ÚNr r r
|
Zcompensation_addZcompensation_removeZ is_monotonic_increasing_boundsÚoutputÚiÚsÚer r Újr Úresultr r r Ú sliding_mean< s¢ ÿþ ùùú ÿùùú
|
r* ) Ú__doc__Ú
|
__future__r ZnumbaÚnumpyr Z!pandas.core._numba.kernels.sharedr Zjitr r r* r r r r Ú<module> s
|
