
    A>i                      % S SK Jr  S SKrS SKJr  S SKJr  S SKrS SKJr  S SK	r	S SK
JrJrJrJrJr  S SKrS SKJr  S SKJrJrJrJrJrJrJrJr  S S	KJr  S S
KJ r   S SK!J"r"J#r#  S SK$J%r%  S SK&J'r'J(r(J)r)J*r*J+r+  S SK,J-r-  S SK.J/r/J0r0J1r1  S SK2J3r3  S SK4J5s  J6r7  S SK8J9r9  S SK:J;r;J<r<  \(       a*  S SKJ=r=J>r>J?r?J@r@JArA  S SKBJCrCJDrDJErE  S SKFJGrG  S SKHJIrI  S SKJJKrK  \L\M\4   rNS\OS'   \#" S5       " S S\R                  5      5       rQ        S9                 S:S jjrR " S S \R                  S!9rT " S" S#\T5      rU " S$ S\U5      rV " S% S&\V5      rW " S' S(\V5      rX " S) S*\U5      rY " S+ S,\T5      rZ " S- S.\Z5      r[    S;S/ jr\S<S0 jr]    S=S1 jr^    S>S2 jr_          S?S3 jr`S@S4 jraSAS5 jrbSBS6 jrc    SCS7 jrdSDS8 jreg)E    )annotationsN)defaultdict)Callable)partial)TYPE_CHECKINGAnyLiteral	TypeAliascast)BlockValuesRefs)AggFuncTypeAggFuncTypeBaseAggFuncTypeDict
AggObjTypeAxisAxisIntNDFrameTnpt)import_optional_dependency)SpecificationError)cache_readonly
set_module)is_nested_object)is_dict_likeis_extension_array_dtypeis_list_likeis_numeric_dtypeis_sequence)ExtensionDtype)ABCDataFrame
ABCNDFrame	ABCSeries)generate_apply_looper)ensure_wrapped_if_datetimelike)get_jit_argumentsprepare_function_arguments)	GeneratorHashableIterableMutableMappingSequence)	DataFrameIndexSeries)GroupBy)	Resampler)
BaseWindowr
   ResTypezpandas.api.executorsc                      \ rS rSrSr\\R                              SS j5       5       r\\R                              SS j5       5       r	Sr
g)	BaseExecutionEngineQ   a  
Base class for execution engines for map and apply methods.

An execution engine receives all the parameters of a call to
``apply`` or ``map``, such as the data container, the function,
etc. and takes care of running the execution.

Supporting different engines allows functions to be JIT compiled,
run in parallel, and others. Besides the default executor which
simply runs the code with the Python interpreter and pandas.
c                    g)a  
Executor method to run functions elementwise.

In general, pandas uses ``map`` for running functions elementwise,
but ``Series.apply`` with the default ``by_row='compat'`` will also
call this executor function.

Parameters
----------
data : Series, DataFrame or NumPy ndarray
    The object to use for the data. Some methods implement a ``raw``
    parameter which will convert the original pandas object to a
    NumPy array, which will then be passed here to the executor.
func : function or NumPy ufunc
    The function to execute.
args : tuple
    Positional arguments to be passed to ``func``.
kwargs : dict
    Keyword arguments to be passed to ``func``.
decorator : function, optional
    For JIT compilers and other engines that need to decorate the
    function ``func``, this is the decorator to use. While the
    executor may already know which is the decorator to use, this
    is useful as for a single executor the user can specify for
    example ``numba.jit`` or ``numba.njit(nogil=True)``, and this
    decorator parameter will contain the exact decorator from the
    executor the user wants to use.
skip_na : bool
    Whether the function should be called for missing values or not.
    This is specified by the pandas user as ``map(na_action=None)``
    or ``map(na_action='ignore')``.
N )datafuncargskwargs	decoratorskip_nas         P/var/www/html/land-tabula/venv/lib/python3.13/site-packages/pandas/core/apply.pymapBaseExecutionEngine.map_           c                    g)aW  
Executor method to run functions by an axis.

While we can see ``map`` as executing the function for each cell
in a ``DataFrame`` (or ``Series``), ``apply`` will execute the
function for each column (or row).

Parameters
----------
data : Series, DataFrame or NumPy ndarray
    The object to use for the data. Some methods implement a ``raw``
    parameter which will convert the original pandas object to a
    NumPy array, which will then be passed here to the executor.
func : function or NumPy ufunc
    The function to execute.
args : tuple
    Positional arguments to be passed to ``func``.
kwargs : dict
    Keyword arguments to be passed to ``func``.
decorator : function, optional
    For JIT compilers and other engines that need to decorate the
    function ``func``, this is the decorator to use. While the
    executor may already know which is the decorator to use, this
    is useful as for a single executor the user can specify for
    example ``numba.jit`` or ``numba.njit(nogil=True)``, and this
    decorator parameter will contain the exact decorator from the
    executor the user wants to use.
axis : {0 or 'index', 1 or 'columns'}
    0 or 'index' should execute the function passing each column as
    parameter. 1 or 'columns' should execute the function passing
    each row as parameter. The default executor engine passes rows
    as pandas ``Series``. Other executor engines should probably
    expect functions to be implemented this way for compatibility.
    But passing rows as other data structures is technically possible
    as far as the function ``func`` is implemented accordingly.
Nr7   )r8   r9   r:   r;   r<   axiss         r>   applyBaseExecutionEngine.apply   rA   rB   r7   N)r8   Series | DataFrame | np.ndarrayr9   r   r:   tupler;   dict[str, Any]r<   zCallable | Noner=   bool)r8   rG   r9   r   r:   rH   r;   rI   r<   r   rD   r   )__name__
__module____qualname____firstlineno____doc__staticmethodabcabstractmethodr?   rE   __static_attributes__r7   rB   r>   r4   r4   Q   s    
 '-'' ' 	'
 #' '  'R +-++ + 	+
 + +  +rB   r4   
