U MZf~?@sxdZddlmZddlmZddlmZddlZddlmZm Z m Z m Z ddl Z ddlmZddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZd Zd Z dZ!e e!Z"dZ#GdddZ$Gddde$Z%ddddZ&GdddZ'ddZ(ddZ)dZ*ej+ej,ej-ej.ej/ej0e(e)fZ1e2e3e*e1Z4dZ5ej6ej7ej6ej7fZ8e2e3e5e8Z9d Z:ej;ejej?ej@ejAfZBe2e3e:eBZCd!ZDej?ej@ejAfZEe2e3eDeEZFiZGe4e9eCfD]ZHeGIeHqd"d#d$d%ZJd&d#d'd(ZKGd)d*d*e'ZLd&d#d+d,ZMGd-d.d.eLZNd/ZOejPejQejRejRfZSe2e3eOeSZTGd0d1d1e'ZUGd2d3d3e'ZVGd4d5d5ZWdS)6z Operator classes for eval. ) annotations)datetime)partialN)CallableIterableIteratorLiteral) Timestamp) is_list_like is_scalar)ensure_decodedresult_type_many)DEFAULT_GLOBALS) pprint_thingpprint_thing_encoded)sumprodminmax)sincosexplogexpm1log1psqrtsinhcoshtanhZarcsinZarccosZarctanZarccoshZarcsinhZarctanhabslog10floorceil)Zarctan2Z__pd_eval_local_cseZdZUd'fdd Zded<d(dddd Zed dd d Zd dd dZddZ ddddZ ddZ ddddZ eddddZ eddZeZed dddZeddddZedd Zejddd!d Zed"d#Zed$dd%d&ZZS))TermNcs&t|tstn|}tt|j}||SN) isinstancestrConstantsuperr#__new__)clsnameenvsideencodingklassZsupr_new __class__?/tmp/pip-unpacked-wheel-nbcvw55c/pandas/core/computation/ops.pyr)Gs z Term.__new__boolis_localNonereturncCsB||_||_||_t|}|tp*|tk|_||_ ||_ dSr$) _namer,r-r& startswith LOCAL_TAGrr5 _resolve_name_valuer.)selfr+r,r-r.Ztnamer2r2r3__init__Os z Term.__init__r&cCs|jtdS)N)r+replacer;r>r2r2r3 local_nameYszTerm.local_namecCs t|jSr$)rr+rBr2r2r3__repr__]sz Term.__repr__cOs|jSr$valuer>argskwargsr2r2r3__call__`sz Term.__call__cOs|Sr$r2rGr2r2r3evaluatecsz Term.evaluatecCslt|j}|j}||jjkr2t|jj|tr2d}|jj||d}||t |drh|j dkrht d|S)NF)r5ndimz?N-dimensional objects, where N > 2, are not supported with eval) r&rCr5r,Zscoper%typeresolveupdatehasattrrLNotImplementedError)r>rCr5resr2r2r3r<fs   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,ZswapkeyrCrF)r>rFkeyr2r2r3rPws  z Term.updatecCs t|jSr$)r r=rBr2r2r3r szTerm.is_scalarc CsXz |jjjWStk rRz|jjWYStk rLt|jYYSXYnXdSr$)r=valuesdtypeAttributeErrorrNrBr2r2r3rNs z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rN__name__reprr+rBr2r2r3rawszTerm.rawcCs8z |jj}Wntk r&|j}YnXt|ttjfSr$)rNrX issubclassrnp datetime64r>tr2r2r3 is_datetimes   zTerm.is_datetimecCs|jSr$r=rBr2r2r3rFsz Term.valuecCs ||_dSr$rc)r>rTr2r2r3rFscCs|jSr$r9rBr2r2r3r+sz Term.nameintcCs|jjSr$)r=rLrBr2r2r3rLsz Term.ndim)NN)NN)rZ __module__ __qualname__r)__annotations__r?propertyrCrDrJrKr<rPr rN return_typer\rbrFsetterr+rL __classcell__r2r2r0r3r#Fs6     r#csFeZdZd ddfdd ZddZedd Zd dd d ZZS)r'Nr6r7cstj||||ddS)N)r-r.)r(r?)r>rFr,r-r.r0r2r3r?szConstant.__init__cCs|jSr$rdrBr2r2r3r<szConstant._resolve_namecCs|jSr$rErBr2r2r3r+sz Constant.namer&cCs t|jSr$)r[r+rBr2r2r3rDszConstant.__repr__)NN) rZrfrgr?r<rir+rDrlr2r2r0r3r'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]r6)rtoperandsr8cCst|||_||_||_dSr$) _bool_op_mapgetrtrur.)r>rtrur.r2r2r3r?sz Op.__init__rr7cCs t|jSr$)iterrurBr2r2r3__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)(rYN)r).0Zoprr2r2r3 szOp.__repr__.. )rurrtjoin)r>Zparenedr2r2r3rDsz Op.__repr__cCs,|jttkrtjStddt|DS)Ncss|] }|jVqdSr$rNr{termr2r2r3r|sz!Op.return_type..)rt CMP_OPS_SYMS BOOL_OPS_SYMSr^bool_r comflattenrBr2r2r3rjszOp.return_typer4cCs(|j}ttdg}|jtko&||S)Nobject) operand_types frozensetr^rWrjr)r>typesZ obj_dtype_setr2r2r3has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncss|] }|jVqdSr$rrr2r2r3r|sz#Op.operand_types..)rrrrBr2r2r3rszOp.operand_typescCstdd|jDS)Ncss|] }|jVqdSr$)r )r{operandr2r2r3r|szOp.is_scalar..)allrurBr2r2r3r sz Op.is_scalarcCs8z |jj}Wntk r&|j}YnXt|ttjfSr$)rjrNrXr]rr^r_r`r2r2r3rbs   zOp.is_datetime)N)rZrfrg__doc__rhr?ryrDrirjrrr rbr2r2r2r3rss   rsc Cs\z ||WStk rVt|rJz||WYStk rHYnX||kYSXdS)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. NisinrXr xyr2r2r3_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. Nrrr2r2r3_not_insr)>=z<=z==z!=inznot in)rnrorqrr)+-*/**//%)rrrr6r7c 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^rWrNrFZastyperXrP)Ztermsacceptable_dtypesrWdtrrTr2r2r3 _cast_inplaceMs  rr4cCs t|tSr$)r%r#)objr2r2r3is_termfsrcsTeZdZdZdddfdd 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&r6rtr8c 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>rtrrerrrr0r2r3r?us 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,leftrightr2r2r3rJs  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,) rrKrrtrrFZpandas.core.computation.evalrZadd_tmp) r>r,rrrrrSrrrr+r2r2r3rKs,    zBinOp.evaluater7csfdd}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)rFencoderrBr2r3 stringifys z'BinOp.convert_values..stringifyNUTC)rrrrbr rFr%refloatr r tzZ tz_convertrP)r>rrrvr2rBr3rs$        zBinOp.convert_valuescCsr|j}|j}|j}t|d|}|j}t|d|}|js<|jrn|jtkrnt|tt j frft|tt j fsnt ddS)NrNz$cannot evaluate scalar only bool ops) rrrjgetattrr rt_bool_ops_dictr]r4r^rrR)r>rrZrhs_rtZlhs_rtr2r2r3rs"   z$BinOp._disallow_scalar_only_bool_ops) rZrfrgrr?rJrKrrrlr2r2r0r3rjs  1!rcCstt|jtjSr$)r]r^rWrNnumber)rWr2r2r3 isnumericsrcs&eZdZdZddfdd ZZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. r6r7csjtd||t|jr$t|jsFtd|jd|jd|jdtjtjg}t t ||tjdS)Nrz unsupported operand type(s) for z: 'z' and '') r(r?rrj TypeErrorrtr^Zfloat32Zfloat_rrr)r>rrrr0r2r3r?s z Div.__init__)rZrfrgrr?rlr2r2r0r3rs r)rrrmrpcsVeZdZdZdddfdd 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')]r6rc sft||f||_zt||_Wn<tk r`}ztdt|dt|W5d}~XYnXdS)NzInvalid unary operator r) r(r?r_unary_ops_dictrrrr[UNARY_OPS_SYMS)r>rtrrr0r2r3r?4szUnaryOp.__init__MathCallr7cCs||}||Sr$)rr)r>r,rr2r2r3rJ@s zUnaryOp.__call__r&cCst|jd|jdS)NrzrY)rrtrrBr2r2r3rDEszUnaryOp.__repr__znp.dtypecCsR|j}|jtdkr tdSt|trH|jtks>|jtkrHtdStdS)Nr4re) rrjr^rWr%rsrt _cmp_ops_dictr)r>rr2r2r3rjHs   zUnaryOp.return_type) rZrfrgrr?rJrDrirjrlr2r2r0r3r#s  rcs8eZdZddfdd ZddZdddd ZZS) rr6r7cst|j|||_dSr$)r(r?r+r)r>rrHr0r2r3r?UszMathCall.__init__c sDfdd|jD}tjdd|jj|W5QRSQRXdS)Ncsg|] }|qSr2r2)r{rtrr2r3 [sz%MathCall.__call__..ignore)r)rur^Zerrstater)r>r,rur2rr3rJYszMathCall.__call__r&cCs(tt|j}t|jdd|dS)Nrz,rY)mapr&rurrtr~)r>rur2r2r3rD_s zMathCall.__repr__)rZrfrgr?rJrDrlr2r2r0r3rTsrc@s$eZdZdddddZddZdS) FuncNoder&r6)r+r8cCs.|tkrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr+rr^r)r>r+r2r2r3r?eszFuncNode.__init__cGs t||Sr$)r)r>rHr2r2r3rJkszFuncNode.__call__N)rZrfrgr?rJr2r2r2r3rdsr)Xr __future__rr functoolsroperatortypingrrrrZnumpyr^Zpandas._libs.tslibsr Zpandas.core.dtypes.commonr r Zpandas.core.commoncorecommonrZpandas.core.computation.commonr r Zpandas.core.computation.scoperZpandas.io.formats.printingrrZ REDUCTIONSZ_unary_math_opsZ_binary_math_opsrr;r#r'rvrsrrrgtltgeleeqneZ_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_dictrdrPrrrrrrposneginvertZ_unary_ops_funcsrrrrr2r2r2r3s     t 6  1