U
|
¬ý°d6 ã 8 @ s@ U d Z ddlmZ ddlZddlZddlZddlmZ ddl m
|
Z
|
ddl m Z ddl mZ ddlmZ errddlZdad ed
|
< g ad ed < eadad ed< dad ed< dddddhdddhdZdZdIdddddZdJdd d d Z d d Z dd d!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)ej.d*ejd+ej/d,ej0d-ej1d.ej2d/ej3d0ej4d0ej5d1ej6d1ej7d2ej8d2e9dej:di Z;d3d4 Z<d5d6 Z=eed7 d8d9 Z>d1d0d2d:Z?dd d;d<Z@dKdd=d>d?ZAdLdd=d@dAZBdMddddBdCZCdddDdEdFZDd d dGdHZEdS )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Ú_whereÚint64Úint32Úfloat64Úfloat32Úbool)ÚevaluateÚwherei@B TÚNone)ÚvÚreturnc C s$ t r| atrtntatr tntad S ©N)r Ú USE_NUMEXPRÚ_evaluate_numexprÚ_evaluate_standardr
|
Ú_where_numexprÚ_where_standardr ©r © r úZd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/computation/expressions.pyÚset_use_numexpr* s r )r c C s& t r"tr"| d krt ¡ } t | ¡ d S r )r r ÚneZdetect_number_of_coresZset_num_threads)Únr r r Úset_numexpr_threads7 s r" c C s t r td | ||S )z
|
Standard evaluation.
|
F)r Ú_store_test_result)ÚopÚop_strÚaÚbr r r r @ s r c C sZ |dk rV|j tkrVt }||fD ] }t|dr ||jjhO }q t|rRt| |krVdS dS )z,return a boolean if we WILL be using numexprNÚdtypeTF)ÚsizeÚ _MIN_ELEMENTSÚsetÚhasattrr( ÚnameÚlenÚ_ALLOWED_DTYPES)r$ r% r& r' Z dtype_checkZdtypesÚor r r Ú_can_use_numexprI s
|
|
r1 c C sÎ d }t | |||dr¤| j d¡ d¡}|r4|| }}|}|}z"tjd| d||ddd}W n6 tk
|
rr Y n$ tk
|
r t|||rn Y nX |r¤|| }}t r´t
|
|d k |d krÊt | |||}|S ) Nr Ú_Úrza_value z b_value)Úa_valueÚb_valueÚsafe©Z
|
local_dictZcasting) r1 Ú__name__ÚstripÚ
|
startswithr r Ú TypeErrorÚNotImplementedErrorÚ_bool_arith_fallbackr r# r )r$ r% r& r' ÚresultZ is_reversedr4 r5 r r r r \ s4
|
|
ý
|
|
r ú+Ú*ú-ú/z**z==z!=z<=ú<z>=ú>ú&ú|ú^c C s t | ||¡S r )Únpr )Úcondr& r' r r r r © s r c C sD d }t d d||dr,tjd| ||ddd}|d kr@t| ||}|S )Nr z#where(cond_value, a_value, b_value))Z
|
cond_valuer4 r5 r6 r7 )r1 r r r )rI r& r' r> r r r r ® s
|
ý r zcompute.use_numexprc C s6 z | j tkW S tk
|
r0 t| ttjf Y S X d S r )r( r ÚAttributeErrorÚ
|
isinstancerH Zbool_)Úxr r r Ú_has_bool_dtypeà s rM )r? r@ rA c C sJ t |rFt |rF| tkrFtjdt| dtt| dt d dS dS )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)rM Ú_BOOL_OP_UNSUPPORTEDÚwarningsÚwarnÚreprr )r% r& r' r r r r= Í s ür= )Ú use_numexprc C s0 t | }|dk r"|r"t| |||S t| |||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' rS r% r r r r ß s
|
r c C s( t dk s t|r t | ||S t| ||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 )rI r& r' rS r r r r ó s r c C s | a g adS )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 r r r r Ú set_test_mode s rV )Ú used_numexprr c C s | rt | ¡ d S r )r Úappend)rW r r r r# s r# c C s t } g a | S )z1
|
Get test result and reset test_results.
|
)r )Úresr r r Úget_test_result s rZ )T)N)T)T)T)FÚ__doc__Ú
|
__future__r ÚoperatorrP ÚnumpyrH Zpandas._configr Zpandas._typingr Zpandas.util._exceptionsr Z pandas.core.computation.checkr Zpandas.core.opsr Znumexprr r Ú__annotations__r r r
|
r r/ r* r r" r r1 r ÚaddZraddÚmulZrmulÚsubZrsubÚtruedivZrtruedivÚfloordivZ rfloordivÚmodZrmodÚpowZrpowÚeqÚleÚltÚgeÚgtÚand_Zrand_Úor_Zror_ÚxorZrxorÚdivmodZrdivmodrT r r rM rO r= r r rV r# rZ r r r r Ú<module> sº þ * à$
|
