E g,?@dZddlmZddlmZddlmZddlZddlmZm Z m Z ddl Z ddl mZddlmZmZddlmcmZdd lmZmZdd lmZdd lmZmZerdd lmZm Z d Z!dZ"dZ#e"e#zZ$dZ%GddZ&Gdde&Z'ddddZ(GddZ)dZ*dZ+dZ,ej-ej.ej/ej0ej1ej2e*e+fZ3e4e5e,e3Z6dZ7ej8ej9ej8ej9fZ:e4e5e7e:Z;dZej?ej@ejAejBejCfZDe4e5e3u%%-xboolis_localreturnNonec||_||_||_t|}|t p|t v|_||_ ||_ dSr0) _namer8r9r2 startswith LOCAL_TAGrrA _resolve_name_valuer:)selfr7r8r9r:tnames r>__init__z Term.__init__Ts_  D ((33Ou7O ((**   r?r2cB|jtdS)N)r7replacerGrJs r> local_namezTerm.local_name^sy  B///r?c*t|jSr0)rr7rPs r>__repr__z Term.__repr__bsDI&&&r?c|jSr0valuerJargskwargss r>__call__z Term.__call__e zr?c|Sr0rWs r>evaluatez Term.evaluatehs r?c\t|j}|j}||jjvr't |jj|t rd}|j||}||t|dr|j dkrtd|S)NF)rAndimz?N-dimensional objects, where N > 2, are not supported with eval) r2rQrAr8scoper1typeresolveupdatehasattrr`NotImplementedError)rJrQrAress r>rHzTerm._resolve_nameks)) =  ' 'J HN: &- - 'HhzH== C 3   CHqLL%Q  r?c|j}t|tr"|j|j||||_dS)z search order for local (i.e., @variable) variables: scope, key_variable [('locals', 'local_name'), ('globals', 'local_name'), ('locals', 'key'), ('globals', 'key')] ) new_valueN)r7r1r2r8swapkeyrQrV)rJrVkeys r>rez Term.update|sJi c3   D H  T_cU  C C C r?c*t|jSr0)r rIrPs r>r zTerm.is_scalars%%%r?c |jjjS#t$r6 |jjcYS#t$rt |jcYcYSwxYwwxYwr0)rIvaluesdtypeAttributeErrorrcrPs r>rcz Term.types} );%+ + ) ) ) ){((((! ) ) )DK((((((( )  )s) A ,AA AAAcjt|jdt|jd|jdS)Nz(name=z, type=))rc__name__reprr7rPs r>rawzTerm.raws4t**%QQT$)__QQTYQQQQr?c |jj}n#t$r |j}YnwxYwt|ttjfSr0)rcrq issubclassrnp datetime64rJts r> is_datetimezTerm.is_datetimesQ  AA    AAA !h 6777 ##c|jSr0rIrPs r>rVz Term.values {r?c||_dSr0r)rJrjs r>rVz Term.values  r?c|jSr0rErPs r>r7z Term.name zr?intc|jjSr0)rIr`rPs r>r`z Term.ndims {r?)NNrBrCrBr2)rBr.rBr@)rBr)rt __module__ __qualname__r5__annotations__rLpropertyrQrSrZr^rHrer rc return_typervr}rVsetterr7r` __classcell__r=s@r>r.r.Ks NNN!!!!!000X0''''"$&&&X& ) )X )K RRRXR888X8X \   \ X   X     r?r.c2eZdZdZedZddZdS)r3c|jSr0rrPs r>rHzConstant._resolve_namer[r?c|jSr0rUrPs r>r7z Constant.namerr?rBr2c*t|jSr0)rur7rPs r>rSzConstant.__repr__sDIr?Nr)rtrrrHrr7rSr]r?r>r3r3sRXr?r3~&|)notandorceZdZUdZded<ddd Zdd Zdd Zed Z eddZ edZ eddZ eddZ dS)Opz. Hold an operator of arbitrary arity. r2opNoperandsIterable[Term | Op]rBrCcbt|||_||_||_dSr0) _bool_op_mapgetrrr:)rJrrr:s r>rLz Op.__init__s+""2r**    r?rc*t|jSr0)iterrrPs r>__iter__z Op.__iter__sDM"""r?czd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. c3<K|]}dt|dVdS)(rsN)r).0oprs r> zOp.__repr__..s6EE+|C((+++EEEEEEr? )rrrjoin)rJpareneds r>rSz Op.__repr__s@ FEt}EEENNNN//88999r?c|jttzvr tjSt dt j|DS)Nc3$K|] }|jV dSr0rcrterms r>rz!Op.return_type..s$!J!J$)!J!J!J!J!J!Jr?)r CMP_OPS_SYMS BOOL_OPS_SYMSrybool_r comflattenrPs r>rzOp.return_typesA 7|m3 4 48O!J!J D8I8I!J!J!JKKr?r@c~|j}ttjdg}|jt ko||z S)Nobject) operand_types frozensetryrprr)rJtypes obj_dtype_sets r>has_invalid_return_typezOp.has_invalid_return_types<"!28H#5#5"677 6)Cem.CCr?cXtdtj|DS)Nc3$K|] }|jV dSr0rrs r>rz#Op.operand_types..s$AAtAAAAAAr?)rrrrPs r>rzOp.operand_typess)AAs{4/@/@AAAAAAr?c>td|jDS)Nc3$K|] }|jV dSr0)r )roperands r>rzOp.is_scalar..s%BB7$BBBBBBr?)allrrPs r>r z Op.is_scalars!BBDMBBBBBBr?c |jj}n#t$r |j}YnwxYwt|tt jfSr0)rrcrqrxrryrzr{s r>r}zOp.is_datetimesT ! %AA ! ! ! AAA !!h 6777r~r0)rr2rrrBrC)rBrrr)rtrr__doc__rrLrrSrrrrr r}r]r?r>rrs GGG!!!!! ####::::LLXL DDDXD BBXBCCCXC888X888r?rc ||S#t$r>t|r( ||cYS#t$rYnwxYw||vcYSwxYw)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. isinrqr xys r>_inrs vvayy  ??  vvayy   !    Av s2AAA AAAAAc ||S#t$r?t|r) ||cYS#t$rYnwxYw||vcYSwxYw)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. rrs r>_not_inrs q z  ??  q z!!!!    z s2A!A A! AA!AA! A!)>=z<=z==z!=inznot in)rrrr)+-*/**//%)rrrrBrCctj|}|D]g}|j|vr |j|}n*#t $r||j}YnwxYw||hdS)a$ Cast an expression inplace. Parameters ---------- terms : Op The expression that should cast. acceptable_dtypes : list of acceptable numpy.dtype Will not cast if term's dtype in this list. dtype : str or numpy.dtype The dtype to cast to. N)ryrprcrVastyperqre)termsacceptable_dtypesrpdtrrjs r> _cast_inplacerOs %B 9) ) )  , ))"--II , , , ++III , Is?$A&%A&r@c,t|tSr0)r1r.)objs r>is_termrhs c4  r?c@eZdZdZd fd ZdZd d Zdd Zd ZxZ S)BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op rr2rBrCct|||f||_||_|| t ||_dS#t$rN}tt }tdt|d||d}~wwxYw)NzInvalid binary operator , valid operators are ) r4rLlhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorru)rJrrrerrrr=s r>rLzBinOp.__init__ws c3Z((( ++---  (,DIII   (--//00DQ488QQ4QQ  sA11 C ;A CC c||}||}|||S)z Recursively evaluate an expression in Python space. Parameters ---------- env : Scope Returns ------- object The result of an evaluated expression. )rrr)rJr8leftrights r>rZzBinOp.__call__s6xx}} yyu%%%r?enginecd|dkr ||}n}|j|||||}|j|||||}|j|vr!||j|j}nddlm} | ||||}||} || |S)al Evaluate a binary operation *before* being passed to the engine. Parameters ---------- env : Scope engine : str parser : str term_type : type eval_in_python : list Returns ------- term_type The "pre-evaluated" expression as an instance of ``term_type`` python)rparser term_typeeval_in_pythonr)eval) local_dictrr)r8) rr^rrrrVpandas.core.computation.evalradd_tmp) rJr8rrrrrhrrrr7s r>r^zBinOp.evaluates" X  $s))CC8$$#- %DH%%#- &Ew.((ii EK88======d4CvNNN{{3y3''''r?cfd}jj}}t|r|jrt|r|jr|j}t |ttfr ||}tt|}|j | d}j |t|r|jrt|r|jr|j}t |ttfr ||}tt|}|j | d}j |dSdSdSdSdS)zK Convert datetimes to a comparable value in an expression. cnjttj}nt}||S)N)r:)r:rrr)rVencoderrJs r> stringifyz'BinOp.convert_values..stringifys5}(!"6OOO&75>> !r?NUTC)rrrr}r rVr1rfloatr r tz tz_convertre)rJrrrvs` r>rzBinOp.convert_valuess  " " " " "8TXS 3<< CO     A!c5\** !IaLL.++,,AtLL'' HOOA    3<< CO     A!c5\** !IaLL.++,,AtLL'' HOOA             r?cd|j}|j}|j}t|d|}|j}t|d|}|js|jra|jt vrQt|ttj fr!t|ttj fstddSdSdS)Nrcz$cannot evaluate scalar only bool ops) rrrgetattrr r_bool_ops_dictrxr@ryrrg)rJrrrhs_rtlhs_rts r>rz$BinOp._disallow_scalar_only_bool_opsshh0000 ] N!m N>))vbh'788*#6D"(+;<< *&&LMM M*) N N))r?)rr2rBrC)rr2r) rtrrrrLrZr^rrrrs@r>rrls$&&&&/(/(/(/(bBNNNNNNNr?rcdttj|jtjSr0)rxryrprcnumber)rps r> isnumericrs bhuoo*BI 6 66r?c$eZdZdZdfd ZxZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. rBrCc xtd||t|jrt|js(t d|jd|jd|jdt jt jg}ttj ||t jdS)Nrz unsupported operand type(s) for z: 'z' and '') r4rLrr TypeErrorrryfloat32float64rrr)rJrrrr=s r>rLz Div.__init__s c3''')) 3?1K1K ?47??O??,/O???   Z4ck$''):BJGGGGGr?r)rtrrrrLrrs@r>rrsQ H H H H H H H H H Hr?r)rrrrcLeZdZdZd fd ZddZdd Zedd ZxZ S)UnaryOpaK Hold a unary operator and its operands. Parameters ---------- op : str The token used to represent the operator. operand : Term or Op The Term or Op operand to the operator. Raises ------ ValueError * If no function associated with the passed operator token is found. rLiteral['+', '-', '~', 'not']rBrCct||f||_ t||_dS#t $r-}t dt|dt|d}~wwxYw)NzInvalid unary operator r) r4rLr_unary_ops_dictrrrruUNARY_OPS_SYMS)rJrrrr=s r>rLzUnaryOp.__init__6s gZ(((  '+DIII   8$r((88'588  sA A8 (A33A8MathCallcV||}||Sr0)rr)rJr8rs r>rZzUnaryOp.__call__Bs%,,s##yy!!!r?r2c@t|jd|jdS)Nrrs)rrrrPs r>rSzUnaryOp.__repr__Gs%tw88888999r?np.dtypec$|j}|jtjdkrtjdSt |t r0|jtvs|jtvrtjdStjdS)Nr@r) rrryrpr1rr _cmp_ops_dictr )rJrs r>rzUnaryOp.return_typeJsz,  "(6"2"2 2 28F## # gr " " $ J- ' '7:+G+G8F## #xr?)rrrBrCrBrr)rBr") rtrrrrLrZrSrrrrs@r>rr%s      """" ::::Xr?rc.eZdZdfd ZdZddZxZS) rrBrCcdt|j|||_dSr0)r4rLr7r)rJrrXr=s r>rLzMathCall.__init__Ws* D))) r?cHfd|jD}|jj|S)Nc&g|] }|Sr]r])rrr8s r> z%MathCall.__call__..]s!444BBsGG444r?)rr)rJr8rs ` r>rZzMathCall.__call__[s/4444dm444ty~x((r?r2ctt|j}t|jdd|dS)Nr,rs)mapr2rrrr)rJrs r>rSzMathCall.__repr__`s@sDM**tw>>(););>>>???r?rr)rtrrrLrZrSrrs@r>rrVsh))) @@@@@@@@r?rceZdZd dZd dZdS) FuncNoder7r2rBrCc|tvrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr7r ryr)rJr7s r>rLzFuncNode.__init__fsC w  DDDDEE E B%% r?rc"t||Sr0)r)rJrXs r>rZzFuncNode.__call__lsd###r?N)r7r2rBrCr%)rtrrrLrZr]r?r>r/r/es<&&&& $$$$$$r?r/rr)Zr __future__rr functoolsroperatortypingrrrnumpyrypandas._libs.tslibsr pandas.core.dtypes.commonr r pandas.core.commoncorecommonrpandas.core.computation.commonr r pandas.core.computation.scoperpandas.io.formats.printingrrcollections.abcrr REDUCTIONS_unary_math_ops_binary_math_opsr2rGr.r3rrrrrgtltgeleeqne_cmp_ops_funcsdictzipr$rand_or__bool_ops_funcsr ARITH_OPS_SYMSaddsubmultruedivpowfloordivmod_arith_ops_funcs_arith_ops_dictSPECIAL_CASE_ARITH_OPS_SYMS_special_case_arith_ops_funcs_special_case_arith_ops_dictrdrerrrrrrposneginvert_unary_ops_funcsrrrr/r]r?r>rcs#""""" )))))) ! :99999  + ,  , ,  q q q q q q q q h     t   3c22 3838383838383838l       B  K K K K K K  SS~6677 ' =(, x|Lcc-99::6 L L L  L  L$ss>+;<<==/!)x/@(, O#tC#%BCC    9AA2!!!!UNUNUNUNUNBUNUNUNp7777HHHHH%HHH0(L(,Q$ss>+;<<==.....b...b @ @ @ @ @r @ @ @$$$$$$$$$$r?