Owg9dZddlmZddlmZddlmZddlZddlmZm 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,ejZej\ej^ej`ejbejde*e+fZ3e4e5e,e3Z6dZ7ejpejrejpejrfZ:e4e5e7e:Z;dZ__init__z Term.__init__TsS  D ((3Ou7O ((*   r?cB|jjtdS)N)r7replacerErHs r> local_namezTerm.local_name^syy  B//r?c,t|jSr0)rr7rNs r>__repr__z Term.__repr__bsDII&&r?c|jSr0valuerHargskwargss r>__call__z Term.__call__e zzr?c|Sr0rUs r>evaluatez Term.evaluatehs r?crt|j}|j}||jjvr)t |jj|t rd}|jj||}|j|t|dr|jdkDr td|S)NF)rAndimz?N-dimensional objects, where N > 2, are not supported with eval) r2rOrAr8scoper1typeresolveupdatehasattrr^NotImplementedError)rHrOrAress r>rFzTerm._resolve_nameks) ==  'J HHNN: &- HhhzH= C 3 CHHqL%Q  r?c|j}t|tr(|jj |j ||||_y)z search order for local (i.e., @variable) variables: scope, key_variable [('locals', 'local_name'), ('globals', 'local_name'), ('locals', 'key'), ('globals', 'key')] ) new_valueN)r7r1r2r8swapkeyrOrT)rHrTkeys r>rcz Term.update|s=ii c3  HH  T__cU  C r?c,t|jSr0)r rGrNs r>r zTerm.is_scalars%%r?c |jjjS#t$r@ |jjcYS#t$rt |jcYcYSwxYwwxYwr0)rGvaluesdtypeAttributeErrorrarNs r>raz Term.types_ );;%%++ + ) ){{(((! )DKK(( )  )s," A+AA+A'"A+&A''A+cxt|jdt|jd|jdS)Nz(name=z, type=))ra__name__reprr7rNs r>rawzTerm.raws3t*%%&fT$))_,=WTYYKqQQr?c |jj}t|ttj fS#t$r|j}Y7wxYwr0)raro issubclassrnp datetime64rHts r> is_datetimezTerm.is_datetimesF  A!h 677  A 8AAc|jSr0rGrNs r>rTz Term.values {{r?c||_yr0r~)rHrhs r>rTz Term.values  r?c|jSr0rCrNs r>r7z Term.name zzr?c.|jjSr0)rGr^rNs r>r^z Term.ndims{{r?)NNreturnNonerr2)rr.rr@)rint)rr __module__ __qualname__r5__annotations__rJpropertyrOrQrXr\rFrcr ra return_typertr{rTsetterr7r^ __classcell__r=s@r>r.r.Ks N!00'"$&& ) )K RR88 \\    r?r.c*eZdZdZedZddZy)r3c|jSr0rrNs r>rFzConstant._resolve_namerYr?c|jSr0rSrNs r>r7z Constant.namerr?c,t|jSr0)rsr7rNs r>rQzConstant.__repr__sDIIr?Nr)rrrrrFrr7rQr[r?r>r3r3s r?r3~&|)notandorceZdZUdZded<d ddZddZddZedZ edd Z ed Z edd Z edd Z y)Opz. Hold an operator of arbitrary arity. r2opNcVtj|||_||_||_yr0) _bool_op_mapgetroperandsr:)rHrrr:s r>rJz Op.__init__s$""2r*    r?c,t|jSr0)iterrrNs r>__iter__z Op.__iter__sDMM""r?cxd|jD}td|jdj|S)zW Print a generic n-ary operator and its operands using infix notation. c3:K|]}dt|dyw)(rqN)r).0oprs r> zOp.__repr__..sEQ|C()+Es )rrrjoin)rHpareneds r>rQz Op.__repr__s5 Ft}}EayN//899r?c|jttzvrtjSt dt j|DS)Nc34K|]}|jywr0rarterms r>rz!Op.return_type..s!J$))!J)r CMP_OPS_SYMS BOOL_OPS_SYMSrwbool_r comflattenrNs r>rzOp.return_types8 77|m3 488O!J D8I!JKKr?c|j}ttjdg}|jt k(xr||z S)Nobject) operand_types frozensetrwrnrr)rHtypes obj_dtype_sets r>has_invalid_return_typezOp.has_invalid_return_types>""!288H#5"67 6)Cem.CCr?cLtdtj|DS)Nc34K|]}|jywr0rrs r>rz#Op.operand_types..sAtAr)rrrrNs r>rzOp.operand_typessAs{{4/@AAAr?c:td|jDS)Nc34K|]}|jywr0)r )roperands r>rzOp.is_scalar..sB7$$Br)allrrNs r>r z Op.is_scalarsBDMMBBBr?c |jj}t|tt j fS#t$r|j}Y7wxYwr0)rrarorvrrwrxrys r>r{zOp.is_datetimesL !  %%A!h 677 !  A !r|r0)rr2rzIterable[Term | Op]rr)rrrr)rrrr__doc__rrJrrQrrrrr r{r[r?r>rrs G! #:LL DD BBCC88r?rc |j|S#t$r5t|r# |j|cYS#t$rYnwxYw||vcYSwxYw)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. isinror xys r>_inrsY vvay  ? vvay !  Av s/A;A AAAAAc |j|S#t$r6t|r$ |j|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 s/A=A A AA  AA)>=z<=z==z!=inznot in)rrrr)+-*/**//%)rrrc"t|tSr0)r1r.)objs r>is_termrOs c4  r?c@eZdZdZdfd ZdZddZd dZdZxZ S) BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op c6t||||f||_||_|j |j  t ||_y#t$r=}tt j}tdt|d||d}~wwxYw)NzInvalid binary operator , valid operators are ) r4rJlhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorrs)rHrrrerrrr=s r>rJzBinOp.__init__^s c3Z( ++-  (,DI (--/0D*48*4J4&Q  sA B8BBcj|j|}|j|}|j||S)z Recursively evaluate an expression in Python space. Parameters ---------- env : Scope Returns ------- object The result of an evaluated expression. )rrr)rHr8leftrights r>rXzBinOp.__call__ps/xx} yyu%%r?cb|dk(r ||}n|jj|||||}|jj|||||}|j|vr'|j |j |j }nddlm} | ||||}|j|} || |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)engineparser term_typeeval_in_pythonr)eval) local_dictrr)r8) rr\rrrrTpandas.core.computation.evalradd_tmp) rHr8rrrrrfrrrr7s r>r\zBinOp.evaluates" X s)C88$$#- %DHH%%#- &Eww.(ii EKK8=4CvN{{33''r?cfd}jj}}t|r|jrt|r|jrv|j }t |ttfr||}tt|}|j|jd}jj|t|r|jrt|r|jrw|j }t |ttfr||}tt|}|j|jd}jj|yyyyy)zK Convert datetimes to a comparable value in an expression. cpjttj}nt}||S)N)r:)r:rrr)rTencoderrHs r> stringifyz'BinOp.convert_values..stringifys-}}(!"6O&5> !r?NUTC)rrrr{r rTr1rfloatr r tz tz_convertrc)rHrrrvs` r>rzBinOp.convert_valuess  "88TXXS 3rz$BinOp._disallow_scalar_only_bool_opsshhhh00 ]]cmm>)vbhh'78"6D"((+;<&&LM M = *,r?)rr2rr)rr2r) rrrrrrJrXr\rrrrs@r>rrSs$$&&/(bBNr?rcpttj|jtjSr0)rvrwrnranumber)rns r> isnumericr s! bhhuo**BII 66r?)rrrrcFeZdZdZdfd ZddZddZed dZxZ 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. ct|||f||_ t||_y#t $r$}t dt|dt|d}~wwxYw)NzInvalid unary operator r) r4rJr_unary_ops_dictrrrrsUNARY_OPS_SYMS)rHrrrr=s r>rJzUnaryOp.__init__si gZ(  '+DI )$r(4''5&68  s* AAAcF|j|}|j|Sr0)rr)rHr8rs r>rXzUnaryOp.__call__s,,s#yy!!r?cLt|jd|jdS)Nrrq)rrrrNs r>rQzUnaryOp.__repr__s"twwiqa899r?cD|j}|jtjdk(rtjdSt |t r9|j tvs|j tvrtjdStjdS)Nr@r) rrrwrnr1rr _cmp_ops_dictr)rHrs r>rzUnaryOp.return_typeso,,   "((6"2 288F# # gr " JJ- '7::+G88F# #xxr?)rzLiteral['+', '-', '~', 'not']rrrMathCallr)rznp.dtype) rrrrrrJrXrQrrrrs@r>r r s+ " :r?r c.eZdZdfd ZdZddZxZS)rcHt||j|||_yr0)r4rJr7r)rHrrVr=s r>rJzMathCall.__init__&s D) r?c||jDcgc] }|| }}|jj|Scc}wr0)rr)rHr8rrs r>rXzMathCall.__call__*s7&*mm4BsG44tyy~~x((5s9ctt|j}t|jddj |dS)Nr,rq)mapr2rrrr)rHrs r>rQzMathCall.__repr__/s8sDMM*twwiq();(??r?rr)rrrrrJrXrQrrs@r>rr%s) @r?rceZdZddZddZy)FuncNodecj|tvrtd|d||_tt||_y)N"z" is not a supported function)MATHOPSrr7rrwr)rHr7s r>rJzFuncNode.__init__5s4 w q&CDE E B% r?ct||Sr0)r)rHrVs r>rXzFuncNode.__call__;sd##r?N)r7r2rrr)rrrrrJrXr[r?r>rr4s & $r?rr)Xr __future__rr functoolsroperatortypingrrrnumpyrwpandas._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_opsr!rEr.r3rrrrrgtltgeleeqne_cmp_ops_funcsdictziprrand_or__bool_ops_funcsrARITH_OPS_SYMSaddsubmultruedivpowfloordivmod_arith_ops_funcs_arith_ops_dictSPECIAL_CASE_ARITH_OPS_SYMS_special_case_arith_ops_funcs_special_case_arith_ops_dictrdrcrrr rposneginvert_unary_ops_funcsrr rrr[r?r>rRsl# ) ! :  + ,  , ,  q q h t 3c2 3838l   B  KK KK KK KK KK KK  S~67 ' ==(,, x||Lc-9:6 LL LL LL  LL  LLs>+;<=/!)x/@/@(,, O##%BC   9AA!UNBUNp7(LL(,,Qs>+;<=.b.b @r @$$r?