U cF?@sndZddlmZddlmZddlmZddlZddlmZm Z m Z ddl Z ddl mZddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZd Zd ZdZ ee Z!dZ"GdddZ#Gddde#Z$ddddZ%GdddZ&ddZ'ddZ(dZ)ej*ej+ej,ej-ej.ej/e'e(fZ0e1e2e)e0Z3dZ4ej5ej6ej5ej6fZ7e1e2e4e7Z8d Z9ej:ej;ejej?ej@fZAe1e2e9eAZBd!ZCej>ej?ej@fZDe1e2eCeDZEiZFe3e8eBfD]ZGeFHeGqd"d#ZId$d%d&d'ZJGd(d)d)e&ZKd$d%d*d+ZLGd,d-d-eKZMd.ZNejOejPejQejQfZRe1e2eNeRZSGd/d0d0e&ZTGd1d2d2e&ZUGd3d4d4ZVdS)5z Operator classes for eval. ) annotations)datetime)partialN)CallableIterableLiteral) Timestamp) is_list_like is_scalar)ensure_decodedresult_type_many)DEFAULT_GLOBALS) pprint_thingpprint_thing_encoded)sumprod)sincosexplogexpm1log1psqrtsinhcoshtanhZarcsinZarccosZarctanZarccoshZarcsinhZarctanhabslog10floorceil)Zarctan2Z__pd_eval_local_cseZdZUd'fdd Zded<d(dddd Zed dd d Zd dd dZddZ ddddZ ddZ ddddZ eddddZ eddZeZed dddZeddddZedd Zejd!d Zed"d#Zed$dd%d&ZZS))TermNcs&t|tstn|}tt|j}||SN) isinstancestrConstantsuperr __new__)clsnameenvsideencodingklassZsupr_new __class__?/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/computation/ops.pyr&Fs z Term.__new__boolis_localNonereturncCsB||_||_||_t|}|tp*|tk|_||_ ||_ dSr!) _namer)r*r# startswith LOCAL_TAGr r2 _resolve_name_valuer+)selfr(r)r*r+Ztnamer/r/r0__init__Ns z Term.__init__r#cCs|jtdS)N)r(replacer8r;r/r/r0 local_nameXszTerm.local_namecCs t|jSr!)rr(r?r/r/r0__repr__\sz Term.__repr__cOs|jSr!valuer;argskwargsr/r/r0__call___sz Term.__call__cOs|Sr!r/rDr/r/r0evaluatebsz Term.evaluatecCslt|j}|j}||jjkr2t|jj|tr2d}|jj||d}||t |drh|j dkrht d|S)NF)r2ndimz?N-dimensional objects, where N > 2, are not supported with eval) r#r@r2r)Zscoper"typeresolveupdatehasattrrINotImplementedError)r;r@r2resr/r/r0r9es   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)Zswapkeyr@rC)r;rCkeyr/r/r0rMvs  z Term.updatecCs t|jSr!)r r:r?r/r/r0r szTerm.is_scalarc CsXz |jjjWStk rRz|jjWYStk rLt|jYYSXYnXdSr!)r:valuesdtypeAttributeErrorrKr?r/r/r0rKs z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rK__name__reprr(r?r/r/r0rawszTerm.rawcCs8z |jj}Wntk r&|j}YnXt|ttjfSr!)rKrU issubclassrnp datetime64r;tr/r/r0 is_datetimes   zTerm.is_datetimecCs|jSr!r:r?r/r/r0rCsz Term.valuecCs ||_dSr!r`)r;rQr/r/r0rCscCs|jSr!r6r?r/r/r0r(sz Term.nameintcCs|jjSr!)r:rIr?r/r/r0rIsz Term.ndim)NN)NN)rW __module__ __qualname__r&__annotations__r<propertyr@rArGrHr9rMr rK return_typerYr_rCsetterr(rI __classcell__r/r/r-r0r Es6      r csFeZdZd ddfdd ZddZedd Zd dd d ZZS)r$Nr3r4cstj||||ddS)N)r*r+)r%r<)r;rCr)r*r+r-r/r0r<szConstant.__init__cCs|jSr!rar?r/r/r0r9szConstant._resolve_namecCs|jSr!rBr?r/r/r0r(sz Constant.namer#cCs t|jSr!)rXr(r?r/r/r0rAszConstant.__repr__)NN) rWrcrdr<r9rfr(rArir/r/r-r0r$s  r$~&|)notandorc@seZdZUdZded<ddddddd Zd d Zdd d dZeddZ edd ddZ eddZ edd ddZ edd ddZ dS)Opz. Hold an operator of arbitrary arity. r#opNzIterable[Term | Op]r3)rqoperandsr5cCst|||_||_||_dSr!) _bool_op_mapgetrqrrr+)r;rqrrr+r/r/r0r<sz Op.__init__cCs t|jSr!)iterrrr?r/r/r0__iter__sz Op.__iter__r4cCs(dd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. css|]}dt|dVqdS)(rVN)r).0Zoprr/r/r0 szOp.__repr__.. )rrrrqjoin)r;Zparenedr/r/r0rAsz Op.__repr__cCs,|jttkrtjStddt|DS)Ncss|] }|jVqdSr!rKrxtermr/r/r0rysz!Op.return_type..)rq CMP_OPS_SYMS BOOL_OPS_SYMSr[bool_r comflattenr?r/r/r0rgszOp.return_typer1cCs(|j}ttdg}|jtko&||S)Nobject) operand_types frozensetr[rTrgr)r;typesZ obj_dtype_setr/r/r0has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncss|] }|jVqdSr!r|r}r/r/r0rysz#Op.operand_types..)rrrr?r/r/r0rszOp.operand_typescCstdd|jDS)Ncss|] }|jVqdSr!)r )rxoperandr/r/r0ryszOp.is_scalar..)allrrr?r/r/r0r sz Op.is_scalarcCs8z |jj}Wntk r&|j}YnXt|ttjfSr!)rgrKrUrZrr[r\r]r/r/r0r_s   zOp.is_datetime)N)rWrcrd__doc__rer<rvrArfrgrrr r_r/r/r/r0rps   rpc Cs\z ||WStk rVt|rJz||WYStk rHYnX||kYSXdS)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. NisinrUr xyr/r/r0_ins rc Cs`z||WStk rZt|rNz||WYStk rLYnX||kYSXdS)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. Nrrr/r/r0_not_insr)>=z<=z==z!=inznot in)rkrlrnro)+-*/**//%)rrrc Cs`t|}|D]L}|j|krqz|j|}Wn tk rN||j}YnX||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[rTrKrCZastyperUrM)Ztermsacceptable_dtypesrTdtr~rQr/r/r0 _cast_inplaceLs  rr1r4cCs t|tSr!)r"r )objr/r/r0is_termesrcsTeZdZdZdddfdd 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#r3rqr5c st|||f||_||_||zt||_WnHtk r}z*t t }t dt |d||W5d}~XYnXdS)NzInvalid binary operator , valid operators are ) r%r<lhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorrX)r;rqrrerrrr-r/r0r<ts 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)leftrightr/r/r0rGs  zBinOp.__call__)enginec Cs|dkr||}nd|jj|||||d}|jj|||||d}|j|krZ||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)) rrHrrqrrCZpandas.core.computation.evalrZadd_tmp) r;r)rrrrrPrrrr(r/r/r0rHs,    zBinOp.evaluater4csfdd}jj}}t|r~|jr~t|r~|jr~|j}t|ttfrR||}t t |}|j dk rr| d}j |t|r|jrt|r|jr|j}t|ttfr||}t t |}|j dk r| d}j |dS)zK Convert datetimes to a comparable value in an expression. cs&jdk rttjd}nt}||S)N)r+)r+rrr)rCencoderr?r/r0 stringifys z'BinOp.convert_values..stringifyNUTC)rrrr_r rCr"rbfloatrr tzZ tz_convertrM)r;rrrvr/r?r0rs$        zBinOp.convert_valuescCsr|j}|j}|j}t|d|}|j}t|d|}|js<|jrn|jtkrnt|tt j frft|tt j fsnt ddS)NrKz$cannot evaluate scalar only bool ops) rrrggetattrr rq_bool_ops_dictrZr1r[rrO)r;rrZrhs_rtZlhs_rtr/r/r0rs"   z$BinOp._disallow_scalar_only_bool_ops) rWrcrdrr<rGrHrrrir/r/r-r0ris  1!rcCstt|jtjSr!)rZr[rTrKnumber)rTr/r/r0 isnumericsrcs&eZdZdZddfdd ZZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. r3r4csjtd||t|jr$t|jsFtd|jd|jd|jdtjtjg}t t ||tjdS)Nrz unsupported operand type(s) for z: 'z' and '') r%r<rrg TypeErrorrqr[Zfloat32Zfloat_rrr)r;rrrr-r/r0r<s z Div.__init__)rWrcrdrr<rir/r/r-r0rs r)rrrjrmcsVeZdZdZdddfdd 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')]r3rc sft||f||_zt||_Wn<tk r`}ztdt|dt|W5d}~XYnXdS)NzInvalid unary operator r) r%r<r_unary_ops_dictrrrrXUNARY_OPS_SYMS)r;rqrrr-r/r0r<3szUnaryOp.__init__MathCallr4cCs||}||Sr!)rr)r;r)rr/r/r0rG?s zUnaryOp.__call__r#cCst|jd|jdS)NrwrV)rrqrr?r/r/r0rADszUnaryOp.__repr__znp.dtypecCsR|j}|jtdkr tdSt|trH|jtks>|jtkrHtdStdS)Nr1rb) rrgr[rTr"rprq _cmp_ops_dictr)r;rr/r/r0rgGs   zUnaryOp.return_type) rWrcrdrr<rGrArfrgrir/r/r-r0r"s  rcs8eZdZddfdd ZddZdddd ZZS) rr3r4cst|j|||_dSr!)r%r<r(r)r;rrEr-r/r0r<TszMathCall.__init__c sDfdd|jD}tjdd|jj|W5QRSQRXdS)Ncsg|] }|qSr/r/)rxrqrr/r0 Zsz%MathCall.__call__..ignore)r)rrr[Zerrstater)r;r)rrr/rr0rGXszMathCall.__call__r#cCs(tt|j}t|jdd|dS)Nrw,rV)mapr#rrrrqr{)r;rrr/r/r0rA^s zMathCall.__repr__)rWrcrdr<rGrArir/r/r-r0rSsrc@s$eZdZdddddZddZdS) FuncNoder#r3)r(r5cCs.|tkrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr(rr[r)r;r(r/r/r0r<dszFuncNode.__init__cGs t||Sr!)r)r;rEr/r/r0rGjszFuncNode.__call__N)rWrcrdr<rGr/r/r/r0rcsr)Wr __future__rr functoolsroperatortypingrrrZnumpyr[Zpandas._libs.tslibsrZpandas.core.dtypes.commonr r Zpandas.core.commoncorecommonrZpandas.core.computation.commonr r Zpandas.core.computation.scoper Zpandas.io.formats.printingrrZ REDUCTIONSZ_unary_math_opsZ_binary_math_opsrr8r r$rsrprrrgtltgeleeqneZ_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_dictrdrMrrrrrrposneginvertZ_unary_ops_funcsrrrrr/r/r/r0s     t 6  1