E gO&UdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZerddlZerdd lmZdad ed <gad ed <eadaded<daded<hdhddZdZd:d;dZddZ dZ!iej"dej#dej$d ej%d ej&d!ej'd!ej(d"ej)d"ej*dej+dej,dej-dej.d#ej/d#ej0d$ejd%ej1d&ej2d'ej3d(ej4d)ej5d*ej6d*ej7d+ej8d+ej9d,ej:d,e;dej<di Z=d-Z>d.Z?ee d/d0Z@d+d*d,d1ZAd>d2ZBd:d?d4ZCd:d?d5ZDd:d;d6ZEd@d8ZFdAd9ZGdS)BzL Expressions ----------- Offer fast expression evaluation through numexpr ) annotationsN) TYPE_CHECKING) get_option)find_stack_level) roperator)NUMEXPR_INSTALLED)FuncTypez bool | None _TEST_MODE list[bool] _TEST_RESULTzFuncType | None _evaluate_where>boolint32int64float32float64>rrr)evaluatewherei@BTvrreturnNonecltr|atrtntatrt nt adSN)r USE_NUMEXPR_evaluate_numexpr_evaluate_standardr _where_numexpr_where_standardrrs X/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/core/computation/expressions.pyset_use_numexprr"-s6 &1H!!6HI* ?^^FFFc|tr2tr-|tj}tj|dSdSdSr)rrnedetect_number_of_coresset_num_threads)ns r!set_numexpr_threadsr):sQ[ 9)++A 1r#cFtrtd|||S)z Standard evaluation. F)r _store_test_result)opop_strabs r!rrCs)"5!!! 2a88Or#c|g|jtkrWt}||fD]"}t|dr||jjhz}#t |rt||krdSdS)z,return a boolean if we WILL be using numexprNdtypeTF)size _MIN_ELEMENTSsethasattrr1namelen_ALLOWED_DTYPES)r,r-r.r/ dtype_checkdtypesos r!_can_use_numexprr<Ls}  6M ! !"uuFV - -1g&&-qw|n,Fv;; /+">&"H"Ht 5r#cd}t||||dr|jdd}|r||}}|}|} t jd|d||dd}n.#t $rYn"t$rt|||rnYnwxYw|r||}}trt|du|t||||}|S) Nr_rza_value z b_value)a_valueb_valuesafe local_dictcasting) r<__name__strip startswithr%r TypeErrorNotImplementedError_bool_arith_fallbackr r+r)r,r-r.r/result is_reversedr@rAs r!rr_s8 FFAq*55k'',,77<<  aqA [+6+++'.7CCFF     D"   #FAq11    aqA/6-... ~#B155 MsA-- B9BB+*-/z**z==z!=z<==>&|^c.tj|||Sr)npr)condr.r/s r!rrs 8D!Q  r#cd}tdd||drtjd|||dd}|t|||}|S)Nrz#where(cond_value, a_value, b_value)) cond_valuer@rArBrC)r<r%rr)rYr.r/rLs r!rrse Fgq!W55  1&*qQGG    ~ q!,, Mr#zcompute.use_numexprc |jtkS#t$r$t|ttjfcYSwxYwr)r1rAttributeError isinstancerXbool_)xs r!_has_bool_dtyperasK/w$ ///!dBH-...../s+AA)rNrOrPc t|rht|rY|tvrPtjdt |dt t|dt dSdS)z Check if we should fallback to the python `_evaluate_standard` in case of an unsupported operation by numexpr, which is the case for some boolean ops. z'evaluating in Python space because the z> operator is not supported by numexpr for the bool dtype, use z instead.) stacklevelTF)ra_BOOL_OP_UNSUPPORTEDwarningswarnreprr)r-r.r/s r!rKrKs qoa00 ) ) ) ME$v,,EE0899EEE,--      4 5r# use_numexprclt|}||rt||||St||||S)z Evaluate and return the expression of the op on a and b. Parameters ---------- op : the actual operand a : left operand b : right operand use_numexpr : bool, default True Whether to try to use numexpr. )_op_str_mappingr r)r,r.r/rhr-s r!rrsDR F   /RA.. . b&!Q / //r#c\tJ|rt|||nt|||S)z Evaluate the where condition cond on a and b. Parameters ---------- cond : np.ndarray[bool] a : return if cond is True b : return if cond is False use_numexpr : bool, default True Whether to try to use numexpr. )rr)rYr.r/rhs r!rrs8    !, M6$1   /$12M2MMr#c|agadS)z Keeps track of whether numexpr was used. Stores an additional ``True`` for every successful use of evaluate with numexpr since the last ``get_test_result``. N)r r r s r! set_test_modermsJLLLr# used_numexprcB|rt|dSdSr)r append)rns r!r+r+s.*L)))))**r#ct}ga|S)z1 Get test result and reset test_results. )r )ress r!get_test_resultrss CL Jr#)T)rrrrr)rr)rr)rhr)rnrrr)rr )H__doc__ __future__roperatortypingrrenumpyrXpandas._configrpandas.util._exceptionsr pandas.corerpandas.core.computation.checkrnumexprr%pandas._typingr r __annotations__r rr rr8r3r"r)rr<raddraddmulrmulsubrsubtruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqleltgegtand_rand_or_ror_xorrxordivmodrdivmodrjrrrardrKrrrmr+rsr#r!rs#""""" %%%%%%444444!!!!!!;;;;;;(''''''   ! !!!!A@@ ) ) )   @ @ @ @ @&&&&R! L#! NC! L#!NC !  L# ! NC ! c!! t!! L$! ND!!" L$#!$ND%!& K'!( K)!* K+!, K K K M3 OS L# NC L# NC D tA!!H   $ 011222///!s55$00000( N N N N N     **** r#