Ë
|
nñúh ã óè d Z ddlmZ ddlZddlZddlmZmZmZ ddl Z
|
ddl m Z ddl mZmZ erddlmZ d dZej& d d « Zej& d d
|
« Zy)z6Common utilities for Numba operations with groupby opsé )Ú annotationsN)Ú TYPE_CHECKINGÚAnyÚCallable)Úimport_optional_dependency)ÚNumbaUtilErrorÚjit_user_function)ÚScalarc ó" t | « s t d« t t j | « j
|
j « « }ddg}t |« }t |« |k s|d| |k7 r t d| d| j d| « y)a^
|
Validate user defined function for ops when using Numba with groupby ops.
|
|
The first signature arguments should include:
|
|
def f(values, index, ...):
|
...
|
|
Parameters
|
----------
|
func : function, default False
|
user defined function
|
|
Returns
|
-------
|
None
|
|
Raises
|
------
|
NumbaUtilError
|
z5Numba engine can only be used with a single function.ÚvaluesÚindexNz
|
The first z arguments to z must be )
|
ÚcallableÚNotImplementedErrorÚlistÚinspectÚ signatureÚ
|
parametersÚkeysÚlenr Ú__name__)ÚfuncÚ udf_signatureÚ expected_argsÚmin_number_argss úMH:\Change_password\venv_build\Lib\site-packages\pandas/core/groupby/numba_.pyÚ validate_udfr s¢ ô, D>Ü!Ø Có
|
ð
|
ô ×*Ñ*¨4Ó0×;Ñ;×@Ñ@ÓBÓCMØ wÐ'MÜ-Ó(Oä MÓ_Ò,Ø Ð)/Ð *¨mÒ ;ä Ø Ð)¨ ¸¿ ¹ °ÀiØ oð ó
|
ð
|
ð <ó c ó¤ t | « t rddln t d« j |||¬« dfd « }|S )a
|
Generate a numba jitted agg function specified by values from engine_kwargs.
|
|
1. jit the user's function
|
2. Return a groupby agg function with the jitted function inline
|
|
Configurations specified in engine_kwargs apply to both the user's
|
function _AND_ the groupby evaluation loop.
|
|
Parameters
|
----------
|
func : function
|
function to be applied to each group 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 ó( t |« t |« k( sJ t |« }t j ||f« } j |« D ]A }||| || } j |« D ] }
|
| || ||
|
|
f } | | g|¢ |||
|
f< " C |S ©N©r ÚnpÚemptyÚprange©r r ÚbeginÚendÚ num_columnsÚargsÚ
|
num_groupsÚresultÚiÚ group_indexÚjÚgroupr Ú
|
numba_funcs r Ú group_aggz*generate_numba_agg_func.<locals>.group_agge s» ø ô 5zS XÒ%Ð%Ð%ÜZ
|
ä: {Ð3Ó4Ø jÓ)ò EAØ a¡¨3¨q©6Ð2KØ\\ +Ó.ò EØ u Qx¨#¨a©&Ð 0°!Ð 3Ñ4Ù )¨%°Ð D¸tÒ Dq!t ñ Eð Eð
|
r ©r ú
|
np.ndarrayr r8 r+ r8 r, r8 r- Úintr. r Úreturnr8 ©r r r r Újit)r r! r" r# r6 r r5 s @@r Úgenerate_numba_agg_funcr= @ s ù ô> # 4Ó(JÝÜä*¨7Ó3à
|
YY¨¸YÓAðØðàðð ðð ð ð
|
ð ð ð ð
|
õó Bðð& Ðr c ó¤ t | « t rddln t d« j |||¬« dfd « }|S )a
|
Generate a numba jitted transform function specified by values from engine_kwargs.
|
|
1. jit the user's function
|
2. Return a groupby transform function with the jitted function inline
|
|
Configurations specified in engine_kwargs apply to both the user's
|
function _AND_ the groupby evaluation loop.
|
|
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 óJ t |« t |« k( sJ t |« }t j t | « |f« } j |« D ]I }||| || } j |« D ]( }
|
| || ||
|
|
f } | | g|¢ ||| ||
|
|
f< * K |S r% r& r* s r Úgroup_transformz6generate_numba_transform_func.<locals>.group_transform¡ sÎ ø ô 5zS XÒ%Ð%Ð%ÜZ
|
ä3v;¨ Ð4Ó5Ø jÓ)ò UAØ a¡¨3¨q©6Ð2KØ\\ +Ó.ò UØ u Qx¨#¨a©&Ð 0°!Ð 3Ñ4Ù/9¸%ÀÐ/TÈtÒ/TuQx # a¡&Ð(¨!Ð+Ò,ñ Uð Uð
|
r r7 r; )r r! r" r# r@ r r5 s @@r Ú generate_numba_transform_funcrA | s ù ô> # 4Ó(JÝÜä*¨7Ó3à
|
YY¨¸YÓAðØðàðð ðð ð ð
|
ð ð ð ð
|
õó Bðð& Ðr )r r r: ÚNone)
|
r zCallable[..., Scalar]r! Úboolr" rC r# rC r: úPCallable[[np.ndarray, np.ndarray, np.ndarray, np.ndarray, int, Any], np.ndarray])
|
r zCallable[..., np.ndarray]r! rC r" rC r# rC r: rD )Ú__doc__Ú
|
__future__r Ú functoolsr Útypingr r r Únumpyr' Úpandas.compat._optionalr Úpandas.core.util.numba_r r Úpandas._typingr
|
r Úcacher= rA © r r ú<module>rO sÅ ðÙ <Ý "ã Û ÷ñ ó å >÷ñ
|
Ý%ó$
|
ðN ð8Ø
|
ð8àð8ð ð8ð ð 8ð
|
Vò 8ó ð8ðv ð8Ø
|
#ð8àð8ð ð8ð ð 8ð
|
Vò 8ó ñ8r
|
