U f/e ?@szdZddlmZddlmZddlmZddlZddlmZm Z ddl Z ddl m Z ddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZd Zd ZdZeeZ dZ!Gddde"Z#GdddZ$Gddde$Z%ddddZ&GdddZ'ddZ(ddZ)d Z*ej+ej,ej-ej.ej/ej0e(e)fZ1e2e3e*e1Z4d!Z5ej6ej7ej6ej7fZ8e2e3e5e8Z9d"Z:ej;ejej?ej@ejAfZBe2e3e:eBZCd#ZDej?ej@ejAfZEe2e3eDeEZFiZGe4e9eCfD]ZHeGIeHqd$d%ZJd&d'd(d)ZKGd*d+d+e'ZLd&d'd,d-ZMGd.d/d/eLZNd0ZOejPejQejRejRfZSe2e3eOeSZTGd1d2d2e'ZUGd3d4d4e'ZVGd5d6d6ZWdS)7z Operator classes for eval. ) annotations)datetime)partialN)CallableIterable) Timestamp) is_list_like is_scalar)ensure_decodedresult_type_many)DEFAULT_GLOBALS) pprint_thingpprint_thing_encoded)sumprod)sincosexplogexpm1log1psqrtsinhcoshtanhZarcsinZarccosZarctanZarccoshZarcsinhZarctanhabslog10floorceil)Zarctan2Z__pd_eval_local_cs*eZdZdZddddfdd ZZS) UndefinedVariableErrorz1 NameError subclass for local variables. Nstrz bool | None)nameis_localcs8t|d}|rd|}n d|}t|dS)Nz is not definedzlocal variable zname )reprsuper__init__)selfr!r"Zbase_msgmsg __class__?/tmp/pip-unpacked-wheel-tiezk1ph/pandas/core/computation/ops.pyr%Is   zUndefinedVariableError.__init__)N__name__ __module__ __qualname____doc__r% __classcell__r*r*r(r+rDsrcseZdZUd&fdd Zded<d'ddZedd d d Zdd d d ZddZ ddZ ddZ ddZ edd ddZ eddZeZedd ddZedd ddZeddZejd dZed!d"Zed#d d$d%ZZS)(TermNcs&t|tstn|}tt|j}||SN) isinstancer Constantr$r2__new__)clsr!envsideencodingklassZsupr_newr(r*r+r6Ss z Term.__new__boolr"cCsB||_||_||_t|}|tp*|tk|_||_ ||_ dSr3) _namer8r9r startswith LOCAL_TAGr r" _resolve_name_valuer:)r&r!r8r9r:Ztnamer*r*r+r%[s z Term.__init__r returncCs|jtdS)N)r!replacer?r&r*r*r+ local_nameeszTerm.local_namecCs t|jSr3)r r!rFr*r*r+__repr__isz Term.__repr__cOs|jSr3valuer&argskwargsr*r*r+__call__lsz Term.__call__cOs|Sr3r*rKr*r*r+evaluateosz Term.evaluatecCs>|jj|j|jd}||t|dr:|jdkr:td|S)N)r"ndimz?N-dimensional objects, where N > 2, are not supported with eval)r8resolverGr"updatehasattrrPNotImplementedError)r&resr*r*r+r@rs 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!r4r r8ZswapkeyrGrJ)r&rJkeyr*r*r+rS|s  z Term.updatecCs t|jSr3)r rArFr*r*r+r szTerm.is_scalarc CsXz |jjjWStk rRz|jjWYStk rLt|jYYSXYnXdSr3)rAvaluesdtypeAttributeErrortyperFr*r*r+r\s z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))r\r-r#r!rFr*r*r+rawszTerm.rawcCs8z |jj}Wntk r&|j}YnXt|ttjfSr3)r\r[ issubclassrnp datetime64r&tr*r*r+ is_datetimes   zTerm.is_datetimecCs|jSr3rArFr*r*r+rJsz Term.valuecCs ||_dSr3re)r&rWr*r*r+rJscCs|jSr3r=rFr*r*r+r!sz Term.nameintcCs|jjSr3)rArPrFr*r*r+rPsz Term.ndim)NN)NN)r-r.r/r6__annotations__r%propertyrGrHrNrOr@rSr r\ return_typer^rdrJsetterr!rPr1r*r*r(r+r2Rs6       r2cs@eZdZd fdd ZddZeddZdd d d ZZS) r5Ncstj||||ddS)N)r9r:)r$r%)r&rJr8r9r:r(r*r+r%szConstant.__init__cCs|jSr3rfrFr*r*r+r@szConstant._resolve_namecCs|jSr3rIrFr*r*r+r!sz Constant.namer rBcCs t|jSr3)r#r!rFr*r*r+rHszConstant.__repr__)NN) r-r.r/r%r@rir!rHr1r*r*r(r+r5s  r5~&|)notandorc@seZdZUdZded<ddddddZd d Zdd d d ZeddZ edd ddZ eddZ edd ddZ edd ddZ dS)Opz. Hold an operator of arbitrary arity. r opNzIterable[Term | Op])rsoperandscCst|||_||_||_dSr3) _bool_op_mapgetrsrtr:)r&rsrtr:r*r*r+r%sz Op.__init__cCs t|jSr3)iterrtrFr*r*r+__iter__sz Op.__iter__rBcCs(dd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. css|]}dt|dVqdS)(r]N)r ).0Zoprr*r*r+ szOp.__repr__.. )rtr rsjoin)r&Zparenedr*r*r+rHsz Op.__repr__cCs,|jttkrtjStddt|DS)Ncss|] }|jVqdSr3r\rztermr*r*r+r{sz!Op.return_type..)rs CMP_OPS_SYMS BOOL_OPS_SYMSr`bool_r comflattenrFr*r*r+rjszOp.return_typer<cCs(|j}ttdg}|jtko&||S)Nobject) operand_types frozensetr`rZrjr)r&typesZ obj_dtype_setr*r*r+has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncss|] }|jVqdSr3r~rr*r*r+r{sz#Op.operand_types..)rrrrFr*r*r+rszOp.operand_typescCstdd|jDS)Ncss|] }|jVqdSr3)r )rzoperandr*r*r+r{szOp.is_scalar..)allrtrFr*r*r+r sz Op.is_scalarcCs8z |jj}Wntk r&|j}YnXt|ttjfSr3)rjr\r[r_rr`rarbr*r*r+rds   zOp.is_datetime)N)r-r.r/r0rhr%rxrHrirjrrr rdr*r*r*r+rrs   rrc Cs\z ||WStk rVt|rJz||WYStk rHYnX||kYSXdS)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. Nisinr[rxyr*r*r+_in s 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*r+_not_insr)>=z<=z==z!=inznot in)rmrnrprq)+-*/**//%)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`rZr\rJZastyper[rS)Ztermsacceptable_dtypesrZdtrrWr*r*r+ _cast_inplaceRs  rr<rBcCs t|tSr3)r4r2)objr*r*r+is_termksrcsLeZdZdZddfdd ZddZddd d Zd d Zd dZZ S)BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op r rsc 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 ValueErrorr#)r&rsrrerrrr(r*r+r%zs 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&r8leftrightr*r*r+rNs  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_dictrrr8) rrOrrsrrJZpandas.core.computation.evalrZadd_tmp) r&r8rrrrrVrrrr!r*r*r+rOs,    zBinOp.evaluatecsfdd}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 )rJencoderrFr*r+ stringifys z'BinOp.convert_values..stringifyNUTC)rrrrdr rJr4rgfloatrr tzZ tz_convertrS)r&rrrvr*rFr+rs$        zBinOp.convert_valuescCsr|j}|j}|j}t|d|}|j}t|d|}|js<|jrn|jtkrnt|tt j frft|tt j fsnt ddS)Nr\z$cannot evaluate scalar only bool ops) rrrjgetattrr rs_bool_ops_dictr_r<r`rrU)r&rrZrhs_rtZlhs_rtr*r*r+rs"   z$BinOp._disallow_scalar_only_bool_ops) r-r.r/r0r%rNrOrrr1r*r*r(r+ros  1!rcCstt|jtjSr3)r_r`rZr\number)rZr*r*r+ isnumericsrcs eZdZdZfddZZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. csjtd||t|jr$t|jsFtd|jd|jd|jdtjtjg}t t ||tjdS)Nrz unsupported operand type(s) for z: 'z' and '') r$r%rrj TypeErrorrsr`Zfloat32Zfloat_rrr)r&rrrr(r*r+r%s z Div.__init__r,r*r*r(r+r s r)rrrlrocsNeZdZdZddfdd ZddZddd d Zed dd d ZZ 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. r rc sft||f||_zt||_Wn<tk r`}ztdt|dt|W5d}~XYnXdS)NzInvalid unary operator r) r$r%r_unary_ops_dictrrrr#UNARY_OPS_SYMS)r&rsrrr(r*r+r%9szUnaryOp.__init__cCs||}||Sr3)rr)r&r8rr*r*r+rNEs zUnaryOp.__call__rBcCst|jd|jdS)Nryr])r rsrrFr*r*r+rHIszUnaryOp.__repr__znp.dtypecCsR|j}|jtdkr tdSt|trH|jtks>|jtkrHtdStdS)Nr<rg) rrjr`rZr4rrrs _cmp_ops_dictr)r&rr*r*r+rjLs   zUnaryOp.return_type) r-r.r/r0r%rNrHrirjr1r*r*r(r+r(s  rcs2eZdZfddZddZddddZZS) MathCallcst|j|||_dSr3)r$r%r!r)r&rrLr(r*r+r%YszMathCall.__init__c sDfdd|jD}tjdd|jj|W5QRSQRXdS)Ncsg|] }|qSr*r*)rzrsrr*r+ _sz%MathCall.__call__..ignore)r)rtr`Zerrstater)r&r8rtr*rr+rN]szMathCall.__call__r rBcCs(tt|j}t|jdd|dS)Nry,r])mapr rtr rsr})r&rtr*r*r+rHcs zMathCall.__repr__)r-r.r/r%rNrHr1r*r*r(r+rXs rc@s"eZdZddddZddZdS)FuncNoder )r!cCs.|tkrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr!rr`r)r&r!r*r*r+r%iszFuncNode.__init__cGs t||Sr3)r)r&rLr*r*r+rNoszFuncNode.__call__N)r-r.r/r%rNr*r*r*r+rhsr)Xr0 __future__rr functoolsroperatortypingrrZnumpyr`Zpandas._libs.tslibsrZpandas.core.dtypes.commonrr Zpandas.core.commoncorecommonrZpandas.core.computation.commonr r Zpandas.core.computation.scoper Zpandas.io.formats.printingr rZ REDUCTIONSZ_unary_math_opsZ_binary_math_opsrr? NameErrorrr2r5rurrrrrgtltgeleeqneZ_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_dictrdrSrrrrrrposneginvertZ_unary_ops_funcsrrrrr*r*r*r+s     m 6  0