o ŀg9@s`dZddlmZddlmZddlmZddlZddlmZm Z m Z ddl Z ddl mZddlmZmZddlmmZdd lmZmZdd lmZdd lmZmZer]dd 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/rMr?r5scoper.typeresolveupdatehasattrrZNotImplementedError)rGrMr?resr<r<r=rEks   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)r4r.r/r5swapkeyrMrS)rGrSkeyr<r<r=r_|s  z Term.updatecCrNr-)r rFrLr<r<r=r  zTerm.is_scalarc CsLz|jjjWSty%z|jjWYSty$t|jYYSwwr-)rFvaluesdtypeAttributeErrorr]rLr<r<r=r]s   z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))r]__name__reprr4rLr<r<r=raws$zTerm.rawcCs6z|jj}Wn ty|j}Ynwt|ttjfSr-)r]ri issubclassrnp datetime64rGtr<r<r= is_datetime    zTerm.is_datetimecCrQr-rFrLr<r<r=rSz Term.valuecCs ||_dSr-ru)rGrcr<r<r=rSrfcCrQr-rBrLr<r<r=r4rvz Term.nameintcCs|jjSr-)rFrZrLr<r<r=rZsz Term.ndim)NNr@rAr@r/)r@r,r@r>)r@rx)rk __module__ __qualname__r2__annotations__rIpropertyrMrOrWrYrEr_r r] return_typermrsrSsetterr4rZ __classcell__r<r<r:r=r,Ks8              r,c@s*eZdZddZeddZd ddZd S) r0cCrQr-rwrLr<r<r=rErXzConstant._resolve_namecCrQr-rRrLr<r<r=r4rvz Constant.namer@r/cCrNr-)rlr4rLr<r<r=rOs zConstant.__repr__Nrz)rkr|r}rErr4rOr<r<r<r=r0s  r0~&|)notandorc@s|eZdZUdZded<ddd d Zdd d ZdddZeddZ edddZ eddZ edddZ edddZ dS) Opz. Hold an operator of arbitrary arity. r/opNoperandsIterable[Term | Op]r@rAcCst|||_||_||_dSr-) _bool_op_mapgetrrr7)rGrrr7r<r<r=rIs z Op.__init__rcCrNr-)iterrrLr<r<r=__iter__rPz Op.__iter__cCs(dd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. css |] }dt|dVqdS)(rjN)r).0oprr<r<r= szOp.__repr__.. )rrrjoin)rGparenedr<r<r=rOsz Op.__repr__cCs,|jttvr tjStddt|DS)Ncs|]}|jVqdSr-r]rtermr<r<r=rz!Op.return_type..)r CMP_OPS_SYMS BOOL_OPS_SYMSrobool_r comflattenrLr<r<r=rszOp.return_typer>cCs(|j}ttdg}|jtko||S)Nobject) operand_types frozensetrorhrr)rGtypes obj_dtype_setr<r<r=has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncsrr-rrr<r<r=rrz#Op.operand_types..)rrrrLr<r<r=rszOp.operand_typescCstdd|jDS)Ncsrr-)r )roperandr<r<r=rrzOp.is_scalar..)allrrLr<r<r=r sz Op.is_scalarcCs6z|jj}Wn ty|j}Ynwt|ttjfSr-)rr]rirnrrorprqr<r<r=rsrtzOp.is_datetimer-)rr/rrr@rA)r@rrzr{)rkr|r}__doc__r~rIrrOrrrrr rsr<r<r<r=rs        rc CsTz||WSty)t|r#z||WYSty"Ynw||vYSw)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. isinrir xyr<r<r=_ins    rc CsXz||WSty+t|r%z ||WYSty$Ynw||vYSw)zd Compute the vectorized membership of ``x not in y`` if possible, otherwise use Python. rrr<r<r=_not_ins   r)>=z<=z==z!=inznot in)rrrr)+-*/**//%)rrrr@r>cCs t|tSr-)r.r,)objr<r<r=is_termOrPrcsFeZdZdZdfdd Zdd Zdd d Zdd dZddZZ S)BinOpz Hold a binary operator and its operands. Parameters ---------- op : str lhs : Term or Op rhs : Term or Op rr/r@rAc s~t|||f||_||_||zt||_WdSty>}zt t }t dt |d||d}~ww)NzInvalid binary operator , valid operators are ) r1rIlhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorrl)rGrrrerrrr:r<r=rI^s  zBinOp.__init__cCs ||}||}|||S)z Recursively evaluate an expression in Python space. Parameters ---------- env : Scope Returns ------- object The result of an evaluated expression. )rrr)rGr5leftrightr<r<r=rWps   zBinOp.__call__enginec Cs|dkr ||}n2|jj|||||d}|jj|||||d}|j|vr-||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) local_dictrrr5) rrYrrrrSpandas.core.computation.evalradd_tmp) rGr5rrrrrbrrrr4r<r<r=rYs,     zBinOp.evaluatecsfdd}jj}}t|r?|jr?t|r?|jr?|j}t|ttfr)||}t t |}|j dur9| d}j |t|rs|jrut|rw|jry|j}t|ttfr[||}t t |}|j durk| d}j |dSdSdSdSdS)zK Convert datetimes to a comparable value in an expression. cs,jdurttjd}||St}||S)N)r7)r7rrr)rSencoderrLr<r= stringifys z'BinOp.convert_values..stringifyNUTC)rrrrsr rSr.rxfloatrr tz tz_convertr_)rGrrrvr<rLr=rs&        zBinOp.convert_valuescCsz|j}|j}|j}t|d|}|j}t|d|}|js|jr9|jtvr7t|tt j fr3t|tt j fs;t ddSdSdS)Nr]z$cannot evaluate scalar only bool ops) rrrgetattrr r_bool_ops_dictrnr>rorra)rGrrrhs_rtlhs_rtr<r<r=rs&   z$BinOp._disallow_scalar_only_bool_ops)rr/r@rA)rr/ry) rkr|r}rrIrWrYrrrr<r<r:r=rSs   1!rcCstt|jtjSr-)rnrorhr]number)rhr<r<r= isnumericsr)rrrrcsDeZdZdZdfdd Zdd d Zdd d ZedddZZ 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. rLiteral['+', '-', '~', 'not']r@rAc sZt||f||_zt||_WdSty,}z tdt|dt|d}~ww)NzInvalid unary operator r) r1rIr_unary_ops_dictrrrrlUNARY_OPS_SYMS)rGrrrr:r<r=rIs zUnaryOp.__init__MathCallcCs||}||Sr-)rr)rGr5rr<r<r=rWs  zUnaryOp.__call__r/cCst|jd|jdS)Nrrj)rrrrLr<r<r=rOszUnaryOp.__repr__np.dtypecCsR|j}|jtdkrtdSt|tr$|jtvs|jtvr$tdStdS)Nr>rx) rrrorhr.rr _cmp_ops_dictr)rGrr<r<r=rs    zUnaryOp.return_type)rrr@rAr@rrz)r@r) rkr|r}rrIrWrOrrrr<r<r:r=rs rcs0eZdZd fdd ZddZd dd ZZS) rr@rAcst|j|||_dSr-)r1rIr4r)rGrrUr:r<r=rI&s zMathCall.__init__cs fdd|jD}|jj|S)Ncsg|]}|qSr<r<)rrrr<r= ,sz%MathCall.__call__..)rr)rGr5rr<rr=rW*s zMathCall.__call__r/cCs(tt|j}t|jdd|dS)Nr,rj)mapr/rrrr)rGrr<r<r=rO/s zMathCall.__repr__ryrz)rkr|r}rIrWrOrr<r<r:r=r%src@s eZdZd ddZd dd Zd S) FuncNoder4r/r@rAcCs.|tvr td|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr4rror)rGr4r<r<r=rI5szFuncNode.__init__rcGs t||Sr-)r)rGrUr<r<r=rW;rPzFuncNode.__call__N)r4r/r@rAr)rkr|r}rIrWr<r<r<r=r4s rr{)Xr __future__rr functoolsroperatortypingrrrnumpyropandas._libs.tslibsrpandas.core.dtypes.commonr r pandas.core.commoncorecommonrpandas.core.computation.commonr r pandas.core.computation.scoper pandas.io.formats.printingrrcollections.abcrr REDUCTIONS_unary_math_ops_binary_math_opsrrDr,r0rrrrrgtltgeleeqne_cmp_ops_funcsdictziprrand_or__bool_ops_funcsrARITH_OPS_SYMSaddsubmultruedivpowfloordivmod_arith_ops_funcs_arith_ops_dictSPECIAL_CASE_ARITH_OPS_SYMS_special_case_arith_ops_funcs_special_case_arith_ops_dictrdr_rrrrposneginvert_unary_ops_funcsrrrrr<r<r<r=s     t 6     1