j!xUdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZdd lmZdd lmZer!ddlZeejed kZerdd lmZdad ed<gaded<eadaded<da ded<hdhddZ!dZ"d=d>dZ#d?d@dZ$dZ%dAd Z&d!Z'iej(d"ej)d"ej*d#ej+d#ej,d$ej-d$ej.d%ej/d%ej0dej1dej2dej3dej4d&ej5d&ej6d'ejd(ej7d)ej8d*ej9d+ej:d,ej;d-ej<d-ej=d.ej>d.ej?d/ej@d/eAdejBdi ZCd0ZDd1ZEe#e d2d3ZFd.d-d/d4ZGdAd5ZHd=dBd7ZId=dBd8ZJd=d>d9ZKdCd;ZLdDd<ZMdS)EzL 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@BTvrreturnNonectrtr tsdan|atrtnt atrtnta dS)NF) r r ne_gt_211 USE_NUMEXPR_evaluate_numexpr_evaluate_standardr_where_numexpr_where_standardrrs QC:\Projects\DSCons\.venv\Lib\site-packages\pandas\core\computation\expressions.pyset_use_numexprr$2sN  I  KKK &1H!!6HI* ?^^FFFc|tr2tr-|tj}tj|dSdSdSN)r rnedetect_number_of_coresset_num_threads)ns r#set_numexpr_threadsr,DsQ[ 9)++A 1r%cFtrtd|||S)z Standard evaluation. F)r _store_test_result)opop_strleft_opright_ops r#rrMs,"5!!! 2gx  r%c|g|jtkrWt}||fD]"}t|dr||jjhz}#t |rt||krdSdS)z2return left_op boolean if we WILL be using numexprNdtypeTF)size _MIN_ELEMENTSsethasattrr4namelen_ALLOWED_DTYPES)r/r0r1r2 dtype_checkdtypesos r#_can_use_numexprr?Vs~  <- ' '"uuFx( - -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_rz left_value z right_value) left_value right_valuesafe local_dictcasting) r?__name__strip startswithr(r TypeErrorNotImplementedError_bool_arith_fallbackr r.r)r/r0r1r2result is_reversedrCrDs r#rris8 FFGXzBB2k'',,77<<  2 ('XG   [2f222*4[QQFF     D"   #FGX>>    2 ('XG/6-... ~#BBB MsA-- B9BB+*-/z**z==z!=z<==>&|^c.tj|||Sr')npr)condr1r2s r#r!r!s 8D'8 , ,,r%cd}tdd||drtjd|||dd}|t|||}|S)Nrz#where(cond_value, a_value, b_value)) cond_valuea_valueb_valuerErF)r?r(rr!)r\r1r2rOs r#r r se Fgw'BB  1&*w8TT    ~ w99 Mr%zcompute.use_numexprc |jtkS#t$r$t|ttjfcYSwxYwr')r4rAttributeError isinstancer[bool_)xs r#_has_bool_dtyperfsK/w$ ///!dBH-...../s+AA)rQrRrSct|rNt|r?|tvr6tjd|dt|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)rf_BOOL_OP_UNSUPPORTEDwarningswarnr)r0r1r2s r#rNrNs wOH$=$= ) ) ) MA&AA+F3AAA,--      4 5r% use_numexprclt|}||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/r1r2rlr0s r#rrsDR F   <R(;; ; b&'8 < <rs)#""""" %%%%%%000000444444!!!!!!;;;;;;''''''=''778+<+<