a Of,?@sdZddlmZddlmZddlmZddlZddlmZm Z m Z ddl Z ddl mZddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZerdd 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@ejAejBejCfZDe4e5e 2, are not supported with eval) r'rDr6r-scoper&typeresolveupdatehasattrrMNotImplementedError)r?rDr6resr3r3r4r=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-ZswapkeyrDrG)r?rGkeyr3r3r4rR|s  z Term.updatecCs t|jSr%)r r>rCr3r3r4r szTerm.is_scalarc CsTz |jjjWStyNz|jjWYStyHt|jYYS0Yn0dSr%)r>valuesdtypeAttributeErrorrPrCr3r3r4rPs   z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rP__name__reprr,rCr3r3r4rawszTerm.rawcCs6z |jj}Wnty$|j}Yn0t|ttjfSr%)rPrZ issubclassrnp datetime64r?tr3r3r4 is_datetimes    zTerm.is_datetimecCs|jSr%r>rCr3r3r4rGsz Term.valuecCs ||_dSr%re)r?rVr3r3r4rGscCs|jSr%r:rCr3r3r4r,sz Term.nameintcCs|jjSr%)r>rMrCr3r3r4rMsz Term.ndim)NN)NN)r\ __module__ __qualname__r*__annotations__r@propertyrDrErKrLr=rRr rP return_typer^rdrGsetterr,rM __classcell__r3r3r1r4r$Ks6     r$c@s.eZdZddZeddZddddZd S) r(cCs|jSr%rfrCr3r3r4r=szConstant._resolve_namecCs|jSr%rFrCr3r3r4r,sz Constant.namer'r8cCs t|jSr%)r]r,rCr3r3r4rEszConstant.__repr__N)r\rhrir=rkr,rEr3r3r3r4r(s r(~&|)notandorc@seZdZUdZded<ddddddd Zd d d d Zdd ddZeddZ edd ddZ eddZ edd ddZ edd ddZ dS)Opz. Hold an operator of arbitrary arity. r'opNzIterable[Term | Op]r7)rvoperandsr9cCst|||_||_||_dSr%) _bool_op_mapgetrvrwr/)r?rvrwr/r3r3r4r@sz Op.__init__rr8cCs t|jSr%)iterrwrCr3r3r4__iter__sz 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 zOp.__repr__.. )rwrrvjoin)r?Zparenedr3r3r4rEsz Op.__repr__cCs,|jttvrtjStddt|DS)Ncss|] }|jVqdSr%rPr}termr3r3r4r~rz!Op.return_type..)rv CMP_OPS_SYMS BOOL_OPS_SYMSr`bool_r comflattenrCr3r3r4rlszOp.return_typer5cCs(|j}ttdg}|jtko&||S)Nobject) operand_types frozensetr`rYrlr)r?typesZ obj_dtype_setr3r3r4has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncss|] }|jVqdSr%rrr3r3r4r~rz#Op.operand_types..)rrrrCr3r3r4rszOp.operand_typescCstdd|jDS)Ncss|] }|jVqdSr%)r )r}operandr3r3r4r~rzOp.is_scalar..)allrwrCr3r3r4r sz Op.is_scalarcCs6z |jj}Wnty$|j}Yn0t|ttjfSr%)rlrPrZr_rr`rarbr3r3r4rds    zOp.is_datetime)N)r\rhri__doc__rjr@r{rErkrlrrr rdr3r3r3r4rus   ruc CsXz ||WStyRt|rFz||WYStyDYn0||vYS0dS)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. NisinrZr xyr3r3r4_ins   rc Cs\z||WStyVt|rJz||WYStyHYn0||vYS0dS)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. Nrrr3r3r4_not_ins  r)>=z<=z==z!=inznot in)rprqrsrt)+-*/**//%)rrrr7r8c Cs^t|}|D]J}|j|vrqz|j|}WntyL||j}Yn0||qdS)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)r`rYrPrGZastyperZrR)Ztermsacceptable_dtypesrYdtrrVr3r3r4 _cast_inplaceOs   rr5cCs t|tSr%)r&r$)objr3r3r4is_termhsrcsTeZdZdZdddfdd ZddZdd d d Zdd d dZddZZ S)BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op r'r7rvr9c st|||f||_||_||zt||_WnJty}z2t t }t dt |d||WYd}~n d}~00dS)NzInvalid binary operator , valid operators are ) r)r@lhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorr])r?rvrrerrrr1r3r4r@ws 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-leftrightr3r3r4rKs  zBinOp.__call__)enginec Cs|dkr||}nd|jj|||||d}|jj|||||d}|j|vrZ||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-) rrLrrvrrGZpandas.core.computation.evalrZadd_tmp) r?r-rrrrrUrrrr,r3r3r4rLs,    zBinOp.evaluater8csfdd}jj}}t|r~|jr~t|r~|jr~|j}t|ttfrR||}t t |}|j durr| d}j |t|r|jrt|r|jr|j}t|ttfr||}t t |}|j dur| d}j |dS)zK Convert datetimes to a comparable value in an expression. cs&jdurttjd}nt}||S)N)r/)r/rrr)rGencoderrCr3r4 stringifys z'BinOp.convert_values..stringifyNUTC)rrrrdr rGr&rgfloatrr tzZ tz_convertrR)r?rrrvr3rCr4rs$        zBinOp.convert_valuescCsr|j}|j}|j}t|d|}|j}t|d|}|js<|jrn|jtvrnt|tt j frft|tt j fsnt ddS)NrPz$cannot evaluate scalar only bool ops) rrrlgetattrr rv_bool_ops_dictr_r5r`rrT)r?rrZrhs_rtZlhs_rtr3r3r4rs"   z$BinOp._disallow_scalar_only_bool_ops) r\rhrirr@rKrLrrrnr3r3r1r4rls  1!rcCstt|jtjSr%)r_r`rYrPnumber)rYr3r3r4 isnumericsrcs&eZdZdZddfdd ZZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. r7r8csjtd||t|jr$t|jsFtd|jd|jd|jdtjtjg}t t ||tjdS)Nrz unsupported operand type(s) for z: 'z' and '') r)r@rrl TypeErrorrvr`Zfloat32Zfloat64rrr)r?rrrr1r3r4r@s  z Div.__init__)r\rhrirr@rnr3r3r1r4rs r)rrrorrcsVeZdZdZdddfdd Zddd d Zd dd d ZeddddZZ 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. zLiteral['+', '-', '~', 'not']r7rc sht||f||_zt||_Wn>tyb}z&tdt|dt|WYd}~n d}~00dS)NzInvalid unary operator r) r)r@r_unary_ops_dictrrrr]UNARY_OPS_SYMS)r?rvrrr1r3r4r@6s zUnaryOp.__init__MathCallr8cCs||}||Sr%)rr)r?r-rr3r3r4rKBs zUnaryOp.__call__r'cCst|jd|jdS)Nr|r[)rrvrrCr3r3r4rEGszUnaryOp.__repr__znp.dtypecCsR|j}|jtdkr tdSt|trH|jtvs>|jtvrHtdStdS)Nr5rg) rrlr`rYr&rurv _cmp_ops_dictr)r?rr3r3r4rlJs   zUnaryOp.return_type) r\rhrirr@rKrErkrlrnr3r3r1r4r%s  rcs8eZdZddfdd ZddZdddd ZZS) rr7r8cst|j|||_dSr%)r)r@r,r)r?rrIr1r3r4r@WszMathCall.__init__cs fdd|jD}|jj|S)Ncsg|] }|qSr3r3)r}rvrr3r4 ]rz%MathCall.__call__..)rwr)r?r-rwr3rr4rK[szMathCall.__call__r'cCs(tt|j}t|jdd|dS)Nr|,r[)mapr'rwrrvr)r?rwr3r3r4rE`s zMathCall.__repr__)r\rhrir@rKrErnr3r3r1r4rVsrc@s*eZdZdddddZdddd Zd S) FuncNoder'r7)r,r9cCs.|tvrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr,rr`r)r?r,r3r3r4r@fszFuncNode.__init__rr8cGs t||Sr%)r)r?rIr3r3r4rKlszFuncNode.__call__N)r\rhrir@rKr3r3r3r4resr)Zr __future__rr functoolsroperatortypingrrrnumpyr`Zpandas._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(rxrurrrgtltgeleeqneZ_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_dictrdrRrrrrrrposneginvertZ_unary_ops_funcsrrrrr3r3r3r4s     t 6  1