o $j9@s`dZddlmZddlmZddlmZddlZddlmZm Z m Z ddl Z ddl mZddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZer]dd lmZm Z d Z!dZ"dZ#e"e#Z$dZ%GdddZ&Gddde&Z'ddddZ(GdddZ)ddZ*ddZ+dZ,ej-ej.ej/ej0ej1ej2e*e+fZ3e4e5e,e3Z6d Z7ej8ej9ej8ej9fZ:e4e5e7e:Z;d!Zej?ej@ejAejBejCfZDe4e5er3r3r4 local_name^szTerm.local_namecC t|jSr%)rr,rBr3r3r4__repr__b z Term.__repr__cO|jSr%valuer>argskwargsr3r3r4__call__ez Term.__call__cOs|Sr%r3rJr3r3r4evaluatehsz Term.evaluatecCslt|j}|j}||jjvrt|jj|trd}|jj||d}||t |dr4|j dkr4t d|S)NF)r6ndimz?N-dimensional objects, where N > 2, are not supported with eval) r'rCr6r-scoper&typeresolveupdatehasattrrPNotImplementedError)r>rCr6resr3r3r4r<ks   zTerm._resolve_namecCs.|j}t|tr|jj|j||d||_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)r,r&r'r-ZswapkeyrCrI)r>rIkeyr3r3r4rU|s  z Term.updatecCrDr%)r r=rBr3r3r4r  zTerm.is_scalarc CsLz|jjjWSty%z|jjWYSty$t|jYYSwwr%)r=valuesdtypeAttributeErrorrSrBr3r3r4rSs   z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rS__name__reprr,rBr3r3r4raws$zTerm.rawcCs6z|jj}Wn ty|j}Ynwt|ttjfSr%)rSr^ issubclassrnp datetime64r>tr3r3r4 is_datetime    zTerm.is_datetimecCrGr%r=rBr3r3r4rIz Term.valuecCs ||_dSr%rj)r>rYr3r3r4rIr[cCrGr%r9rBr3r3r4r,rkz Term.nameintcCs|jjSr%)r=rPrBr3r3r4rPsz Term.ndim)NNr7r8r7r')r7r$r7r5)r7rm)r` __module__ __qualname__r*__annotations__r?propertyrCrErMrOr<rUr rS return_typerbrhrIsetterr,rP __classcell__r3r3r1r4r$Ks8              r$c@s*eZdZddZeddZd ddZd S) r(cCrGr%rlrBr3r3r4r<rNzConstant._resolve_namecCrGr%rHrBr3r3r4r,rkz Constant.namer7r'cCrDr%)rar,rBr3r3r4rEs zConstant.__repr__Nro)r`rqrrr<rtr,rEr3r3r3r4r(s  r(~&|)notandorc@s|eZdZUdZded<ddd d Zdd d ZdddZeddZ edddZ eddZ edddZ edddZ dS) Opz. Hold an operator of arbitrary arity. r'opNoperandsIterable[Term | Op]r7r8cCst|||_||_||_dSr%) _bool_op_mapgetrrr/)r>rrr/r3r3r4r?s z Op.__init__rcCrDr%)iterrrBr3r3r4__iter__rFz Op.__iter__cCs(dd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. css |] }dt|dVqdS)(r_N)r).0Zoprr3r3r4 szOp.__repr__.. )rrrjoin)r>Zparenedr3r3r4rEsz Op.__repr__cCs,|jttvr tjStddt|DS)Ncs|]}|jVqdSr%rSrtermr3r3r4rz!Op.return_type..)r CMP_OPS_SYMS BOOL_OPS_SYMSrdbool_r comflattenrBr3r3r4ruszOp.return_typer5cCs(|j}ttdg}|jtko||S)Nobject) operand_types frozensetrdr]rur)r>typesZ obj_dtype_setr3r3r4has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncsrr%rrr3r3r4rrz#Op.operand_types..)rrrrBr3r3r4rszOp.operand_typescCstdd|jDS)Ncsrr%)r )roperandr3r3r4rrzOp.is_scalar..)allrrBr3r3r4r sz Op.is_scalarcCs6z|jj}Wn ty|j}Ynwt|ttjfSr%)rurSr^rcrrdrerfr3r3r4rhrizOp.is_datetimer%)rr'rrr7r8)r7rrorp)r`rqrr__doc__rsr?rrErtrurrr rhr3r3r3r4r~s        r~c CsTz||WSty)t|r#z||WYSty"Ynw||vYSw)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. isinr^r xyr3r3r4_ins    rc CsXz||WSty+t|r%z ||WYSty$Ynw||vYSw)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. rrr3r3r4_not_ins   r)>=z<=z==z!=inznot in)ryrzr|r})+-*/**//%)rrrr7r5cCs t|tSr%)r&r$)objr3r3r4is_termOrFrcsFeZdZdZdfdd Zdd Zdd d Zdd dZddZZ S)BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op rr'r7r8c s~t|||f||_||_||zt||_WdSty>}zt t }t dt |d||d}~ww)NzInvalid binary operator , valid operators are ) r)r?lhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorra)r>rrrerrrr1r3r4r?^s  zBinOp.__init__cCs ||}||}|||S)z Recursively evaluate an expression in Python space. Parameters ---------- env : Scope Returns ------- object The result of an evaluated expression. )rrr)r>r-leftrightr3r3r4rMps   zBinOp.__call__enginec Cs|dkr ||}n2|jj|||||d}|jj|||||d}|j|vr-||j|j}nddlm} | ||||d}||} || |dS)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)Z local_dictrrr-) rrOrrrrIZpandas.core.computation.evalrZadd_tmp) r>r-rrrrrXrrrr,r3r3r4rOs,     zBinOp.evaluatecsfdd}jj}}t|r?|jr?t|r?|jr?|j}t|ttfr)||}t t |}|j dur9| d}j |t|rs|jrut|rw|jry|j}t|ttfr[||}t t |}|j durk| d}j |dSdSdSdSdS)zK Convert datetimes to a comparable value in an expression. cs,jdurttjd}||St}||S)N)r/)r/rrr)rIencoderrBr3r4 stringifys z'BinOp.convert_values..stringifyNUTC)rrrrhr rIr&rmfloatrr tzZ tz_convertrU)r>rrrvr3rBr4rs&        zBinOp.convert_valuescCsz|j}|j}|j}t|d|}|j}t|d|}|js|jr9|jtvr7t|tt j fr3t|tt j fs;t ddSdSdS)NrSz$cannot evaluate scalar only bool ops) rrrugetattrr r_bool_ops_dictrcr5rdrrW)r>rrZrhs_rtZlhs_rtr3r3r4rs&   z$BinOp._disallow_scalar_only_bool_ops)rr'r7r8)rr'rn) r`rqrrrr?rMrOrrrwr3r3r1r4rSs   1!rcCstt|jtjSr%)rcrdr]rSnumber)r]r3r3r4 isnumericsr)rrrxr{csDeZdZdZdfdd Zdd d Zdd d ZedddZZ 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']r7r8c sZt||f||_zt||_WdSty,}z tdt|dt|d}~ww)NzInvalid unary operator r) r)r?r_unary_ops_dictrrrraUNARY_OPS_SYMS)r>rrrr1r3r4r?s zUnaryOp.__init__MathCallcCs||}||Sr%)rr)r>r-rr3r3r4rMs  zUnaryOp.__call__r'cCst|jd|jdS)Nrr_)rrrrBr3r3r4rEszUnaryOp.__repr__np.dtypecCsR|j}|jtdkrtdSt|tr$|jtvs|jtvr$tdStdS)Nr5rm) rrurdr]r&r~r _cmp_ops_dictr)r>rr3r3r4rus    zUnaryOp.return_type)rrr7r8r7rro)r7r) r`rqrrrr?rMrErtrurwr3r3r1r4rs rcs0eZdZd fdd ZddZd dd ZZS) rr7r8cst|j|||_dSr%)r)r?r,r)r>rrKr1r3r4r?&s zMathCall.__init__cs fdd|jD}|jj|S)Ncsg|]}|qSr3r3)rrrr3r4 ,sz%MathCall.__call__..)rr)r>r-rr3rr4rM*s zMathCall.__call__r'cCs(tt|j}t|jdd|dS)Nr,r_)mapr'rrrr)r>rr3r3r4rE/s zMathCall.__repr__rnro)r`rqrrr?rMrErwr3r3r1r4r%src@s eZdZd ddZd dd Zd S) FuncNoder,r'r7r8cCs.|tvr td|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr,rrdr)r>r,r3r3r4r?5szFuncNode.__init__rcGs t||Sr%)r)r>rKr3r3r4rM;rFzFuncNode.__call__N)r,r'r7r8r)r`rqrrr?rMr3r3r3r4r4s rrp)Xr __future__rr functoolsroperatortypingrrrnumpyrdZpandas._libs.tslibsrZpandas.core.dtypes.commonr r Zpandas.core.commoncorecommonrZpandas.core.computation.commonr r Zpandas.core.computation.scoper Zpandas.io.formats.printingrrcollections.abcrrZ REDUCTIONSZ_unary_math_opsZ_binary_math_opsrr;r$r(rr~rrrgtltgeleeqneZ_cmp_ops_funcsdictziprrand_or_Z_bool_ops_funcsrZARITH_OPS_SYMSaddsubmultruedivpowfloordivmodZ_arith_ops_funcsZ_arith_ops_dictZSPECIAL_CASE_ARITH_OPS_SYMSZ_special_case_arith_ops_funcsZ_special_case_arith_ops_dictrdrUrrrrposneginvertZ_unary_ops_funcsrrrrr3r3r3r4s     t 6     1