B W0d ?@s~dZddlmZddlmZddlmZddlZddlmZm 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@ejAfZEe2e3eDeEZFiZGxe4e9eCfD]ZHeGIeHqWd$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__M/var/www/html/venv/lib/python3.7/site-packages/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}||S)N) isinstancer Constantr$r0__new__)clsr!envsideencodingklassZsupr_new)r(r)r*r3Ss z Term.__new__boolr"cCsB||_||_||_t|}|tp*|tk|_||_ ||_ dS)N) _namer5r6r startswith LOCAL_TAGr r" _resolve_name_valuer7)r&r!r5r6r7Ztnamer)r)r*r%[s z Term.__init__r )returncCs|jtdS)N)r!replacer<)r&r)r)r* local_nameeszTerm.local_namecCs t|jS)N)r r!)r&r)r)r*__repr__isz Term.__repr__cOs|jS)N)value)r&argskwargsr)r)r*__call__lsz Term.__call__cOs|S)Nr))r&rErFr)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)r5resolverBr"updatehasattrrINotImplementedError)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!r1r r5ZswapkeyrBrD)r&rDkeyr)r)r*rL|s  z Term.updatecCs t|jS)N)r r>)r&r)r)r*r szTerm.is_scalarc CsHy |jjjStk rBy|jjStk r<t|jSXYnXdS)N)r>valuesdtypeAttributeErrortype)r&r)r)r*rUs z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rUr+r#r!)r&r)r)r*rawszTerm.rawcCs8y |jj}Wntk r&|j}YnXt|ttjfS)N)rUrT issubclassrnp datetime64)r&tr)r)r* is_datetimes   zTerm.is_datetimecCs|jS)N)r>)r&r)r)r*rDsz Term.valuecCs ||_dS)N)r>)r&rPr)r)r*rDscCs|jS)N)r:)r&r)r)r*r!sz Term.nameintcCs|jjS)N)r>rI)r&r)r)r*rIsz Term.ndim)NN)NN)r+r,r-r3__annotations__r%propertyrBrCrGrHr=rLr rU return_typerWr\rDsetterr!rIr/r)r))r(r*r0Rs$      r0cs@eZdZd fdd ZddZeddZdd d d ZZS) r2Ncstj||||ddS)N)r6r7)r$r%)r&rDr5r6r7)r(r)r*r%szConstant.__init__cCs|jS)N)r:)r&r)r)r*r=szConstant._resolve_namecCs|jS)N)rD)r&r)r)r*r!sz Constant.namer )r?cCs t|jS)N)r#r!)r&r)r)r*rCszConstant.__repr__)NN) r+r,r-r%r=r_r!rCr/r)r))r(r*r2s r2~&|)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])rioperandscCst|||_||_||_dS)N) _bool_op_mapgetrirjr7)r&rirjr7r)r)r*r%sz Op.__init__cCs t|jS)N)iterrj)r&r)r)r*__iter__sz Op.__iter__)r?cCs(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)r* szOp.__repr__.. )rjr rijoin)r&Zparenedr)r)r*rCsz Op.__repr__cCs,|jttkrtjStddt|DS)Ncss|] }|jVqdS)N)rU)rptermr)r)r*rqsz!Op.return_type..)ri CMP_OPS_SYMS BOOL_OPS_SYMSrYbool_r comflatten)r&r)r)r*r`szOp.return_typer9cCs(|j}ttdg}|jtko&||S)Nobject) operand_types frozensetrYrSr`rz)r&typesZ obj_dtype_setr)r)r*has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncss|] }|jVqdS)N)rU)rprtr)r)r*rqsz#Op.operand_types..)r|rxry)r&r)r)r*r{szOp.operand_typescCstdd|jDS)Ncss|] }|jVqdS)N)r )rpoperandr)r)r*rqszOp.is_scalar..)allrj)r&r)r)r*r sz Op.is_scalarcCs8y |jj}Wntk r&|j}YnXt|ttjfS)N)r`rUrTrXrrYrZ)r&r[r)r)r*r\s   zOp.is_datetime)N)r+r,r-r.r^r%rnrCr_r`r~r{r r\r)r)r)r*rhs   rhc CsPy ||Stk rJt|rBy ||Stk r@YnX||kSXdS)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. N)isinrTr)xyr)r)r*_in s  rc CsTy ||Stk rNt|rFy ||Stk rDYnX||kSXdS)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. N)rrTr)rrr)r)r*_not_ins  r)>=z<=z==z!=inznot in)rcrdrfrg)+-*/z**z//%)z**z//rc Csdt|}xT|D]L}|j|kr qy|j|}Wn tk rP||j}YnX||qWdS)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)rYrSrUrDZastyperTrL)Ztermsacceptable_dtypesrSdtrtrPr)r)r* _cast_inplaceRs   rr9)r?cCs t|tS)N)r1r0)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 )ric st|||f||_||_||yt||_WnHtk r}z*t t }t dt |d||Wdd}~XYnXdS)NzInvalid binary operator z, valid operators are ) r$r%lhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorr#)r&rirrerrr)r(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&r5leftrightr)r)r*rGs  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_dictrr)r5) rrHrrirrDZpandas.core.computation.evalrZadd_tmp) r&r5rrrrrOrrrr!r)r)r*rHs(    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)r7)r7rrr )rDencoder)r&r)r* stringifys z'BinOp.convert_values..stringifyNUTC)rrrr\r rDr1r]floatrr tzZ tz_convertrL)r&rrrvr))r&r*rs$        zBinOp.convert_valuescCsr|j}|j}|j}t|d|}|j}t|d|}|js<|jrn|jtkrnt|tt j frft|tt j fsnt ddS)NrUz$cannot evaluate scalar only bool ops) rrr`getattrr ri_bool_ops_dictrXr9rYrwrN)r&rrZrhs_rtZlhs_rtr)r)r*rs    z$BinOp._disallow_scalar_only_bool_ops) r+r,r-r.r%rGrHrrr/r)r))r(r*ros  1!rcCstt|jtjS)N)rXrYrSrUnumber)rSr)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%rr` TypeErrorrirYZfloat32Zfloat_rrxry)r&rrr)r(r)r*r%s   z Div.__init__)r+r,r-r.r%r/r)r))r(r*r sr)rrrbrecsNeZdZdZddfdd 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 )ric sft||f||_yt||_Wn<tk r`}ztdt|dt|Wdd}~XYnXdS)NzInvalid unary operator z, valid operators are ) r$r%r_unary_ops_dictrrrr#UNARY_OPS_SYMS)r&rirr)r(r)r*r%9szUnaryOp.__init__cCs||}||S)N)rr)r&r5rr)r)r*rGEs zUnaryOp.__call__)r?cCst|jd|jdS)NrorV)r rir)r&r)r)r*rCIszUnaryOp.__repr__znp.dtypecCsR|j}|jtdkr tdSt|trH|jtks>|jtkrHtdStdS)Nr9r]) rr`rYrSr1rhri _cmp_ops_dictr)r&rr)r)r*r`Ls   zUnaryOp.return_type) r+r,r-r.r%rGrCr_r`r/r)r))r(r*r(s  rcs2eZdZfddZddZddddZZS) MathCallcst|j|||_dS)N)r$r%r!r)r&rrE)r(r)r*r%YszMathCall.__init__c s8fdd|jD}tjdd|jj|SQRXdS)Ncsg|] }|qSr)r))rpri)r5r)r* _sz%MathCall.__call__..ignore)r)rjrYZerrstater)r&r5rjr))r5r*rG]szMathCall.__call__r )r?cCs(tt|j}t|jdd|dS)Nro,rV)mapr rjr rirs)r&rjr)r)r*rCcs zMathCall.__repr__)r+r,r-r%rGrCr/r)r))r(r*rXs rc@s"eZdZddddZddZdS)FuncNoder )r!cCs.|tkrtd|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr!rrYr)r&r!r)r)r*r%iszFuncNode.__init__cGs t||S)N)r)r&rEr)r)r*rGoszFuncNode.__call__N)r+r,r-r%rGr)r)r)r*rhsr)Xr. __future__rr functoolsroperatortypingrrnumpyrYZpandas._libs.tslibsrZpandas.core.dtypes.commonrr Zpandas.core.commoncorecommonrxZpandas.core.computation.commonr r Zpandas.core.computation.scoper Zpandas.io.formats.printingr rZ REDUCTIONSZ_unary_math_opsZ_binary_math_opsrr< NameErrorrr0r2rkrhrrrugtltgeleeqneZ_cmp_ops_funcsdictziprrvand_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_dictrdrLrrrrrrposneginvertZ_unary_ops_funcsrrrrr)r)r)r*s     m 6 0