U
|
¬ý°dÚ ã @ sh d dl mZ d dlZd dlmZmZ d dlZd dlm Z d dl
|
m Z ej ddddddd d
|
d Z dS ) é )Ú annotationsN)Ú TYPE_CHECKINGÚCallable)ÚScalar)Úimport_optional_dependency)ÚmaxsizezCallable[..., Scalar]Úbool)ÚfuncÚnopythonÚnogilÚparallelc sD t rddlntdj|||dddddd fdd }|S )
|
aÓ
|
Generate a Numba function that loops over the columns 2D object and applies
|
a 1D numba kernel over each column.
|
|
Parameters
|
----------
|
func : function
|
aggregation function to be applied to each column
|
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)r
|
r r z
|
np.ndarrayÚint)ÚvaluesÚstartÚendÚ min_periodsc sd t jt|| jd ft jd} | jd ¡D ]0} | d d
|f |||f| |d d
|f< q.|S )Né )Zdtype)ÚnpÚemptyÚlenÚshapeÚfloat64Zprange)r r r r ÚargsÚresultÚi©r r © úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/_numba/executor.pyÚ column_looper. s .z1generate_shared_aggregator.<locals>.column_looper)r r r Zjit)r r
|
r r r r r r Úgenerate_shared_aggregator s
|
r )Ú
|
__future__r Ú functoolsÚtypingr r Únumpyr Zpandas._typingr Zpandas.compat._optionalr Ú lru_cacher r r r r Ú<module> s
|
|