FrameApplyc
                    [        U40 U	D6u  ppU R                  U5      nUS:X  a  [        nO#US:X  a  U(       a  [        SU< S35      e[        nW" U UUUUUUUU	S9	$ )z=construct and return a row or column based frame apply objectr      z-Named aggregation is not supported when axis=.)rawresult_typeby_rowengineengine_kwargsr:   r;   )reconstruct_func_get_axis_numberFrameRowApplyNotImplementedErrorFrameColumnApply)objr9   rD   rX   rY   rZ   r[   r\   r:   r;   _columnsklasss                r>   frame_applyrf      s     +4:6:AW%Dqy	%@4'C  !#
 
rB   c                     \ rS rSr% S\S'   SSSS.               SS jjr\R                  SS	 j5       r\R                      SS
 j5       r	\R                      SS j5       r
SS jrSS jrSS jrSS jrSS jr        S S jr    S!S jrSS jr          S"S jr      S#S jrSS jrSS jr        S$S jrS%S jrSrg)&Apply   r   rD   compatpythonNrZ   r[   r\   c                   Xl         X0l        USL d  US;   d   eXPl        U=(       d    SU l        U	=(       d    0 U l        X`l        Uc  0 OUU l        US;  a  [        S5      eX@l        X l	        g )NF)rj   _compatr7   )Nreduce	broadcastexpandzUinvalid value for result_type, must be one of {None, 'reduce', 'broadcast', 'expand'})
rb   rX   rZ   r:   r;   r[   r\   
ValueErrorrY   r9   )
selfrb   r9   rX   rY   rZ   r[   r\   r:   r;   s
             r>   __init__Apply.__init__   sz     &,A"AAAJB	l#0#8RmEE= 
 '	rB   c                    g Nr7   rs   s    r>   rE   Apply.apply      rB   c                    g rw   r7   rs   op_names     r>   agg_or_apply_list_likeApply.agg_or_apply_list_like       	rB   c                    g rw   r7   r|   s     r>   agg_or_apply_dict_likeApply.agg_or_apply_dict_like  r   rB   c                    U R                   n[        U[        5      (       a  U R                  5       $ [	        U5      (       a  U R                  5       $ [        U5      (       a  U R                  5       $ g)z
Provide an implementation for the aggregators.

Returns
-------
Result of aggregation, or None if agg cannot be performed by
this method.
N)r9   
isinstancestr	apply_strr   agg_dict_liker   agg_list_like)rs   r9   s     r>   agg	Apply.agg  s`     yydC  >>##%%''$%%'' rB   c                f   U R                   nU R                  nU R                  nU R                  nU R                  nUR
                  S:H  nUR                  U5      S:X  a4  U(       a   eUR                  R                  " US/UQ70 UD6R                  $ [        U5      (       at  [        U5      (       dd  [        [        [           U5      nU(       a1  U Vs0 s H#  n[        R                  " U5      =(       d    UU_M%     nnO[         R#                  X5      n[        U5      (       a!  [        [$        U5      nU R'                  U5      $ [        [        U5      n U R)                  U5      n[1        U[2        [4        45      (       a-  UR6                  (       a  UR6                  (       d  [/        S5      e[1        U[2        [4        45      (       a*  UR8                  R;                  UR8                  5      (       d  [/        S5      eU$ s  snf ! [*         a    e [,         a  n	[/        S5      U	eSn	A	ff = f)z
Transform a DataFrame or Series.

Returns
-------
DataFrame or Series
    Result of applying ``func`` along the given axis of the
    Series or DataFrame.

Raises
------
ValueError
    If the transform function fails or does not transform.
rV   r   zTransform function failedNzFunction did not transform)rb   r9   rD   r:   r;   ndimr^   T	transformr   r   r   listr   comget_callable_namedictfromkeysr   transform_dict_liketransform_str_or_callable	TypeError	Exceptionrr   r   r"   r    emptyindexequals)
rs   rb   r9   rD   r:   r;   	is_seriesvresulterrs
             r>   r   Apply.transform.  s    hhyyyyyyHHM	%*  =55??4<T<V<>>>l4&8&8_-t4DBFG$Q--a05Aq8$G}}S/.D++D11 OT*	C33D9F v	<899II899&9l";<<FLLDWDWIIE
 E
 9::= H  	 	C89sB	Cs   *HH H0H++H0c                r   SSK Jn  U R                  nU R                  nU R                  n[        U[        5      (       d   e[        U5      S:X  a  [        S5      eU R                  SX15      n0 nUR                  5        H.  u  pxUR                  USS9n	U	R                  " US/UQ70 UD6Xg'   M0     U" USS9$ )z3
Compute transform in the case of a dict-like func
r   concatz$No transform functions were providedr   rV   r   rD   )pandas.core.reshape.concatr   rb   r:   r;   r   r!   lenrr   normalize_dictlike_argitems_gotitemr   )
rs   r9   r   rb   r:   r;   resultsnamehowcolgs
             r>   r   Apply.transform_dict_likem  s     	6hhyy #z****t9>CDD**;B68ID<<1<-D NN3CDCFCGM & gA&&rB   c                   U R                   nU R                  nU R                  n[        U[        5      (       a  U R
                  " X!/UQ70 UD6$  UR                  " U4SU0UD6$ ! [         a    U" U/UQ70 UD6s $ f = f)z<
Compute transform in the case of a string or callable func
r:   )rb   r:   r;   r   r   
_apply_strrE   r   )rs   r9   rb   r:   r;   s        r>   r   Apply.transform_str_or_callable  s     hhyydC  ??3>t>v>>	.99T7777 	.-d-f--	.s   A& &A?>A?c                     U R                  SS9$ )zb
Compute aggregation in the case of a list-like argument.

Returns
-------
Result of aggregation.
r   r}   )r~   rx   s    r>   r   Apply.agg_list_like       **5*99rB   c                l   [        [        [           U R                  5      nU R                  n/ n/ nUR
                  S:X  a  U H  nUR                  UR                  SUS9n	[        X5      (       a  U R                  /U R                  QOU R                  n
[        X5      " U/U
Q70 UD6nUR                  U5        [        R                  " U5      =(       d    UnUR                  U5        M     Xv4$ / n[        U5       H  u  pUR                  USUR                   SS2U4   S9n	[        X5      (       a  U R                  /U R                  QOU R                  n
[        X5      " U/U
Q70 UD6nUR                  U5        UR                  U5        M     UR"                  R%                  U5      nXv4$ )a  
Compute agg/apply results for like-like input.

Parameters
----------
op_name : {"agg", "apply"}
    Operation being performed.
selected_obj : Series or DataFrame
    Data to perform operation on.
kwargs : dict
    Keyword arguments to pass to the functions.

Returns
-------
keys : list[Hashable] or Index
    Index labels for result.
results : list
    Data for result. When aggregating with a Series, this can contain any
    Python objects.
rV   r   subsetN)r   r   r   r9   rb   r   r   r   include_axisrD   r:   getattrappendr   r   	enumerateilocrd   take)rs   r}   selected_objr;   r9   rb   r   keysar   r:   new_resr   indicesr   cols                   r>   compute_list_likeApply.compute_list_like  s   4 D)4995hh !||L$5$5Al|S $G22 YY++ 
 "$0DTDVDw' ,,Q/41D! < } G'5
||Ca8I8I!U(8S|T $G22 YY++ 
 "$0GGGw'u% 6  '',,W5D}rB   c                    SSK Jn  U R                  n U" X!SSS9$ ! [         a?  nSSKJn  U" X!UR                  S9n[        U5      (       a  [        S5      UeUs S nA$ S nAff = f)	Nr   r   rV   F)r   rD   sortr.   r   r   z3cannot combine transform and aggregation operations)	r   r   rb   r   pandasr.   r   r   rr   )rs   r   r   r   rb   r   r.   r   s           r>   wrap_results_list_likeApply.wrap_results_list_like  sn     	6hh	'15AA 
	 &Gchh?F'' I M
	s    
A%4A A% A%c                     U R                  SS9$ )zb
Compute aggregation in the case of a dict-like argument.

Returns
-------
Result of aggregation.
r   r   )r   rx   s    r>   r   Apply.agg_dict_like  r   rB   c                   SSK JnJn  U R                  n[	        XuU45      n[        [        U R                  5      n	U R                  XU	5      n	UR                  S:H  =(       a0    UR                  R                  5       [        UR                  5      :  n
UR                  S:X  a`  UR                  USS9nU	R                  5        VVs/ s H  u  p[        X5      " U40 UD6PM     nnn[!        U	R#                  5       5      nX4$ U(       d  U
(       a  / n/ nU	R                  5        H  u  nnUR                  R%                  U/5      nUR                  R'                  U5      n[)        [         5      n[+        UUSS9 H  u  nnUU   R-                  U5        M     UR                  5        VVVs/ s H1  u  nnU  H$  n[        UR/                  USS9U5      " U40 UD6PM&     M3     nnnnUU/[        U5      -  -  nUU-  nM     X4$ U(       a  Un/ / pU	R                  5        H  u  nnUU   nUR                  S:X  aF  UR                  USUS	9nUR-                  [        UU5      " U40 UD65        UR-                  U5        Ma  UR                  5        HJ  u  nnUR                  USUS	9nUR-                  [        UU5      " U40 UD65        UR-                  U5        ML     M     X4$ U	R                  5        VVs/ s H'  u  nn[        UR                  USS9U5      " U40 UD6PM)     nnn[!        U	R#                  5       5      nX4$ s  snnf s  snnnf s  snnf )
a3  
Compute agg/apply results for dict-like input.

Parameters
----------
op_name : {"agg", "apply"}
    Operation being performed.
selected_obj : Series or DataFrame
    Data to perform operation on.
selection : hashable or sequence of hashables
    Used by GroupBy, Window, and Resample if selection is applied to the object.
kwargs : dict
    Keyword arguments to pass to the functions.

Returns
-------
keys : list[hashable]
    Index labels for result.
results : list
    Data for result. When aggregating with a Series, this can contain any
    Python object.
r   DataFrameGroupBySeriesGroupBy   rV   r   Tstrictr   r   )pandas.core.groupby.genericr   r   rb   r   r   r   r9   r   r   rd   nuniquer   r   r   r   r   r   get_indexer_forr   r   zipr   _ixs)rs   r}   r   	selectionr;   r   r   rb   
is_groupbyr9   is_non_unique_colr   rc   r   r   r   keyr   labelslabel_to_indicesr   labelindicekey_datadfcolsseriesr   s                               r>   compute_dict_likeApply.compute_dict_like  sG   :	

 hh%FG
OTYY/**7$G " K$$,,.\5I5I1JJ 	
 !<<	<2DKO::<X<wt-c<V<<GX		$DZ }Y  1 GD JJLS&..>>uE%--227;#.t#4 $'$ELE5$U+2259 %F
 +;*@*@*B*Bw") L--f1-=wGVvV") W*B   H--8# )N }1  BT JJLS#w99> \\#Ad\CFNN767#;C#J6#JKKK$"&**,3!$c#!Fwvw'?'Nv'NOC( #/ )& }	 !%

 ,HC Sq17;CJ6J ,   		$D}] Y4s   M/8M
>.Mc                   SSK Jn  SSKJn  U R                  nU Vs/ s H  n[        U[        5      PM     nn[        U5      (       a  U V	s/ s H  oR                  (       a  M  U	PM     n
n	[        X#SS9 VVs/ s H  u  pUR                  (       a  M  UPM     nnnU/ :X  a  UnUn
UR                  S:X  a/  U" U5      nUR                  UR                  R                  5        Un[        U[        5      (       a  SOSnU" U
UUSS	9n	U	$ [        U5      (       a  [!        S
5      eSSK Jn  UR                  S:X  a  [%        SU5      nUR&                  nOS nU" X2US9n	U	$ s  snf s  sn	f s  snnf )Nr   r-   r   Tr   r   rV   F)rD   r   r   zLcannot perform both aggregation and transformation operations simultaneouslyr   r.   r   )r   r-   r   r   rb   r   r!   allr   r   r   
_set_namesrd   namesr"   anyrr   r.   r   r   )rs   r   result_indexresult_datar-   r   rb   r
is_ndframer   r   kr   keys_to_usekturD   r.   r   s                     r>   wrap_results_dict_likeApply.wrap_results_dict_like`  sh    	!5hh :EEAjJ/
Ez??,7LK&||vKGL ",DIIdaQRQXQXI   b *%  A%K(|3399:!!+C!;!;AD 	F4 ) __!  & xx1}8S)xxK$GF] F Ms   E E%%E%:E*E*c                   [        [        U R                  5      nU R                  nSSKJnJn  [        X!S5      n[        U5      (       a  [        R                  " U5      n/ UR                  QUR                  Q7nU R                  S:w  a  SU;  d  US;   a  [        SU S35      eSU;   a+  [        X$U45      (       d  U R                  U R                   S'   U R"                  " X!/U R                  Q70 U R                   D6$ )zQ
Compute apply in case of a string.

Returns
-------
result: Series or DataFrame
r   r   NrD   )corrwithskewz
Operation z does not support axis=1)r   r   r9   rb   r   r   r   r   callableinspectgetfullargspecr:   
kwonlyargsrD   rr   r   r;   r   )rs   r9   rb   r   r   methodsig	arg_namess           r>   r   Apply.apply_str  s     C#hh	
 D)F((0C4#((4S^^4IyyA~i'43G+G :dV3K!LMM":%56, , '+iiF#sD499DDDrB   c                   U R                   S:X  a  [        S5      eU R                  S:X  aq  [        U R                  [
        5      (       aR  U R                  R                  R                  " U R                  S4SU R                  0U R                  D6R                  $ U R                  nU R                  n[        U5      (       a  U R                  SS9nOU R                  SS9n[        X140 UD6nU$ )z
Compute apply in case of a list-like or dict-like.

Returns
-------
result: Series, DataFrame, or None
    Result when self.func is a list-like or dict-like, None otherwise.
numbazIThe 'numba' engine doesn't support list-like/dict likes of callables yet.rV   r   r:   rE   r   )r[   r`   rD   r   rb   r    r   rE   r9   r:   r;   r   r   r~   reconstruct_and_relabel_result)rs   r9   r;   r   s       r>   apply_list_or_dict_likeApply.apply_list_or_dict_like  s     ;;'!%/ 
 99>j<@@88::##DIIqPtyyPDKKPRRRyy000AF000AF/GGrB   c                  ^	 US;   d   eUS:X  a:  [        U[        5      (       a%  [        S UR                  5        5       5      (       d%  [        S UR                  5        5       5      (       a  [	        S5      eUR
                  S:w  ad  SSKJn  U" [        UR                  5       5      5      R                  UR                  S	S
9n[        U5      S:  a  [        S[        U5       S35      e[        [        [        4m	[        U	4S jUR                  5        5       5      (       a9  0 nUR                  5        H!  u  px[        UT	5      (       d  U/Xg'   M  XU'   M#     UnU$ )z
Handler for dict-like argument.

Ensures that necessary columns exist if obj is a DataFrame, and
that a nested renamer is not passed. Also normalizes to all lists
when values consists of a mix of list and non-lists.
)rE   r   r   r   c              3  <   #    U  H  u  p[        U5      v   M     g 7frw   )r   .0rc   r   s      r>   	<genexpr>/Apply.normalize_dictlike_arg.<locals>.<genexpr>  s     =LOO   c              3  <   #    U  H  u  p[        U5      v   M     g 7frw   )r   r  s      r>   r  r    s     ;lda,q//lr  znested renamer is not supportedrV   r   r   T)r   z	Label(s) z do not existc              3  @   >#    U  H  u  p[        UT5      v   M     g 7frw   r   )r  rc   xaggregator_typess      r>   r  r    s     H<41z!-..<s   )r   r"   r   r   r   r   r   r-   r   r   
differencerd   r   KeyErrorrH   r   )
rs   r   rb   r9   r-   r   new_funcr   r   r  s
            @r>   r   Apply.normalize_dictlike_arg  s%    3333 5L3	**=

===;djjl;;;$%FGG88q=$diik*+66s{{6ND4y1}4:,mDEE %. H4::<HHH(*H

!!%566#$#HK"#QK	 %
 DrB   c                   [        U[        5      (       d   e[        X5      (       aO  [        X5      n[	        U5      (       a  U" U0 UD6$ [        U5      S:X  d   e[        S U 5       5      (       a   eU$ [        [        U5      (       a-  [        US5      (       a  [        [        U5      nU" U/UQ70 UD6$ SU S[        U5      R                   S3n[        U5      e)z
if arg is a string, then try to operate on it:
- try to find a function (or attribute) on obj
- try to find a numpy function
- raise
r   c              3  *   #    U  H	  oS :H  v   M     g7f)rD   Nr7   )r  kwargs     r>   r  #Apply._apply_str.<locals>.<genexpr>  s     ?uF?s   	__array__'z' is not a valid function for 'z' object)r   r   hasattrr   r   r   r   nptyperK   AttributeError)rs   rb   r9   r:   r;   fmsgs          r>   r   Apply._apply_str  s     $$$$$3"A{{$)&)) t9>!>??????HR73#<#<D!AS*4*6**dV:49;M;M:NhWC %%rB   )	r:   rZ   r[   r\   r9   r;   rb   rX   rY   )rb   r   r9   r   rX   rJ   rY   
str | NonerZ   #Literal[False, 'compat', '_compat']r[   r   r\   dict[str, bool] | NonereturnNoner*  DataFrame | Seriesr}   Literal['agg', 'apply']r*  r-  )r*  zDataFrame | Series | None)r*  r,   )r}   r/  r   Series | DataFramer;   rI   r*  z(tuple[list[Hashable] | Index, list[Any]])r   Iterable[Hashable]r   zlist[Series | DataFrame])
r}   r/  r   r0  r   zHashable | Sequence[Hashable]r;   rI   r*  z tuple[list[Hashable], list[Any]])r   r0  r   zlist[Hashable]r   r   )r   r   rb   r-  r9   r   r*  r   )r9   r   )rK   rL   rM   rN   __annotations__rt   rQ   rR   rE   r~   r   r   r   r   r   r   r   r   r   r   r   r   r  r   r   rS   r7   rB   r>   rh   rh      s   
M 7?04!! ! 	!
  ! 4! ! .! 
!F 	  	.	 
 	.	 
.=~'0.":@(@ )@ 	@
 
2@D&1I*:](] )] 1	]
 ] 
*]~:(: %: 	:x!EF>,,/,7F,	,\&rB   rh   )	metaclassc                  p    \ rS rSr% SrS\S'   \SS j5       r\SS j5       r    SS jr	    SS jr
S	rg
)NDFrameApplyi(  z[
Methods shared by FrameApply and SeriesApply but
not GroupByApply or ResamplerWindowApply
r-  rb   c                .    U R                   R                  $ rw   )rb   r   rx   s    r>   r   NDFrameApply.index0  s    xx~~rB   c                L    U R                   R                  U R                  5      $ rw   )rb   _get_agg_axisrD   rx   s    r>   agg_axisNDFrameApply.agg_axis4  s    xx%%dii00rB   c                r   U R                   nU R                  nUS:X  aW  [        U [        5      (       a  U R                  nO-[        U [
        5      (       a  U R                  (       a  SOSnOSn0 UESU0En[        USS5      S:X  a  [        S5      eU R                  XU5      u  pVU R                  XV5      nU$ )	NrE   rn   FrZ   rD   r   rV   "axis other than 0 is not supported)
rb   r;   r   rT   rZ   SeriesApplyr   r`   r   r   )rs   r}   rb   r;   rZ   r   r   r   s           r>   r~   #NDFrameApply.agg_or_apply_list_like8  s     hhg$
++D+..&*kku11&1F3"a'%&JKK..wVD,,T;rB   c                   US;   d   eU R                   n0 nUS:X  a(  U R                  (       a  SOSnUR                  SU05        [        USS5      S:X  a  [	        S	5      eS nU R                  XXS5      u  pgU R                  X&U5      nU$ )
Nr   rE   rE   rn   FrZ   rD   r   rV   r=  )rb   rZ   updater   r`   r   r   )	rs   r}   rb   r;   rZ   r   r   r   r   s	            r>   r   #NDFrameApply.agg_or_apply_dict_likeO  s     ****hhg"&++Y5FMM8V,-3"a'%&JKK	$($:$:)%
! ,,SLrB   r7   Nr*  r-   r.  )rK   rL   rM   rN   rO   r2  propertyr   r:  r~   r   rS   r7   rB   r>   r5  r5  (  s_    
 
  1 1.	..	rB   r5  c                  J  ^  \ rS rSr% S\S'   SSSS.               SU 4S jjjr\\R                  SS	 j5       5       r	\\R                  SS
 j5       5       r
\\R                  S S j5       5       r\\R                  \R                   S! S"S jj5       5       5       r\R                  S 5       rS#S jr\R                        S$S j5       r\SS j5       r\SS j5       r\S 5       rS%S jrU 4S jrS rS&S jrS'S jrS rS(S jrS rS$S jrS%U 4S jjr Sr!U =r"$ ))rT   ie  r,   rb   Frk   Nrl   c               d   > USLa  US:w  a  [        SU S35      e[        T
U ]	  UUUUUUUUU	S9	  g )NFrj   zby_row=z not allowed)rZ   r[   r\   r:   r;   )rr   superrt   )rs   rb   r9   rX   rY   rZ   r[   r\   r:   r;   	__class__s             r>   rt   FrameApply.__init__h  sU     6X#5wvhl;<<' 	 
	
rB   c                    g rw   r7   rx   s    r>   r   FrameApply.result_index       	rB   c                    g rw   r7   rx   s    r>   result_columnsFrameApply.result_columns  rM  rB   c                    g rw   r7   rx   s    r>   series_generatorFrameApply.series_generator  rM  rB   c                    g rw   r7   )r9   nogilnopythonparallels       r>   generate_numba_apply_func$FrameApply.generate_numba_apply_func  s     	rB   c                    g rw   r7   rx   s    r>   apply_with_numbaFrameApply.apply_with_numba  rz   rB   c                    U R                   R                  R                  5        HF  u  p[        U5      (       d  [	        SU SU S35      e[        U5      (       d  M9  [	        SU S35      e   g )NzColumn z# must have a numeric dtype. Found 'z	' insteadzM is backed by an extension array, which is not supported by the numba engine.)rb   dtypesr   r   rr   r   )rs   colnamedtypes      r>   validate_values_for_numba$FrameApply.validate_values_for_numba  s|    "hhoo335NG#E** gY '#WI/  (.. gY 'B C  6rB   c                    g rw   r7   rs   r   	res_indexs      r>   wrap_results_for_axis FrameApply.wrap_results_for_axis  r   rB   c                    U R                   $ rw   )rO  rx   s    r>   res_columnsFrameApply.res_columns  s    """rB   c                .    U R                   R                  $ rw   )rb   rd   rx   s    r>   rd   FrameApply.columns  s    xxrB   c                .    U R                   R                  $ rw   )rb   valuesrx   s    r>   rn  FrameApply.values  s    xxrB   c                   [        U R                  5      (       a+  U R                  S:X  a  [        S5      eU R	                  5       $ [        U R                  5      S:X  a)  [        U R                  5      S:X  a  U R                  5       $ [        U R                  [        5      (       a+  U R                  S:X  a  [        S5      eU R                  5       $ [        U R                  [        R                  5      (       a  U R                  S:X  a  [        S5      e[        R                  " SS9   U R                  R                   R#                  SU R                  S	9nS
S
S
5        U R                  R%                  WUR&                  S9$ U R(                  S:X  a6  U R                  S:X  a  [        S5      eU R+                  U R                  5      $ [-        U R                  R.                  5      (       d  U R                  5       $ U R0                  (       a$  U R3                  U R                  U R4                  S9$ U R7                  5       $ ! , (       d  f       N= f)zcompute the resultsr  z9the 'numba' engine doesn't support lists of callables yetr   zJthe 'numba' engine doesn't support using a string as the callable functionzOthe 'numba' engine doesn't support using a numpy ufunc as the callable functionignorer   rE   )r9   N)axesrp   z:the 'numba' engine doesn't support result_type='broadcast'r[   r\   )r   r9   r[   r`   r  r   rd   r   apply_empty_resultr   r   r   r!  ufuncerrstaterb   _mgrrE   _constructor_from_mgrrs  rY   apply_broadcastr   shaperX   	apply_rawr\   apply_standardrs   r   s     r>   rE   FrameApply.apply  s    		""{{g%)O  //11 t||!c$**o&:**,, dii%%{{g%)8  >>## 		288,,{{g%)C  *((----gDII-F + 8811'1MM {*{{g%)P  ''11 TXX^^$$**,, XX>>DDVDV>WW""$$+ +*s   */I
Ic                  > U R                   nU R                  nU R                  S:X  a  U R                   OU R                   R                  U l         SU l        S n [        TU ]  5       nXl         X l        US:X  a  Ub  UR                  OUnUc>  U R                   R
                  " U R                  U4SU R                  0U R                  D6nU$ ! Xl         X l        f = f)Nr   rV   r:   )	rb   rD   r   rH  r   rE   r9   r:   r;   )rs   rb   rD   r   rI  s       r>   r   FrameApply.agg  s    hhyy  $yyA~488488::		W[]FHI19!'!3VXXF>XX^^DIItS$))St{{SF HIs   C Cc                   [        U R                  5      (       d   eU R                  S;  a  U R                  R	                  5       $ U R                  S:H  nSSKJn  U(       d   U R                  S:X  a>  U R                  " U" / [        R                  S9/U R                  Q70 U R                  D6nOGU R                  " U" U R                  [        R                  S9/U R                  Q70 U R                  D6n[        X25      (       + nU(       a  [        U R                   5      (       a>  U R                  " U" / [        R                  S9/U R                  Q70 U R                  D6nO[        R"                  nU R                  R%                  X0R                   S9$ U R                  R	                  5       $ ! [         a     Nf = f)z
we have an empty result; at least 1 axis is 0

we will try to apply the function to an empty
series in order to see if this is a reduction function
)ro   Nro   r   r   r`  )r   r`  r   )r   r9   rY   rb   copyr   r.   rD   r!  float64r:   r;   rd   r   r   r   r:  nan_constructor_sliced)rs   should_reducer.   r   s       r>   ru  FrameApply.apply_empty_result  sg    		"""" #3388==?" ((H4!:99>		r47;yyDHKKA 		T\\D ++A %/q$9 94==!!IIfRrzz:VTYYV$++VFF88///GG88==?"  s   $BF5 5
GGc                ~   S nUS:X  a  [        U R                  U R                  U R                  SS9u  pE[	        U R                  40 [        U5      D6nU" U R                  U R                  /UQ76 n[        R                  " U5      nOT[        R                  " U" U R                  5      U R                  U R                  /U R                  Q70 U R                  D6nUR                  S:X  a.  U R                  R                  XpR                  U R                  S9$ U R                  R!                  XpR"                  S9$ )z$apply to the values as a numpy arrayc                   ^  U 4S jnU$ )zi
Wrap user supplied function to work around numpy issue.

see https://github.com/numpy/numpy/issues/8352
c                 t   > T" U 0 UD6n[        U[        5      (       a  [        R                  " U[        S9nU$ )Nr  )r   r   r!  arrayobject)r:   r;   r   r9   s      r>   wrapper<FrameApply.apply_raw.<locals>.wrap_function.<locals>.wrapperH  s4    t.v.fc**XXfF;FrB   r7   )r9   r  s   ` r>   wrap_function+FrameApply.apply_raw.<locals>.wrap_functionA  s     NrB   r  rV   num_required_argsr   r   rd   r  )r&   r9   r:   r;   r#   r%   rn  rD   r!  squeezeapply_along_axisr   rb   _constructorr   rd   r  r:  )rs   r[   r\   r  r:   r;   	nb_looperr   s           r>   r|  FrameApply.apply_raw>  s   	 W5				"#	LD .		#M2I t{{DII==FZZ'F((dii(		 	
 ++F ;;!88((zz4<<(XX88//mm/LLrB   c                >   [        U R                  5      (       d   e[        R                  " UR                  5      nUR
                  S   n[        UR                  5       H  u  pEU R                  " X   /U R                  Q70 U R                  D6n[        R                  " U5      R                  nUS:  a  [        S5      eUS:X  a  U[        U5      :w  a  [        S5      eXbS S 2U4'   M     U R                  R                  X!R                   UR                  S9nU$ )Nr   rV   ztoo many dims to broadcastzcannot broadcast resultr  )r   r9   r!  
empty_likern  r{  r   rd   r:   r;   asarrayr   rr   r   rb   r  r   )	rs   targetresult_valuesresult_compareir   resaresr   s	            r>   rz  FrameApply.apply_broadcastq  s    		""""fmm4  a/FA))FKC$))Ct{{CC::c?''D ax !=>>qy!SX-$%>??"%!Q$ 0 &&v~~ ' 
 rB   c                    U R                   S:X  a  U R                  5       u  pOU R                  5       u  pU R                  X5      $ )Nrk   )r[   apply_series_generatorapply_series_numbawrap_resultsrd  s      r>   r}  FrameApply.apply_standard  sE    ;;("!%!<!<!>GY!%!8!8!:G   44rB   c                L   [        U R                  5      (       d   eU R                  nU R                  n0 n[	        U5       H]  u  pEU R                  " U/U R
                  Q70 U R                  D6X4'   [        X4   [        5      (       d  MJ  X4   R                  SS9X4'   M_     X24$ )NF)deep)
