U c8@s,UdZddlmZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZddlmZerrddlZdad ed <gad ed <eadad ed<dad ed<dddddhdddhdZdZdIddddZdJddddZddZd d!Zd"d#Zej d$ej!d$ej"d%ej#d%ej$d&ej%d&ej&d'ej'd'ej(dej)dej*dej+dej,d(ej-d(ej.d)ejd*ej/d+ej0d,ej1d-ej2d.ej3d/ej4d/ej5d0ej6d0ej7d1ej8d1e9dej:diZ;d2d3Zd0d/d1d9Z?d:d;Z@dKdd<d=d>ZAdLd?d@ZBdMdddAdBdCZCdddDdEdFZDd ddGdHZEdS)NzL Expressions ----------- Offer fast expression evaluation through numexpr ) annotationsN) get_option)FuncType)find_stack_level)NUMEXPR_INSTALLED) roperatorz bool | None _TEST_MODEz list[bool] _TEST_RESULTzFuncType | None _evaluate_whereZint64Zint32Zfloat64Zfloat32bool)evaluatewherei@BTNone)returncCs$tr|atrtntatrtntadSN)r USE_NUMEXPR_evaluate_numexpr_evaluate_standardr _where_numexpr_where_standardr vrG/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/computation/expressions.pyset_use_numexpr*s rcCs&tr"tr"|dkrt}t|dSr)rrneZdetect_number_of_coresZset_num_threads)nrrrset_numexpr_threads7srcCstr td|||S)z Standard evaluation. F)r_store_test_result)opop_strabrrrr@srcCsZ|dk rV|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_numexprIs   r-cCsd}t||||dr|jdd}|r4||}}|}|}z"tjd|d||ddd}Wn6tk rrYn$tk rt|||rnYnX|r||}}t rt |dk |dkrt ||||}|S) Nr _rza_value z b_value)a_valueb_valuesafeZ local_dictZcasting) r-__name__strip startswithrr TypeErrorNotImplementedError_bool_arith_fallbackrrr)r r!r"r#resultZ is_reversedr0r1rrrr]s4      r+*-/z**z==z!=z<==>&|^cCst|||Sr)npr)condr"r#rrrrsrcCsDd}tdd||dr,tjd|||ddd}|dkr@t|||}|S)Nrz#where(cond_value, a_value, b_value))Z cond_valuer0r1r2r3)r-rr r)rEr"r#r:rrrrs  rzcompute.use_numexprcCs6z |jtkWStk r0t|ttjfYSXdSr)r$r AttributeError isinstancerDZbool_)xrrr_has_bool_dtypes rI)r;r<r=cCsJt|rFt|rF|tkrFtjdt|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)rI_BOOL_OP_UNSUPPORTEDwarningswarnreprr)r!r"r#rrrr9sr9) use_numexprcCs0t|}|dk r"|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#rOr!rrrr s r cCs(tdk s t|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 AssertionErrorr)rEr"r#rOrrrrs r)rrcCs |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)rr rrrr set_test_modesrR) used_numexprrcCs|rt|dSr)r append)rSrrrrsrcCs t}ga|S)z1 Get test result and reset test_results. )r )resrrrget_test_resultsrV)T)N)T)T)T)F__doc__ __future__roperatorrLZnumpyrDZpandas._configrZpandas._typingrZpandas.util._exceptionsrZpandas.core.computation.checkrZpandas.core.opsrZnumexprrr__annotations__r rr r r+r&rrrr-raddZraddmulZrmulsubZrsubtruedivZrtruedivfloordivZ rfloordivmodZrmodpowZrpoweqleltgegtand_Zrand_or_Zror_xorZrxordivmodZrdivmodrPrrrIrKr9r rrRrrVrrrrs              *$