Ë
|
nñúhn) ã óx d dl mZ d dlZd dlmZmZmZ d dlZd dl m
|
Z
|
d dl m Z erd dl mZ ej d d« Zej d d« Zej dd « Zej d
|
« Zej d d « Zy)é )Ú annotationsN)Ú TYPE_CHECKINGÚAnyÚCallable)Úimport_optional_dependency)Újit_user_function)ÚScalarc ó t | « t rddln t d« j |||¬« dfd « }|S )a
|
Generate a numba jitted apply function specified by values from engine_kwargs.
|
|
1. jit the user's function
|
2. Return a rolling apply function with the jitted function inline
|
|
Configurations specified in engine_kwargs apply to both the user's
|
function _AND_ the rolling apply function.
|
|
Parameters
|
----------
|
func : function
|
function to be applied to each window and will be JITed
|
nopython : bool
|
nopython to be passed into numba.jit
|
nogil : bool
|
nogil to be passed into numba.jit
|
parallel : bool
|
parallel to be passed into numba.jit
|
|
Returns
|
-------
|
Numba function
|
r NÚnumba©ÚnopythonÚnogilÚparallelc óR t j t |« « } j t |« « D ]j }|| }|| }| || } t j t j
|
| « « }
|
t | « |
|
z
|
|k\ r | g|¢ ||< Xt j ||< l |S )N)ÚnpÚemptyÚlenÚprangeÚsumÚisnanÚnan) ÚvaluesÚbeginÚendÚminimum_periodsÚargsÚresultÚiÚstartÚstopÚwindowÚ count_nanr Ú
|
numba_funcs úLH:\Change_password\venv_build\Lib\site-packages\pandas/core/window/numba_.pyÚ
|
roll_applyz-generate_numba_apply_func.<locals>.roll_apply9 s ø ô #e*Ó%Ø c &kÓ*ò #AØ!HEØq6DØE $Ð'FÜrxx¨Ó /Ó0IÜ6{YÑ&¨/Ò9Ù & vÐ 5°Ò 5q ä FFq ð #ð ó ) r ú
|
np.ndarrayr r' r r' r Úintr r Úreturnr' ©r r r r Újit)Úfuncr r r r% r r# s @@r$ Úgenerate_numba_apply_funcr- s ù ô> # 4Ó(JÝÜä*¨7Ó3à
|
YY¨¸YÓAðØðàðð ðð ð ð
|
ð ð
|
õ ó Bðð& Ðr& c ó t rddl n t d« j | ||¬« d fd « }|S )aÍ
|
Generate a numba jitted ewm mean or sum function specified by values
|
from engine_kwargs.
|
|
Parameters
|
----------
|
nopython : bool
|
nopython to be passed into numba.jit
|
nogil : bool
|
nogil to be passed into numba.jit
|
parallel : bool
|
parallel to be passed into numba.jit
|
com : float
|
adjust : bool
|
ignore_na : bool
|
deltas : tuple
|
normalize : bool
|
|
Returns
|
-------
|
Numba function
|
r Nr r c ó t j t | « « }ddz z }d|z
|
}rdn|}j t |« « D ]. }|| } || }
|
| | |
|
} t j t | « « } | d } t t j
|
| « « }||k\ r| nt j | d< d}t dt | « « D ]¡ }| | }t j
|
|« }||z }t j
|
| « sL|ssLr||| |z dz
|
z z }n|| z } |r1r%| |k7 r|| z ||z z } r| ||z z } r||z }n d}n
|
| |z } n|r|} ||k\ r| nt j | |< £ | || |
|
1 |S )Nç ð?r é )r r r r r( r r Úrange)r r r r r ÚalphaÚ old_wt_factorÚnew_wtr r r r! Ú
|
sub_resultÚweightedÚnobsÚold_wtÚjÚcurÚis_observationÚadjustÚcomÚdeltasÚ ignore_naÚ normalizer s r$ Úewmz$generate_numba_ewm_func.<locals>.ewmv s¹ ø ô #f+Ó&ØsSyÑ!Øe Ù Eà c %jÓ)ó ) ,AØ!HEØq6DØE $Ð'Fܤ# f£+Ó.Jà ayHÜ288 HÓ-Ð-Ó.DØ(,°Ò(?HÄRÇVÁVJqMØFä1c &kÓ*ò PØ QiÜ%'§X¡X¨c£]Ð!2ØÑ&Üxx Ô)Ù%©YÙ$ð # m°v¸eÀa¹iÈ!¹mÑ7LÑ&LÑ LFà'4°xÑ'?HÙ)Ù (à#+¨s¢?Ø/5¸Ñ/@À6ÈCÁ<Ñ/O HÙ'0Ø3;¸vȹÑ3O¨Ù#)Ø$*¨fÑ$4¡Fà-0¡Fà (¨C¡¡Ù#Ø "Hà,0°OÒ,C¡ÌÏÉ
|
1 ð9 Pð< ",F5Ò ðS) ,ðV r& ©
|
r r' r r' r r' r r( r) r' ©r r r r+ )
|
r r r r> r= r@ r? rA rB r s
|
````` @r$ Úgenerate_numba_ewm_funcrE P ss ý õB Üä*¨7Ó3à
|
YY¨¸YÓAð6Øð6àð6ð ð6ð ð 6ð
|
|
÷ 6ñ 6ó Bð6ðp Jr& c ó t | « t rddln t d« j |||¬« dfd « }|S )aé
|
Generate a numba jitted function to apply window calculations table-wise.
|
|
Func will be passed a M window size x N number of columns array, and
|
must return a 1 x N number of columns array. Func is intended to operate
|
row-wise, but the result will be transposed for axis=1.
|
|
1. jit the user's function
|
2. Return a rolling apply function with the jitted function inline
|
|
Parameters
|
----------
|
func : function
|
function to be applied to each window and will be JITed
|
nopython : bool
|
nopython to be passed into numba.jit
|
nogil : bool
|
nogil to be passed into numba.jit
|
parallel : bool
|
parallel to be passed into numba.jit
|
|
Returns
|
-------
|
Numba function
|
r Nr r c óò t j t |« | j d f« }t j |j « }j t |« « D ]g }|| }|| } | || }
|
t j
|
t j |
|
« d¬« } |
|
g|¢ } t |
|
« | z
|
|k\ } | ||d d
f< | ||d d
f< i t j ||t j « }|S )Nr1 r )Úaxis) r r r Úshaper r r Úwherer )r r r r r r Úmin_periods_maskr r r r! r" r6 Únan_maskr r# s r$ Ú
|
roll_tablez-generate_numba_table_func.<locals>.roll_tableØ sæ ø ô 3u: v§|¡|°A¡Ð7Ó8Ü 88 F§L¡LÓ1ÐØ c &kÓ*ò &AØ!HEØq6DØE $Ð'FÜrxx¨Ó /°aÔ8IÙ# FÐ2¨TÒ2JÜ6{ YÑ.°/ÑAHØ%-Ð Q¢TÑ "Ø%F1a4Lð &ô Ð*¨F´B·F±FÓ;Ø r& )
|
r r' r r' r r' r r( r r r* )r, r r r rM r r# s @@r$ Úgenerate_numba_table_funcrN ² sx ù ô@ # 4Ó(JÝÜä*¨7Ó3à
|
YY¨¸YÓAðØðàðð ðð ð ð
|
õ ó Bðð* Ðr& c óp t rdd ln t d« j ddd¬« fd« }|S )Nr r Tr c óº t j | j d « }j | j d « D ] }| d d
|f } |« ||< |S )Nr1 )r r rI r )Útabler r Ú partitionÚnan_funcr s r$ Únan_agg_with_axiszBgenerate_manual_numpy_nan_agg_with_axis.<locals>.nan_agg_with_axisû s[ ø ä%++ a.Ó)Ø ekk¨!nÓ-ò ,AØ a d IÙ Ó+F1Ið ,ð r& rD )rS rT r s ` @r$ Ú'generate_manual_numpy_nan_agg_with_axisrU ô s< ù åÜä*¨7Ó3à
|
YY D°4YÓ8ôó 9ðð Ð r& c ó t rddl n t d« j | ||¬« d fd « }|S )aß
|
Generate a numba jitted ewm mean or sum function applied table wise specified
|
by values from engine_kwargs.
|
|
Parameters
|
----------
|
nopython : bool
|
nopython to be passed into numba.jit
|
nogil : bool
|
nogil to be passed into numba.jit
|
parallel : bool
|
parallel to be passed into numba.jit
|
com : float
|
adjust : bool
|
ignore_na : bool
|
deltas : tuple
|
normalize: bool
|
|
Returns
|
-------
|
Numba function
|
r Nr r c óD ddz z }d|z
|
}rdn|}t j | j d « }t j | j « }| d j « } t j
|
| « j t j « }
|
t j |
|
|k\ | t j « |d< t dt | « « D ]@ } | | } t j
|
| « } |
|
| j t j « z }
|
j t | « « D ]¸ }t j
|
| | « s| | sr#r||xx || dz
|
z z cc< n || | z | |< | | sMrK| | | | k7 r*|| | | z || | z z | |< r| | || |z z | |< r||xx |z cc< d||< | |xx | | z cc< «| | s±| | | |< º t j |
|
|k\ | t j « || < C |S )Nr0 r1 r ) r ÚonesrI r Úcopyr ÚastypeÚint64rJ r r2 r r )r r r r r3 r4 r5 r9 r r7 r8 r r; Úis_observationsr: r= r> r? r@ rA r s r$ Ú ewm_tablez0generate_numba_ewm_table_func.<locals>.ewm_table, s ø ð sSyÑ!Øe Ù EÜ a Ó)ä&,,Ó'Ø!9>>Ó#Ü(Ó#Ð#×+Ñ+¬B¯H©HÓ5ÜHHT _Ñ 4°hÄÇÁÓGq Üq#f+Ó&ó LAØ)CÜ !xx¨}nOØ O×*Ñ*¬2¯8©8Ó4Ñ 4DØ\\¤# c£(Ó+ò )Üxx ¨¡ Ô,Ø& qÒ)²Ù$ð # 1I¨ ¸&ÀÀQÁ¹-Ñ)GÑ GIà*7¸(À1¹+Ñ*EH QKØ*¨1Ó-Ù (à#+¨A¡;°#°a±&Ò#8à(.¨q© °H¸Q±KÑ(?À&È3ÈqÉ6Á/Ñ(Qð %-¨Q¡Kñ (1Ø6>¸q±kÀVÈAÁYÐQWÑEWÑ6X¨°© Ù#)Ø$*¨1£I°Ñ$7¤Ià03 F¨1¢Ià (¨£ ¨s°1©vÑ 5¤ Ø$ QÓ'Ø"% a¡&HQKð3 )ô6 ¨ Ñ!8¸(ÄBÇFÁFÓKF1Ið? LðB r& rC rD )
|
r r r r> r= r@ r? rA r] r s
|
````` @r$ Ú generate_numba_ewm_table_funcr^ st ý õB Üä*¨7Ó3à
|
YY¨¸YÓAð0Øð0àð0ð ð0ð ð 0ð
|
|
÷ 0ñ 0ó Bð0ðd Ðr& )r, zCallable[..., Scalar]r Úboolr r_ r r_ )r r_ r r_ r r_ r> Úfloatr= r_ r@ r_ r? ÚtuplerA r_ )r, zCallable[..., np.ndarray]r r_ r r_ r r_ )Ú
|
__future__r Ú functoolsÚtypingr r r Únumpyr Úpandas.compat._optionalr Úpandas.core.util.numba_r Úpandas._typingr Úcacher- rE rN rU r^ © r& r$ ú<module>rk s ðÝ "ã ÷ñ ó å >å 5áÝ%ð ð8Ø
|
ð8àð8ð ð8ð ò 8ó ð8ðv ð^Øð^à ð^ð ð^ð
|
ð ^ð
|
ð ^ð ð ^ð ð^ð ò^ó ð^ðB ð;Ø
|
#ð;àð;ð ð;ð ò ;ó ð;ðB ñ ó ð ð" ðXØðXà ðXð ðXð
|
ð Xð
|
ð Xð ð Xð ðXð òXó ñXr&
|
