o TjO@sDUdZddlmZddlZddlmZddlZddlZddl m Z ddl m Z ddl mZddlmZer9ddlZerAdd lmZdad ed <gad ed <eadaded<daded<hdhddZdZdHdIddZdJdKddZddZdLd d!Z d"d#Z!iej"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/d1d9ZAdLd:d;ZBdHdMd=d>ZCdHdMd?d@ZDdHdIdAdBZEdNdDdEZFdOdFdGZGdS)PzL 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@BTvrreturnNonecCs(tr|atrtntatrtadStadSN)r USE_NUMEXPR_evaluate_numexpr_evaluate_standardr _where_numexpr_where_standardr rrf/home/dinkstrade/pdmp-scanner/venv/lib/python3.10/site-packages/pandas/core/computation/expressions.pyset_use_numexpr-s r!cCs.trtr|dur t}t|dSdSdSr)rrnedetect_number_of_coresset_num_threads)nrrr set_numexpr_threads:s r&cCstrtd|||S)z Standard evaluation. F)r _store_test_result)opop_strabrrr rCs rcCsZ|dur+|jtkr+t}||fD]}t|dr||jjhO}qt|r)t||kr+dSdS)z,return a boolean if we WILL be using numexprNdtypeTF)size _MIN_ELEMENTSsethasattrr,namelen_ALLOWED_DTYPES)r(r)r*r+ dtype_checkdtypesorrr _can_use_numexprLs   r7cCsd}t||||drP|jdd}|r||}}|}|}ztjd|d||ddd}Wnty8YntyHt|||rEnYnw|rP||}}t rXt |du|durct ||||}|S) Nr_rza_value z b_value)a_valueb_valuesafe local_dictcasting) r7__name__strip startswithr"r TypeErrorNotImplementedError_bool_arith_fallbackr r'r)r(r)r*r+result is_reversedr:r;rrr r_s8       r+*-/z**z==z!=z<==>&|^cCst|||Sr)npr)condr*r+rrr rsrcCsDd}tdd||drtjd|||ddd}|dur t|||}|S)Nrz#where(cond_value, a_value, b_value)) cond_valuer:r;r<r=)r7r"rr)rRr*r+rFrrr rs  rzcompute.use_numexprcCs0z|jtkWStyt|ttjfYSwr)r,rAttributeError isinstancerQbool_)xrrr _has_bool_dtypes   rX)rHrIrJcCsJt|r#t|r#|tvr#tjdt|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)rX_BOOL_OP_UNSUPPORTEDwarningswarnreprr)r)r*r+rrr rEs  rE use_numexprcCs0t|}|dur|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. N)_op_str_mappingr r)r(r*r+r^r)rrr rs  rcCs(tdusJ|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)rRr*r+r^rrr rs 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_modesr` used_numexprcCs|r t|dSdSr)r append)rarrr r'sr'cCs t}ga|S)z1 Get test result and reset test_results. )r )resrrr get_test_resultsrd)T)rrrrr)rr)rr)r^r)rarrr)rr )H__doc__ __future__roperatortypingrr[numpyrQpandas._configrpandas.util._exceptionsr pandas.corerpandas.core.computation.checkrnumexprr"pandas._typingrr __annotations__r rr r r3r.r!r&rr7raddraddmulrmulsubrsubtruedivrtruedivfloordiv rfloordivmodrmodpowrpoweqleltgegtand_rand_or_ror_xorrxordivmodrdivmodr_rrrXrZrErrr`r'rdrrrr s              )  $