r   r9   rR  r   r   r:   r;   r   r"   r  )rs   
series_genre  r   r  r   s         r>   r  !FrameApply.apply_series_generator  s    		""""**
%%	j)DA1@tyy@DKK@GJ'*i00 %Z__%_8
 * !!rB   c                J   U R                   R                  SS5      (       a  [        S5      eU R                  R                  R
                  (       a  U R                  R
                  (       d  [        S5      eU R                  5         U R                  5       nXR                  4$ )NrW  FzAParallel apply is not supported when raw=False and engine='numba'zBThe index/columns must be unique when raw=False and engine='numba')
r\   getr`   rb   r   	is_uniquerd   ra  r[  r   r~  s     r>   r  FrameApply.apply_series_numba  s    !!*e44%S  xx~~''t||/E/E%T  	&&('')))))rB   c                   SSK Jn  [        U5      S:  a*  SU;   a$  [        US   5      (       a  U R	                  X5      $ U R
                  R                  n[        U5      S:X  a  XCL a  U" U[        R                  S9nOU" U5      nX%l	        U$ )Nr   r   r  )
r   r.   r   r   rf  rb   r  r!  r  r   )rs   r   re  r.   constructor_slicedr   s         r>   r  FrameApply.wrap_results  s    ! w<!WWQZ1H1H--gAA "XX99w<1!3!='rzzBF'0F rB   c                   > U R                   S:X  a>  U R                  nUR                  U R                     nUR	                  X R
                  S9$ [        TU ]  5       $ )Nsizer  )r9   rb   r{  rD   r  r:  rH  r   )rs   rb   valuerI  s      r>   r   FrameApply.apply_str  sR     99((CIIdii(E**5*FFw ""rB   )rD   rb   )rb   r   r9   r   rX   rJ   rY   r'  rZ   Literal[False, 'compat']r[   r   r\   r)  r*  r+  rD  r*  zGenerator[Series]TTFr*  z5Callable[[npt.NDArray, Index, Index], dict[int, Any]])r*  r+  r   r2   re  r-   r*  r-  r,  )rk   Nr  r,   r*  r,   )r*  ztuple[ResType, Index])#rK   rL   rM   rN   r2  rt   rE  rQ   rR   r   rO  rR  rP   	functoolscacherX  r[  ra  rf  ri  rd   r   rn  rE   r   ru  r|  rz  r}  r  r  r  r   rS   __classcell__rI  s   @r>   rT   rT   e  s   	N ,104

 
 	

  
 )
 
 .
 

 
<          __27	>   
 	  	+0	  # #      4%l.,#\1Mf85""**# #rB   c                      \ rS rSr% SrS\S'   \SS j5       r\\	R                   S SS jj5       5       rSS jr\SS j5       r\SS	 j5       r      SS
 jrSrg)r_   i  r   r   rD   c                T   ^  U 4S j[        [        T R                  5      5       5       $ )Nc              3  X   >#    U  H  nTR                   R                  US S9v   M!     g7f)rV   r   N)rb   r   )r  r  rs   s     r>   r  1FrameRowApply.series_generator.<locals>.<genexpr>  s$     K2JQaa(2Js   '*)ranger   rd   rx   s   `r>   rR  FrameRowApply.series_generator  s    K%DLL8I2JKKrB   c                   ^^^ [        S5      nSSKJm  SSKJm  UR
                  R                  U 5      mUR                  XUS9UUU4S j5       nU$ )Nr  r   r   maybe_cast_strrU  rV  rW  c           	        > 0 n[        U R                  S   5       H%  nT" U S S 2U4   UT	" X   5      S9nT" U/UQ76 XE'   M'     U$ )NrV   r   )r  r{  )
rn  	col_namesdf_indexr:   r   jserr.   
jitted_udfr  s
          r>   
numba_func;FrameRowApply.generate_numba_apply_func.<locals>.numba_func  sY    G6<<?+1a4L~il7S (3d3
 , NrB   r   r   r.   pandas.core._numba.extensionsr  	extendingregister_jitablejit	r9   rU  rV  rW  r  r  r.   r  r  s	         @@@r>   rX  'FrameRowApply.generate_numba_apply_func  sV    
 +73! 	A__55d;
 
H	E	 
F	 rB   c           	        [        [        U R                  5      n[        XR                  U R
                  SS9u  p#U R                  " U40 [        U R                  5      D6nSSK	J
n  U R                  R                  nU R                  R                  nU" U5       oe" U5       n[        U" U R                  Xv/UQ76 5      nS S S 5        S S S 5        W$ ! , (       d  f       N= f! , (       d  f       W$ = fNrV   r  r   )set_numba_datar   r   r9   r&   r:   r;   rX  r%   r\   r  r  rb   r   rd   r   rn  	rs   r9   r:   r;   nb_funcr  r   rd   r  s	            r>   r[  FrameRowApply.apply_with_numba  s    Hdii(1))T[[A
 00
%d&8&89
 	A(("" E"e^G-Dwt{{GBTBCC .E"
 .E-D""
s$   C%$CC%
C"	C%%
C4c                    U R                   $ rw   rd   rx   s    r>   r   FrameRowApply.result_index      ||rB   c                    U R                   $ rw   r  rx   s    r>   rO  FrameRowApply.result_columns      zzrB   c                   U R                   S:X  a#  U R                  R                  U5      nX#l        U$ U R                   cH  [	        S UR                  5        5       5      (       a#  U R                  R                  U5      nX#l        U$  U R                  R                  US9n[        US   [        5      (       d=  [        UR                  5      [        U R                  5      :X  a  U R                  Ul        [        UR                  5      [        U5      :X  a  X$l        U$ ! [         a=  nS[        U5      ;   a(  U R                  R                  U5      nX#l        Us SnA$ e SnAff = f)zreturn the results for the rowsro   Nc              3  B   #    U  H  n[        U[        5      v   M     g 7frw   )r   r   )r  r  s     r>   r  6FrameRowApply.wrap_results_for_axis.<locals>.<genexpr>  s      .
)9AJq$)9s   r8   z%All arrays must be of the same lengthr   )rY   rb   r  r   r   rn  r  rr   r   r   r"   r   ri  rd   )rs   r   re  r  r   r   s         r>   rf  #FrameRowApply.wrap_results_for_axis  s4   
 x'((..w7C!IJ%# .
)0)9.
 +
 +

 ((..w7C!IJ
	XX***8F '!*i006<< C(8(8$99#//v~~#i.0&N#  	6#c(B hh227;%	
	s$   
D" "
E),1E$E)#E$$E)r7   Nr  r  r  r*  zdict[int, Any]rD  r  )rK   rL   rM   rN   rD   r2  rE  rR  rP   r  r  rX  r[  r   rO  rf  rS   r7   rB   r>   r_   r_     s    D'L L __27	>  6&    ''+0'	'rB   r_   c                     ^  \ rS rSr% SrS\S'   SU 4S jjr\SS j5       r\	\
R                   S SS jj5       5       rSS jr\SS	 j5       r\SS
 j5       r      SS jrSS jrSrU =r$ )ra   i=  rV   r   rD   c                N   > [         TU ]  UR                  5      nUR                  $ rw   )rH  rz  r   )rs   r  r   rI  s      r>   rz   FrameColumnApply.apply_broadcast@  s    (2xxrB   c              #    #    U R                   n[        U5      n[        U5      S:  d   eU R                  R	                  SSS9nUR
                  nUR                  S   R                  R                  5       n[        UR                  [        5      (       a:  U R                  n[        [        U5      5       H  nUR	                  USS9v   M     g [        XR                  SS9 Hh  u  pxX2l        UR                  U5        [         R#                  USU5        U(       d*  [%        UR                  S   5      UR                  S   l        Uv   Mj     g 7f)Nr   r   Tr   _name)rn  r$   r   rb   r   rx  blocksrefshas_referencer   r`  r   r  r   r   
set_valuesr  __setattr__r   )	rs   rn  r  mgris_viewrb   r  arrr   s	            r>   rR  !FrameColumnApply.series_generatorD  s
    /76{Q hhmmAAm&hh**Q-$$224cii00 ((C3s8_hhqqh)) % !DA	s#""36 *9A)GCJJqM&	 Bs   EEc                   ^^^ [        S5      nSSKJm  SSKJm  UR
                  R                  U 5      mUR                  XUS9UUU4S j5       nU$ )Nr  r   r   r  r  c           	        > 0 n[        U R                  S   5       H.  nT" X   R                  5       UT	" X%   5      S9nT" U/UQ76 XE'   M0     U$ )Nr   r   )r  r{  r  )
rn  col_names_indexr   r:   r   r  r  r.   r  r  s
          r>   r  >FrameColumnApply.generate_numba_apply_func.<locals>.numba_funcs  s`    G 6<<?+ INN$)'1
 (3d3
 , NrB   r  r  s	         @@@r>   rX  *FrameColumnApply.generate_numba_apply_funch  sQ    
 +73!@__55d;
	H	E	 
F	  rB   c           	        [        [        U R                  5      n[        XR                  U R
                  SS9u  p#U R                  " U40 [        U R                  5      D6nSSK	J
n  U" U R                  R                  5       nU" U R                  5       n[        U" U R                  Xv/UQ76 5      nS S S 5        S S S 5        W$ ! , (       d  f       N= f! , (       d  f       W$ = fr  r  r  s	            r>   r[  !FrameColumnApply.apply_with_numba  s    Hdii(1))T[[A
 00
%d&8&89
 	A
 488>>*e4<<(Gwt{{GBTBCC ) +
 
	 )( +*
 
s$   CC5C
C	C
C'c                    U R                   $ rw   r  rx   s    r>   r   FrameColumnApply.result_index  r  rB   c                    U R                   $ rw   r  rx   s    r>   rO  FrameColumnApply.result_columns  r  rB   c                    U R                   S:X  a  U R                  X5      nU$ [        US   [        5      (       d#  U R                  R                  U5      nX#l        U$ U R                  X5      nU$ )z"return the results for the columnsrq   r   )rY   infer_to_same_shaper   r"   rb   r  r   rs   r   re  r   s       r>   rf  &FrameColumnApply.wrap_results_for_axis  ss     x'--gAF  GAJ	22XX11':F$L  --gAFrB   c                |    U R                   R                  US9nUR                  nX#l        UR	                  5       nU$ )z7infer the results to the same shape as the input objectr  )rb   r  r   r   infer_objectsr  s       r>   r  $FrameColumnApply.infer_to_same_shape  s>    &&G&4 ! %%'rB   r7   r  r  r  r  r  rD  r  )r   r2   re  r-   r*  r,   )rK   rL   rM   rN   rD   r2  rz  rE  rR  rP   r  r  rX  r[  r   rO  rf  r  rS   r  r  s   @r>   ra   ra   =  s    D' ! !F __27	>  8*    +0	* rB   ra   c                     ^  \ rS rSr% S\S'   SrS\S'   S\S'   S	S
.       SU 4S jjjrSS jrU 4S jrSS jr	S r
SS jrSrU =r$ )r>  i  r.   rb   r   r   rD   r(  rZ   rj   rZ   c          
     ,   > [         TU ]  UUSS UUUS9  g )NF)rX   rY   rZ   r:   r;   )rH  rt   )rs   rb   r9   rZ   r:   r;   rI  s         r>   rt   SeriesApply.__init__  s,     	 	 	
rB   c                j   U R                   n[        U5      S:X  a  U R                  5       $ [        U R                  5      (       a  U R                  5       $ [        U R                  [        5      (       a  U R                  5       $ U R                  S:X  a  U R                  5       $ U R                  5       $ )Nr   rn   )rb   r   ru  r   r9   r  r   r   r   rZ   apply_compatr}  rs   rb   s     r>   rE   SeriesApply.apply  s    hhs8q=**,, 		""//11dii%%>>##;;)#$$&& ""$$rB   c                   > [         TU ]  5       nUcJ  U R                  nU R                  n[	        U5      (       d   eU" U/U R
                  Q70 U R                  D6nU$ rw   )rH  r   rb   r9   r   r:   r;   )rs   r   rb   r9   rI  s       r>   r   SeriesApply.agg  sT    >((C99DD>>!>#9		9T[[9FrB   c                ~    U R                   nUR                  UR                  UR                  S9R	                  USS9$ )N)r`  r   rE   r  )rb   r  r`  r   __finalize__r  s     r>   ru  SeriesApply.apply_empty_result  sA    hhciisyyANN O 
 	
rB   c                n   U R                   nU R                  n[        U5      (       aO  [        R                  " U5      nU(       a2  U R
                  (       d!  U R                  (       d  UR                  USS9$  UR                  USS9nU$ ! [        [        [        4 a    UR                  USS9n U$ f = f)zcompat apply method for funcs in listlikes and dictlikes.

 Used for each callable when giving listlikes and dictlikes of callables to
 apply. Needed for compatibility with Pandas < v2.1.

.. versionadded:: 2.1.0
Fr  rj   )rb   r9   r   r   get_cython_funcr:   r;   rE   rr   r#  r   )rs   rb   r9   r$  r   s        r>   r  SeriesApply.apply_compat  s     hhyyD>>##D)A4;;yyey44	3YYtHY5F  NI6 	3YYtEY2F	3s   9B %B43B4c                  ^ ^ [        [        T R                  5      mT R                  n[	        T[
        R                  5      (       a>  [
        R                  " SS9   T" U/T R                  Q70 T R                  D6sS S S 5        $ T R                  (       d   T" U/T R                  Q70 T R                  D6$ T R                  (       d  T R                  (       a  UU 4S jnOTnUR                  US9n[        U5      (       a;  [	        US   [        5      (       a#  UR                  [        U5      UR                   S9$ UR#                  X1R                   S9R%                  USS9$ ! , (       d  f       N= f)	Nrq  rr  c                D   > T" U /TR                   Q70 TR                  D6$ rw   )r:   r;   )r  r9   rs   s    r>   curried+SeriesApply.apply_standard.<locals>.curried%  s     A9		9T[[99rB   )mapperr   r  rE   r  )r   r   r9   rb   r   r!  rv  rw  r:   r;   rZ   _map_valuesr   r"   _constructor_expanddimr   r   r  r  )rs   rb   r$  mappedr9   s   `   @r>   r}  SeriesApply.apply_standard  s   Hdii(hhdBHH%%*C;$));t{{; +*7dii74;;7799: G0v;;:fQi;; --d6l#))-LL##F))#<IIG J  ' +*s    E$$
E2r7   )rb   r.   r9   r   rZ   r(  r*  r+  r,  )r*  r.   )rK   rL   rM   rN   r2  rD   rt   rE   r   ru  r  r}  rS   r  r  s   @r>   r>  r>    sk    	KD'// 7?

 

 4
 

 
&%(
, rB   r>  c                  t   ^  \ rS rSr% S\S'         S
U 4S jjrS rS r    SS jr    SS jr	S	r
U =r$ )GroupByApplyi6  z GroupBy | Resampler | BaseWindowrb   c          	        > UR                  5       nUR                  R                  UR                  SS5      5      U l        [
        TU ]  UUSS UUS9  g )NrD   r   FrX   rY   r:   r;   )r  rb   r^   r  rD   rH  rt   rs   rb   r9   r:   r;   rI  s        r>   rt   GroupByApply.__init__9  sU     GG,,VZZ-BC	 	 	
rB   c                    [         erw   r`   rx   s    r>   rE   GroupByApply.applyL      !!rB   c                    [         erw   r2  rx   s    r>   r   GroupByApply.transformO  r4  rB   c           	        U R                   nU R                  nUS:X  a  0 UESS0En[        USS5      S:X  a  [        S5      eUR                  R
                  S:X  a  UR                  nOUR                  n[        R                  " USS	[        US5      S
9   U R                  XU5      u  pVS S S 5        U R                  WW5      nU$ ! , (       d  f       N"= f)NrE   rZ   FrD   r   rV   r=  as_indexT	condition)rb   r;   r   r`   _selected_objr   _obj_with_exclusionsr   temp_setattrr   r   r   )rs   r}   rb   r;   r   r   r   r   s           r>   r~   #GroupByApply.agg_or_apply_list_likeR  s     hhg00%0F3"a'%&JKK!!Q&,,L33L TWS*-E
 !227&QMD
 ,,T7;
 
s   C
Cc           	        SSK JnJn  US;   d   eU R                  n0 nUS:X  a(  U R                  (       a  SOSnUR                  SU05        [        USS5      S	:X  a  [        S
5      eUR                  nUR                  n[        XBU45      n	U	(       aK  U R                  R                  SS 5      n
U R                  R                  SS 5      nUR                  XS.5        [        R                  " USS[        US5      S9   U R!                  XX5      u  pS S S 5        U R#                  UWW5      nU$ ! , (       d  f       N#= f)Nr   r   rA  rE   rn   FrZ   rD   rV   r=  r[   r\   rt  r8  Tr9  )r   r   r   rb   rZ   rB  r   r`   r;  
_selectionr   r;   r  r   r=  r   r   r   )rs   r}   r   r   rb   r;   rZ   r   r   r   r[   r\   r   r   r   s                  r>   r   #GroupByApply.agg_or_apply_dict_likel  s+   	

 ****hh!#g"&++Y5FMM8V,-3"a'%&JKK((NN	%FG
 [[__Xt4F KKOOOTBMMMVLMTWS*-E
 )-(>(>y)%L
 ,,\<U
 
s   D33
Er   )rb   zGroupBy[NDFrameT]r9   r   r*  r+  r.  )rK   rL   rM   rN   r2  rt   rE   r   r~   r   rS   r  r  s   @r>   r,  r,  6  sc    	))

 
 

&"".	4%.%	% %rB   r,  c                  ^   ^  \ rS rSr% SrS\S'   S\S'         SU 4S jjrS rS	 rS
r	U =r
$ )ResamplerWindowApplyi  r   r   rD   Resampler | BaseWindowrb   c          	     2   > [         [        U ]  UUSS UUS9  g )NFr.  )rH  r,  rt   r/  s        r>   rt   ResamplerWindowApply.__init__  s+     	lD* 	+ 	
rB   c                    [         erw   r2  rx   s    r>   rE   ResamplerWindowApply.apply  r4  rB   c                    [         erw   r2  rx   s    r>   r   ResamplerWindowApply.transform  r4  rB   r7   )rb   rD  r9   r   r*  r+  )rK   rL   rM   rN   rD   r2  rt   rE   r   rS   r  r  s   @r>   rC  rC    sC    D'	
#
 
 

""" "rB   rC  c                  ^	 SSK Jm	  U SL =(       a5    [        S0 UD6=(       d#    [        U	4S jUR	                  5        5       5      nSnSnU(       dO  [        U [        5      (       a,  [        U 5      [        [        U 5      5      :  a  [        S5      eU c  [        S5      eU(       a  0 nUR                  5        Hu  u  pg[        UT	5      (       a[  UR                  nUR                  (       d  UR                  (       a  XR                  UR                  4S jnUR                  U4XV'   Mq  XuU'   Mw     [!        U5      u  pnU c   eX X44$ )	aR  
This is the internal function to reconstruct func given if there is relabeling
or not and also normalize the keyword to get new order of columns.

