]j tUdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZerddlZeej.ed k\Zerdd lmZdad ed<gaded<eadaded<da ded<hdhddZ!dZ"d6d7dZ#d8d9dZ$dZ%d:dZ&dZ'iejPdejRdejTdejVdejXdejZdej\d ej^d ej`dejbdejddejfdejhd!ejjd!ejld"ej,d#ejnd$ejpd%ejrd&ejtd'ejvd(ejxd(ejzd)ej|d)ej~d*ejd*eAdejdi ZCd+ZDd,ZEe#e d-d.ZFd)d(d*d/ZGd:d0ZHd6d;d1ZId6d;d2ZJd6d7d3ZKdzL Expressions ----------- Offer fast expression evaluation through numexpr ) annotationsN) TYPE_CHECKING) get_option)np_version_gt2_3)find_stack_level) roperator)NUMEXPR_INSTALLED)Versionz2.11.0)FuncTypez bool | None _TEST_MODE list[bool] _TEST_RESULTzFuncType | None _evaluate_where>boolint32int64float32float64>rrr)evaluatewherei@Bc|trtr tsdan|atrtnt atrta yta y)NF) r r ne_gt_211 USE_NUMEXPR_evaluate_numexpr_evaluate_standardr_where_numexpr_where_standardrvs M/root/env/lib/python3.12/site-packages/pandas/core/computation/expressions.pyset_use_numexprr"2s0 I KK &1!6HI*^FFcvtr3tr,|tj}tj|yyyN)r rnedetect_number_of_coresset_num_threads)ns r!set_numexpr_threadsr*Ds2[ 9))+A 1)r#c6tr td|||S)z Standard evaluation. F)r _store_test_result)opop_strleft_opright_ops r!rrMs5! gx  r#c|e|jtkDrRt}||fD])}t|ds||jj hz}+t |r t||k\ryy)z2return left_op boolean if we WILL be using numexprdtypeTF)size _MIN_ELEMENTSsethasattrr2namelen_ALLOWED_DTYPES)r-r.r/r0 dtype_checkdtypesos r!_can_use_numexprr=Vsh  <<- '"uFx( -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_rz left_value z right_value) left_value right_valuesafe local_dictcasting) r=__name__strip startswithr&r TypeErrorNotImplementedError_bool_arith_fallbackr r,r)r-r.r/r0result is_reversedrArBs r!rris FFGXzBkk'',77<  ('XG   [[fX\2*4[QF  ('XG6-. ~#BB M)  " #FGX> sB B6B65B6+*-/z**z==z!=z<==>&|^c0tj|||Sr%)npr)condr/r0s r!rrs 88D'8 ,,r#c~d}tdd||drtjd|||dd}| t|||}|S)Nrz#where(cond_value, a_value, b_value)) cond_valuea_valueb_valuerCrD)r=r&rr)rZr/r0rMs r!rrsO Fgw'B 1&*w8T  ~ w9 Mr#zcompute.use_numexprc |jtk(S#t$r#t|ttj fcYSwxYwr%)r2rAttributeError isinstancerYbool_)xs r!_has_bool_dtyperds:/ww$ /!dBHH-../s)AA)rOrPrQct|rAt|r6|tvr.tjd|dt|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)rd_BOOL_OP_UNSUPPORTEDwarningswarnr)r.r/r0s r!rLrLsX wOH$= ) ) MM9&D+F36iA,-    r#cTt|}||rt||||St||||S)a Evaluate and return the expression of the op on left_op and right_op. Parameters ---------- op : the actual operand left_op : left operand right_op : right operand use_numexpr : bool, default True Whether to try to use numexpr. )_op_str_mappingrr)r-r/r0 use_numexprr.s r!rrs9R F  R(; ; b&'8 <rs# %04!;''78+<