a OfO8@sHUdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZerrddlZerdd lmZdad ed <gad ed <eadaded<daded<hdhddZdZdHdddddZdIddddZddZddd d!Z d"d#Z!ej"d$ej#d$ej$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/ej7d0ej8d0ej9d1ej:d1e;dejd4d5Z?ee d6d7d8Z@d0d/d1d9ZAddd:d;ZBdJdd<d=d>ZCdKdd<d?d@ZDdLddddAdBZEdddCdDdEZFd ddFdGZGdS)MzL Expressions ----------- Offer fast expression evaluation through numexpr ) annotationsN) TYPE_CHECKING) get_option)find_stack_level) roperator)NUMEXPR_INSTALLED)FuncTypez bool | None _TEST_MODEz list[bool] _TEST_RESULTzFuncType | None _evaluate_where>int64Zfloat32boolZint32float64>r rr)evaluatewherei@BTrNone)vreturncCs$tr|atrtntatrtntadSN)r USE_NUMEXPR_evaluate_numexpr_evaluate_standardr _where_numexpr_where_standardr rr\/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/computation/expressions.pyset_use_numexpr-s r)rcCs&tr"tr"|durt}t|dSr)rrneZdetect_number_of_coresZset_num_threads)nrrrset_numexpr_threads:sr!cCstr td|||S)z Standard evaluation. F)r _store_test_result)opop_strabrrrrCsrcCsZ|durV|jtkrVt}||fD]}t|dr ||jjhO}q t|rRt||krVdSdS)z,return a boolean if we WILL be using numexprNdtypeTF)size _MIN_ELEMENTSsethasattrr'namelen_ALLOWED_DTYPES)r#r$r%r&Z dtype_checkZdtypesorrr_can_use_numexprLs   r0cCsd}t||||dr|jdd}|r4||}}|}|}z"tjd|d||ddd}Wn2typYn"tyt|||rnYn0|r||}}t rt |du|durt ||||}|S) Nr_rza_value z b_value)a_valueb_valuesafeZ local_dictZcasting) r0__name__strip startswithrr TypeErrorNotImplementedError_bool_arith_fallbackr r"r)r#r$r%r&resultZ is_reversedr3r4rrrr_s4        r+*-/z**z==z!=z<==>&|^cCst|||Sr)npr)condr%r&rrrrsrcCsDd}tdd||dr,tjd|||ddd}|dur@t|||}|S)Nrz#where(cond_value, a_value, b_value))Z cond_valuer3r4r5r6)r0rrr)rHr%r&r=rrrrs  rzcompute.use_numexprcCs4z |jtkWSty.t|ttjfYS0dSr)r'rAttributeError isinstancerGZbool_)xrrr_has_bool_dtypes  rL)r>r?r@cCsJt|rFt|rF|tvrFtjdt|dtt|dtddSdS)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)rL_BOOL_OP_UNSUPPORTEDwarningswarnreprr)r$r%r&rrrr<s  r<) use_numexprcCs0t|}|dur"|r"t||||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. N)_op_str_mappingr r)r#r%r&rRr$rrrrs rcCs(tdus J|rt|||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. N)r r)rHr%r&rRrrrrs rcCs |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 rrrr set_test_modesrT) used_numexprrcCs|rt|dSr)r append)rUrrrr"sr"cCs t}ga|S)z1 Get test result and reset test_results. )r )resrrrget_test_resultsrX)T)N)T)T)T)H__doc__ __future__roperatortypingrrOnumpyrGZpandas._configrZpandas.util._exceptionsrZ pandas.corerZpandas.core.computation.checkrZnumexprrZpandas._typingrr __annotations__r rr r r.r)rr!rr0raddZraddmulZrmulsubZrsubtruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqleltgegtand_Zrand_or_Zror_xorZrxordivmodZrdivmodrSrrrLrNr<rrrTr"rXrrrrs              *$