o 5cF?@sddZddlmZddlmZddlmZddlZddlmZm 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#ZId5d&d'ZJGd(d)d)e&ZKd5d*d+ZLGd,d-d-eKZMd.ZNejOejPejQejQfZRe1e2eNeRZSGd/d0d0e&ZTGd1d2d2e&ZUGd3d4d4ZVdS)6z 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)sincosexplogexpm1log1psqrtsinhcoshtanharcsinarccosarctanarccosharcsinharctanhabslog10floorceil)arctan2__pd_eval_local_cseZdZUd'fdd Zded<d'd(dd Zed)d d Zd)d dZddZ d*ddZ ddZ d(ddZ ed+ddZ eddZeZed)ddZed+ddZedd Zejd!d Zed"d#Zed,d%d&ZZS)-TermNcs&t|tstn|}tt|j}||SN) isinstancestrConstantsuperr(__new__)clsnameenvsideencodingklasssupr_new __class__Q/var/www/html/gps/gps/lib/python3.10/site-packages/pandas/core/computation/ops.pyr.Fs z Term.__new__boolis_localreturnNonecCsB||_||_||_t|}|tp|tv|_||_ ||_ dSr)) _namer1r2r+ startswith LOCAL_TAGr r; _resolve_name_valuer3)selfr0r1r2r3tnamer8r8r9__init__Ns  z Term.__init__r+cCs|jtdS)N)r0replacer@rCr8r8r9 local_nameXszTerm.local_namecC t|jSr))rr0rHr8r8r9__repr__\ z Term.__repr__cO|jSr)valuerCargskwargsr8r8r9__call___z Term.__call__cOs|Sr)r8rPr8r8r9evaluatebsz Term.evaluatecCslt|j}|j}||jjvrt|jj|trd}|jj||d}||t |dr4|j dkr4t d|S)NF)r;ndimz?N-dimensional objects, where N > 2, are not supported with eval) r+rIr;r1scoper*typeresolveupdatehasattrrVNotImplementedError)rCrIr;resr8r8r9rAes   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)r0r*r+r1swapkeyrIrO)rCrOkeyr8r8r9r[vs  z Term.updatecCrJr))r rBrHr8r8r9r  zTerm.is_scalarc CsLz|jjjWSty%z|jjWYSty$t|jYYSwwr))rBvaluesdtypeAttributeErrorrYrHr8r8r9rYs   z Term.typecCs$t|jdt|jd|jdS)Nz(name=z, type=))rY__name__reprr0rHr8r8r9raws$zTerm.rawcCs6z|jj}Wn ty|j}Ynwt|ttjfSr))rYre issubclassrnp datetime64rCtr8r8r9 is_datetime    zTerm.is_datetimecCrMr)rBrHr8r8r9rOz Term.valuecCs ||_dSr)rq)rCr_r8r8r9rOrbcCrMr)r>rHr8r8r9r0rrz Term.nameintcCs|jjSr))rBrVrHr8r8r9rVsz Term.ndimNNr<r=r<r+)r<r(r<r:)r<rt)rg __module__ __qualname__r.__annotations__rEpropertyrIrKrSrUrAr[r rY return_typerirorOsetterr0rV __classcell__r8r8r6r9r(Es8              r(cs>eZdZd dfdd ZddZedd Zdd d ZZS)r,Nr<r=cstj||||ddS)N)r2r3)r-rE)rCrOr1r2r3r6r8r9rEzConstant.__init__cCrMr)rsrHr8r8r9rArTzConstant._resolve_namecCrMr)rNrHr8r8r9r0rrz Constant.namer+cCrJr))rhr0rHr8r8r9rKs zConstant.__repr__rurvrw) rgryrzrErAr|r0rKrr8r8r6r9r,s  r,~&|)notandorc@szeZdZUdZded<ddd d Zd d Zdd dZeddZ edddZ eddZ edddZ edddZ dS)Opz. Hold an operator of arbitrary arity. r+opNoperandsIterable[Term | Op]r<r=cCst|||_||_||_dSr)) _bool_op_mapgetrrr3)rCrrr3r8r8r9rEs z Op.__init__cCrJr))iterrrHr8r8r9__iter__rLz Op.__iter__cCs(dd|jD}td|jd|S)zW Print a generic n-ary operator and its operands using infix notation. css |] }dt|dVqdS)(rfN)r).0oprr8r8r9 szOp.__repr__.. )rrrjoin)rCparenedr8r8r9rKsz Op.__repr__cCs,|jttvr tjStddt|DS)Ncs|]}|jVqdSr)rYrtermr8r8r9rz!Op.return_type..)r CMP_OPS_SYMS BOOL_OPS_SYMSrkbool_r comflattenrHr8r8r9r}szOp.return_typer:cCs(|j}ttdg}|jtko||S)Nobject) operand_types frozensetrkrdr}r)rCtypes obj_dtype_setr8r8r9has_invalid_return_typeszOp.has_invalid_return_typecCstddt|DS)Ncsrr)rrr8r8r9rrz#Op.operand_types..)rrrrHr8r8r9rszOp.operand_typescCstdd|jDS)Ncsrr))r )roperandr8r8r9rrzOp.is_scalar..)allrrHr8r8r9r sz Op.is_scalarcCs6z|jj}Wn ty|j}Ynwt|ttjfSr))r}rYrerjrrkrlrmr8r8r9rorpzOp.is_datetimer))rr+rrr<r=rwrx)rgryrz__doc__r{rErrKr|r}rrr ror8r8r8r9rs       rc CsTz||WSty)t|r#z||WYSty"Ynw||vYSw)z` Compute the vectorized membership of ``x in y`` if possible, otherwise use Python. isinrer xyr8r8r9_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. rrr8r8r9_not_ins   r)>=z<=z==z!=inznot in)rrrr)+-*/**//%)rrrc Cs^t|}|D]%}|j|vrqz|j|}Wnty&||j}Ynw||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)rkrdrYrOastyperer[)termsacceptable_dtypesrddtrr_r8r8r9 _cast_inplaceLs    rr<r:cCs t|tSr))r*r()objr8r8r9is_termerLrcsFeZdZdZdfdd 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<r=c s~t|||f||_||_||zt||_WdSty>}zt t }t dt |d||d}~ww)NzInvalid binary operator , valid operators are ) r-rElhsrhs_disallow_scalar_only_bool_opsconvert_values_binary_ops_dictfuncKeyErrorlistkeys ValueErrorrh)rCrrrerrrr6r8r9rEts  zBinOp.__init__cCs ||}||}|||S)z Recursively evaluate an expression in Python space. Parameters ---------- env : Scope Returns ------- object The result of an evaluated expression. )rrr)rCr1leftrightr8r8r9rSs   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_dictrrr1) rrUrrrrOpandas.core.computation.evalradd_tmp) rCr1rrrrr^rrrr0r8r8r9rUs,     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)r3)r3rrr)rOencoderrHr8r9 stringifys z'BinOp.convert_values..stringifyNUTC)rrrror rOr*rtfloatrr tz tz_convertr[)rCrrrvr8rHr9rs&        zBinOp.convert_valuescCsz|j}|j}|j}t|d|}|j}t|d|}|js|jr9|jtvr7t|tt j fr3t|tt j fs;t ddSdSdS)NrYz$cannot evaluate scalar only bool ops) rrr}getattrr r_bool_ops_dictrjr:rkrr])rCrrrhs_rtlhs_rtr8r8r9rs$   z$BinOp._disallow_scalar_only_bool_ops)rr+r<r=)rr+rv) rgryrzrrErSrUrrrr8r8r6r9ris   1!rcCstt|jtjSr))rjrkrdrYnumber)rdr8r8r9 isnumericsrcs"eZdZdZdfdd ZZS)Divz Div operator to special case casting. Parameters ---------- lhs, rhs : Term or Op The Terms or Ops in the ``/`` expression. r<r=csjtd||t|jrt|js#td|jd|jd|jdtjtjg}t t ||tjdS)Nrz unsupported operand type(s) for z: 'z' and '') r-rErr} TypeErrorrrkfloat32float_rrr)rCrrrr6r8r9rEs  z Div.__init__rv)rgryrzrrErr8r8r6r9rs r)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<r=c sZt||f||_zt||_WdSty,}z tdt|dt|d}~ww)NzInvalid unary operator r) r-rEr_unary_ops_dictrrrrhUNARY_OPS_SYMS)rCrrrr6r8r9rE3s zUnaryOp.__init__MathCallcCs||}||Sr))rr)rCr1rr8r8r9rS?s  zUnaryOp.__call__r+cCst|jd|jdS)Nrrf)rrrrHr8r8r9rKDrzUnaryOp.__repr__np.dtypecCsR|j}|jtdkrtdSt|tr$|jtvs|jtvr$tdStdS)Nr:rt) rr}rkrdr*rr _cmp_ops_dictr)rCrr8r8r9r}Gs    zUnaryOp.return_type)rrr<r=)r<rrw)r<r) rgryrzrrErSrKr|r}rr8r8r6r9r"s rcs0eZdZd fdd ZddZd dd ZZS) rr<r=cst|j|||_dSr))r-rEr0r)rCrrQr6r8r9rETs zMathCall.__init__csPfdd|jD}tjdd|jj|WdS1s!wYdS)Ncsg|]}|qSr8r8)rrrr8r9 Zsz%MathCall.__call__..ignore)r)rrkerrstater)rCr1rr8rr9rSXs $zMathCall.__call__r+cCs(tt|j}t|jdd|dS)Nr,rf)mapr+rrrr)rCrr8r8r9rK^s zMathCall.__repr__rvrw)rgryrzrErSrKrr8r8r6r9rSsrc@seZdZd ddZddZd S) FuncNoder0r+r<r=cCs.|tvr td|d||_tt||_dS)N"z" is not a supported function)MATHOPSrr0rrkr)rCr0r8r8r9rEdszFuncNode.__init__cGs t||Sr))r)rCrQr8r8r9rSjrLzFuncNode.__call__N)r0r+r<r=)rgryrzrErSr8r8r8r9rcs  rrx)Wr __future__rr functoolsroperatortypingrrrnumpyrkpandas._libs.tslibsrpandas.core.dtypes.commonr r pandas.core.commoncorecommonrpandas.core.computation.commonr r pandas.core.computation.scoper pandas.io.formats.printingrr REDUCTIONS_unary_math_ops_binary_math_opsrr@r(r,rrrrrgtltgeleeqne_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[rrrrrrposneginvert_unary_ops_funcsrrrrr8r8r8r9s     t 6     1