If named aggregation is applied, `func` will be None, and kwargs contains the
column and aggregation function information to be parsed;
If named aggregation is not applied, `func` is either string (e.g. 'min') or
Callable, or list of them (e.g. ['min', np.max]), or the dictionary of column name
and str/Callable/list of them (e.g. {'A': 'min'}, or {'A': [np.min, lambda x: x]})

If relabeling is True, will return relabeling, reconstructed func, column
names, and the reconstructed order of columns.
If relabeling is False, the columns and order will be None.

Parameters
----------
func: agg function (e.g. 'min' or Callable) or list of agg functions
    (e.g. ['min', np.max]) or dictionary (e.g. {'A': ['min', np.max]}).
**kwargs: dict, kwargs used in is_multi_agg_with_relabel and
    normalize_keyword_aggregation function for relabelling

Returns
-------
relabelling: bool, if there is relabelling or not
func: normalized and mangled func
columns: tuple of column names
order: array of columns indices

Examples
--------
>>> reconstruct_func(None, **{"foo": ("col", "min")})
(True, defaultdict(<class 'list'>, {'col': ['min']}), ('foo',), array([0]))

>>> reconstruct_func("min")
(False, 'min', None, None)
r   )NamedAggNc              3  <   >#    U  H  n[        UT5      v   M     g 7frw   r  )r  r   rL  s     r>   r  #reconstruct_func.<locals>.<genexpr>  s     @1z!X&&s   zFFunction names must be unique if there is no new column names assignedz4Must provide 'func' or tuples of '(column, aggfunc).c                    U" U /UQ70 UD6$ rw   r7   )r  r9   r   kws       r>   <lambda>"reconstruct_func.<locals>.<lambda>  s     QURR!#RrB   r7   )r   rL  is_multi_agg_with_relabelr   rn  r   r   r   setr   r   r   aggfuncr:   r;   columnnormalize_keyword_aggregation)
r9   r;   
relabelingrd   orderconverted_kwargsr   valrU  rL  s
            @r>   r]   r]     s0   N 5 !+F+ 	A@@@ 
 '+G)-EdD!!c$i#c$i.&@ %X  <RSS HC#x((++88szz-4SZZ G *-W(= %(+% '  = 
