ehO&UdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZerddlZerdd lmZdad ed <gad ed <eadaded<daded<hdhddZdZd3d4dZd5d6dZdZd7dZ dZ!iejDdejFdejHdejJdejLdejNdejPdejRdejTdejVdejXdejZdej\dej^dej`dej$d ejbd!ejdd"ejfd#ejhd$ejjd%ejld%ejnd&ejpd&ejrd'ejtd'e;dejxdi Z=d(Z>d)Z?ee d*d+Z@d&d%d'd,ZAd7d-ZBd3d8d.ZCd3d8d/ZDd3d4d0ZEd9d1ZFd:d2ZGy);zL 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@Bc^tr|atrtntatrt ayt ayN)r USE_NUMEXPR_evaluate_numexpr_evaluate_standardr _where_numexpr_where_standardrvs a/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/computation/expressions.pyset_use_numexprr -s$ &1!6HI*^FFcvtr3tr,|tj}tj|yyyr)rrnedetect_number_of_coresset_num_threads)ns rset_numexpr_threadsr':s2[ 9))+A 1)r!c6tr td|||S)z Standard evaluation. F)r _store_test_result)opop_strabs rrrCs5! a8Or!c|e|jtkDrRt}||fD])}t|ds||jj hz}+t |r t||k\ryy)z,return a boolean if we WILL be using numexprdtypeTF)size _MIN_ELEMENTSsethasattrr/namelen_ALLOWED_DTYPES)r*r+r,r- dtype_checkdtypesos r_can_use_numexprr:Lsg  66M !"uFV -1g&qww||n,F - v;/+">&"H r!crd}t||||drZ|jjdjd}|r||}}|}|} t j d|d||dd}|r||}}tr t|du|t||||}|S#t $rY6t$rt|||rnYOwxYw) 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>r?s rrr_s FFAq*5kk'',77< aqA [[6((+'.7CF aqA6-. ~#B15 M)  " #FAq1 sB B6B65B6+*-/z**z==z!=z<==>&|^c0tj|||Sr)npr)condr,r-s rrrs 88D!Q r!c~d}tdd||drtjd|||dd}| t|||}|S)Nrz#where(cond_value, a_value, b_value)) cond_valuer>r?r@rA)r:r#rr)rWr,r-rJs rrrsO Fgq!W5 1&*qQG  ~ q!, Mr!zcompute.use_numexprc |jtk(S#t$r#t|ttj fcYSwxYwr)r/rAttributeError isinstancerVbool_)xs r_has_bool_dtyper_s:/ww$ /!dBHH-../s)AA)rLrMrNc t|rSt|rH|tvr@tjdt |dt t|dt yy)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)r__BOOL_OP_UNSUPPORTEDwarningswarnreprr)r+r,r-s rrIrIsb qoa0 ) ) MM9$v,H089:)E,-    r!cTt|}||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- use_numexprr+s rrrs9R F  RA. . b&!Q //r!cJtJ|r t|||St|||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)rWr,r-rhs rrrs/   !,6$1 M/$12MMr!c |agay)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 rs r set_test_moderksJLr!c4|rtj|yyr)r append) used_numexprs rr)r)sL)r!ct}ga|S)z1 Get test result and reset test_results. )r )ress rget_test_resultrqs CL Jr!)T)rrreturnNoner)rrrs)rrr)rhr)rnrrrrs)rrr )H__doc__ __future__roperatortypingrrcnumpyrVpandas._configrpandas.util._exceptionsr pandas.corerpandas.core.computation.checkrnumexprr#pandas._typingr r __annotations__r rr rr6r1r r'rr:raddraddmulrmulsubrsubtruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqleltgegtand_rand_or_ror_xorrxordivmodrdivmodrgrrr_rbrIrrrkr)rqr!rrsd# %4!;' K j ! ?!A )   @&&R! LL#! NNC! LL#!NNC !  LL# ! NNC ! c!! t!! LL$! NND!!" LL$#!$NND%!& KK'!( KK)!* KK+!, KK KK KK MM3 OOS LL# NNC LL# NNC D tA!H $ 012/!s5$0( N * r!