u W++rB   c                 l    [        S U R                  5        5       5      =(       a    [        U 5      S:  $ )a8  
Check whether kwargs passed to .agg look like multi-agg with relabeling.

Parameters
----------
**kwargs : dict

Returns
-------
bool

Examples
--------
>>> is_multi_agg_with_relabel(a="max")
False
>>> is_multi_agg_with_relabel(a_max=("a", "max"), a_min=("a", "min"))
True
>>> is_multi_agg_with_relabel()
False
c              3  l   #    U  H*  n[        U[        5      =(       a    [        U5      S :H  v   M,     g7f)r   N)r   rH   r   )r  r   s     r>   r  ,is_multi_agg_with_relabel.<locals>.<genexpr>  s'     M_z!U#3A!3_s   24r   )r   rn  r   )r;   s    r>   rS  rS    s-    * MV]]_MM FarB   c           
     (   SSK Jn  [        [        5      n/ n[	        U R                  5       5      nU R                  5        HH  u  pVX%   R                  U5        UR                  U[        R                  " U5      =(       d    U45        MJ     [        U5      nUR                  5        VVVs/ s H0  u  pXU  H$  nU[        R                  " U5      =(       d    U4PM&     M2     n	nnn[        U	5      n
U" U
5      R                  U5      nX$U4$ s  snnnf )a  
Normalize user-provided "named aggregation" kwargs.
Transforms from the new ``Mapping[str, NamedAgg]`` style kwargs
to the old Dict[str, List[scalar]]].

Parameters
----------
kwargs : dict

Returns
-------
aggspec : dict
    The transformed kwargs.
columns : tuple[str, ...]
    The user-provided keys.
col_idx_order : List[int]
    List of columns indices.

Examples
--------
>>> normalize_keyword_aggregation({"output": ("input", "sum")})
(defaultdict(<class 'list'>, {'input': ['sum']}), ('output',), array([0]))
r   r   )pandas.core.indexes.baser-   r   r   rH   r   rn  r   r   r   _make_unique_kwarg_listr   get_indexer)r;   r-   aggspecrY  rd   rV  rU  uniquified_orderaggfuncsaggspec_orderuniquified_aggspeccol_idx_orders               r>   rW  rW  !  s    < /
 $GEFKKM"G!==?w'fc33G<GHI +
 /u5 !( /FG 
&&w/:7; 	< /  
 1? ,-99:JKM]**s   ,7Dc           
         [        U 5       VVs/ s H=  u  pU R                  U5      S:  a!  US   US    SU SU R                  U5       34OUPM?     snn$ s  snnf )z
Uniquify aggfunc name of the pairs in the order list

Examples:
--------
>>> kwarg_list = [("a", "<lambda>"), ("a", "<lambda>"), ("b", "<lambda>")]
>>> _make_unique_kwarg_list(kwarg_list)
[('a', '<lambda>_0'), ('a', '<lambda>_1'), ('b', '<lambda>')]
rV   r   rc   N)r   count)seqr  pairs      r>   ra  ra  ^  sl     !~%GA :=419LaT!WIQs2Aw}}T2345RVV%  s   AAc                   SSK Jn  [        [        X#SS9S S9 Vs/ s H  oUS   PM	     nn0 nSn[	        U [
        5      (       + =(       a    [        U R                  5      S:  n	UR                  5        H  u  pX
   R                  5       nU	(       a  U Vs/ s H0  n[	        U[        5      (       d  [        R                  " U5      OUPM2     nnU" UR                  SS	9R                  U5      nUS
:g  nUR                  5       (       a  UR                   X      nUR"                  (       d  XhU[        U5      -    Ul        UR%                  U5      Xz'   U[        U5      -   nM     U$ s  snf s  snf )aS  
Internal function to reorder result if relabelling is True for
dataframe.agg, and return the reordered result in dict.

Parameters:
----------
result: Result from aggregation
func: Dict of (column name, funcs)
columns: New columns name for relabelling
order: New order for relabelling

Examples
--------
>>> from pandas.core.apply import relabel_result
>>> result = pd.DataFrame(
...     {"A": [np.nan, 2, np.nan], "C": [6, np.nan, np.nan], "B": [np.nan, 4, 2.5]},
...     index=["max", "mean", "min"],
... )
>>> funcs = {"A": ["max"], "C": ["max"], "B": ["mean", "min"]}
>>> columns = ("foo", "aab", "bar", "dat")
>>> order = [0, 1, 2, 3]
>>> result_in_dict = relabel_result(result, funcs, columns, order)
>>> pd.DataFrame(result_in_dict, index=columns)
       A    C    B
foo  2.0  NaN  NaN
aab  NaN  6.0  NaN
bar  NaN  NaN  4.0
dat  NaN  NaN  2.5
r   r   Tr   c                    U S   $ )NrV   r7   )ts    r>   rQ   relabel_result.<locals>.<lambda>  s    STUVSWrB   )r   rV   F)r  )r`  r-   sortedr   r   r"   r   rd   r   dropnar   r   r   r   rb  r   r   r   reindex)r   r9   rd   rY  r-   rl  reordered_indexesreordered_result_in_dictidxreorder_maskr   funsr$  rh  	valid_idxs                   r>   relabel_resultr|  p  sZ   F / #3wd#CXXDQX   8:
C!&)44PV^^9Lq9PLJJLK , SVSVa
1c0B0B%%a(ISV   "!''6BB3GM%+I}}FF=34 ww'cCHn=AG()		'(: %CHnI !J $#Y>s   E'7E,c                v    SSK Jn  [        U40 UD6u  pApVU(       a  Uc   eUc   e[        XXV5      nU" XuS9n U $ )Nr   )r,   r  )r   r,   r]   r|  )r   r9   r;   r,   rX  rd   rY  result_in_dicts           r>   r  r    sQ     '7'G'G$Jg """   'gE>9MrB   c                    [        U 5      S::  a  U $ Sn/ nU  HI  n[        R                  " U5      S:X  a  [        U5      nSU S3Ul        US-  nUR                  U5        MK     U$ )a  
Possibly mangle a list of aggfuncs.

Parameters
----------
aggfuncs : Sequence

Returns
-------
mangled: list-like
    A new AggSpec sequence, where lambdas have been converted
    to have unique names.

Notes
-----
If just one aggfunc is passed, the name will not be mangled.
rV   r   z<lambda>z<lambda_>)r   r   r   r   rK   r   )re  r  mangled_aggfuncsrU  s       r>   _managle_lambda_listr    sv    $ 8}	A  )Z7g&G!)!AGFA(  rB   c                ,   [        U 5      nU(       d  [        U 5      (       d  U $ [        U 5      " 5       nU(       aM  U R                  5        H7  u  p4[        U5      (       a  [        U5      (       d  [	        U5      nOUnXRU'   M9     U$ [	        U 5      nU$ )a
  
Make new lambdas with unique names.

Parameters
----------
agg_spec : Any
    An argument to GroupBy.agg.
    Non-dict-like `agg_spec` are pass through as is.
    For dict-like `agg_spec` a new spec is returned
    with name-mangled lambdas.

Returns
-------
mangled : Any
    Same type as the input.

Examples
--------
>>> maybe_mangle_lambdas("sum")
'sum'
>>> maybe_mangle_lambdas([lambda: 1, lambda: 2])  # doctest: +SKIP
[<function __main__.<lambda_0>,
 <function pandas...._make_lambda.<locals>.f(*args, **kwargs)>]
)r   r   r"  r   r  )agg_specis_dictmangled_aggspecr   re  r  s         r>   maybe_mangle_lambdasr    s    2 8$G|H--8n&O%^^-MCH%%l8.D.D#7#A #+ #3C  .  /x8rB   c                B   Sn[        U 5      n/ nU R                  5        Hf  n[        U[        5      (       d=  [	        U5      (       d-  [        UR                  [        U5      R                  5      5      eUR                  U5        Mh     U(       d  Sn[        U5      eX#4$ )a  
Validates types of user-provided "named aggregation" kwargs.
`TypeError` is raised if aggfunc is not `str` or callable.

Parameters
----------
kwargs : dict

Returns
-------
columns : List[str]
    List of user-provided keys.
func : List[Union[str, callable[...,Any]]]
    List of user-provided aggfuncs

Examples
--------
>>> validate_func_kwargs({"one": "min", "two": "max"})
(['one', 'two'], ['min', 'max'])
z-func is expected but received {} in **kwargs.z2Must provide 'func' or named aggregation **kwargs.)
r   rn  r   r   r   r   formatr"  rK   r   )r;   tuple_given_messagerd   r9   col_funcno_arg_messages         r>   validate_func_kwargsr  -  s    . J6lGDMMO8S))Xh-?-?/66tH~7N7NOPPH $ M''=rB   c                h    [        U[        5      =(       d    [        U[        5      =(       a    U S:H  $ )Nr   )r   r    r"   )r}   r   s     r>   r   r   Q  s*    dL) 4#85(8rB   )r   FNrj   rk   NNN)rb   r,   r9   r   rD   r   rX   rJ   rY   r'  rZ   r  r[   r   r\   r)  r*  rT   )r9   zAggFuncType | Noner*  zMtuple[bool, AggFuncType, tuple[str, ...] | None, npt.NDArray[np.intp] | None])r*  rJ   )r;   r   r*  z]tuple[MutableMapping[Hashable, list[AggFuncTypeBase]], tuple[str, ...], npt.NDArray[np.intp]])rk  Sequence[tuple[Any, Any]]r*  r  )
r   r-  r9   zdict[str, list[Callable | str]]rd   r1  rY  zIterable[int]r*  zdict[Hashable, Series]r,  )re  Sequence[Any]r*  r  )r  r   r*  r   )r;   r   r*  z0tuple[list[str], list[str | Callable[..., Any]]])r}   r/  r   r0  r*  rJ   )f
__future__r   rQ   collectionsr   collections.abcr   r  r   r   typingr   r   r	   r
   r   numpyr!  pandas._libs.internalsr   pandas._typingr   r   r   r   r   r   r   r   pandas.compat._optionalr   pandas.errorsr   pandas.util._decoratorsr   r   pandas.core.dtypes.castr   pandas.core.dtypes.commonr   r   r   r   r   pandas.core.dtypes.dtypesr   pandas.core.dtypes.genericr    r!   r"   pandas.core._numba.executorr#   pandas.core.commoncorecommonr   pandas.core.constructionr$   pandas.core.util.numba_r%   r&   r'   r(   r)   r*   r+   r   r,   r-   r.   pandas.core.groupbyr/   pandas.core.resampler0   pandas.core.window.rollingr1   r   intr2   r2  ABCr4   rf   ABCMetarh   r5  rT   r_   ra   r>  r,  rC  r]   rS  rW  ra  r|  r  r  r  r  r   r7   rB   r>   <module>r     sv   " 
 # $      2	 	 	 ? ,
 5  5  >     C
   
 ,.5#s(^ # "#e#'' e $eV "'/,0	$	$
$ $ 
	$
 $ %$ $ *$ $ND	&ckk D	&N:5 :zl# l#^fJ fRFz FRm, m`[5 [|"< "8T,
T,RT,n4:+:+:+z	"$Q$Q$
)Q$  Q$ 	Q$
 Q$h6D)X!!5!HrB   