U
|
¬ý°dêA ã @ sx d Z ddlmZ ddlmZ ddlmZ ddlZddlmZm Z m
|
Z
|
m Z ddl Z ddlmZ ddlmZmZ ddlm mZ dd lmZmZ dd
|
lmZ dd l m Z m Z d Z d Z dZ!e e! Z"dZ#G dd dZ$G dd de$Z%ddddZ&G dd dZ'dd Z(d d Z)d Z*ej+ej,ej-ej.ej/ej0e(e)fZ1e2e3e*e1Z4d Z5ej6ej7ej6ej7fZ8e2e3e5e8Z9d Z:ej;ej<ej=ej>ej?ej@ejAfZBe2e3e:eBZCd!ZDej?ej@ejAfZEe2e3eDeEZFi ZGe4e9eCfD ]ZHeG IeH¡ qÄd"d#d$d%ZJd&d#d'd(ZKG d)d* d*e'ZLd&d#d+d,ZMG d-d. d.eLZNd/ZOejPejQejRejRfZSe2e3eOeSZTG d0d1 d1e'ZUG d2d3 d3e'ZVG d4d5 d5ZWdS )6z
|
Operator classes for eval.
|
é )Ú annotations)Údatetime)ÚpartialN)ÚCallableÚIterableÚIteratorÚLiteral)Ú Timestamp)Ú is_list_likeÚ is_scalar)Úensure_decodedÚresult_type_many)ÚDEFAULT_GLOBALS)Ú pprint_thingÚpprint_thing_encoded)ÚsumÚprodÚminÚmax)ÚsinÚcosÚexpÚlogÚexpm1Úlog1pÚsqrtÚsinhÚcoshÚtanhZarcsinZarccosZarctanZarccoshZarcsinhZarctanhÚabsÚlog10ÚfloorÚceil)Zarctan2Z__pd_eval_local_c s e Zd ZU d' fdd Zded< d(dddd Zed
|
dd d Zd
|
dd dZdd Z d dddZ
|
dd Z ddddZ eddddZ edd ZeZed
|
ddd Zeddd d Zed d Zejddd!d Zed"d# Zed$dd%d&Z ZS ))ÚTermNc s& t |tstn| }tt|j}||S ©N)Ú
|
isinstanceÚstrÚConstantÚsuperr# Ú__new__)ÚclsÚnameÚenvÚsideÚencodingÚklassZsupr_new©Ú __class__© úRd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/computation/ops.pyr) G s z Term.__new__ÚboolÚis_localÚNone©Úreturnc C sB || _ || _|| _t|}| t¡p*|tk| _| ¡ | _ || _
|
d S r$ ) Ú_namer, r- r& Ú
|
startswithÚ LOCAL_TAGr r5 Ú _resolve_nameÚ_valuer. )Úselfr+ r, r- r. Ztnamer2 r2 r3 Ú__init__O s
|
z Term.__init__r& c C s | j td¡S )NÚ )r+ Úreplacer; ©r> r2 r2 r3 Ú
|
local_nameY s zTerm.local_namec C s
|
t | jS r$ )r r+ rB r2 r2 r3 Ú__repr__] s z Term.__repr__c O s | j S r$ ©Úvalue©r> ÚargsÚkwargsr2 r2 r3 Ú__call__` s z Term.__call__c O s | S r$ r2 rG r2 r2 r3 Úevaluatec s z Term.evaluatec C sl t | j}| j}|| jjkr2t| jj| tr2d}| jj||d}| |¡ t |drh|j
|
dkrht d|S )NF)r5 Úndimé z?N-dimensional objects, where N > 2, are not supported with eval) r& rC r5 r, Zscoper% ÚtypeÚresolveÚupdateÚhasattrrL ÚNotImplementedError)r> rC r5 Úresr2 r2 r3 r< f s
|
|
ÿ
|
ÿzTerm._resolve_namec C s. | 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, ZswapkeyrC rF )r> rF Úkeyr2 r2 r3 rP w s
|
|
z Term.updatec C s
|
t | jS r$ )r r= rB r2 r2 r3 r s zTerm.is_scalarc C sX z | j jjW S tk
|
rR z| j jW Y S tk
|
rL t| j Y Y S X Y nX d S r$ )r= ÚvaluesÚdtypeÚAttributeErrorrN rB r2 r2 r3 rN s z Term.typec C s$ t | j dt| j d| j dS )Nz(name=z, type=ú))rN Ú__name__Úreprr+ rB r2 r2 r3 Úraw s zTerm.rawc C s8 z | j j }W n tk
|
r& | j }Y nX t|ttjfS r$ )rN rX Ú
|
issubclassr ÚnpÚ
|
datetime64©r> Útr2 r2 r3 Ú is_datetime s
|
zTerm.is_datetimec C s | j S r$ ©r= rB r2 r2 r3 rF © s z
|
Term.valuec C s
|
|| _ d S r$ rc )r> rT r2 r2 r3 rF s c C s | j S r$ ©r9 rB r2 r2 r3 r+ ± s z Term.nameÚintc C s | j jS r$ )r= rL rB r2 r2 r3 rL µ s z Term.ndim)NN)NN)rZ Ú
|
__module__Ú __qualname__r) Ú__annotations__r? ÚpropertyrC rD rJ rK r< rP r rN Ú return_typer\ rb rF Úsetterr+ rL Ú __classcell__r2 r2 r0 r3 r# F s6
|
|
|
|
|
r# c sF e Zd Zd dd fdd Zdd Zedd Zd
|
dd d Z ZS )r' Nr6 r7 c s t j||||d d S )N)r- r. )r( r? )r> rF r, r- r. r0 r2 r3 r? » s zConstant.__init__c C s | j S r$ rd rB r2 r2 r3 r< ¾ s zConstant._resolve_namec C s | j S r$ rE rB r2 r2 r3 r+ Á s z Constant.namer& c C s
|
t | jS r$ )r[ r+ rB r2 r2 r3 rD Å s zConstant.__repr__)NN) rZ rf rg r? r< ri r+ rD rl r2 r2 r0 r3 r' º s
|
|
r' ú~ú&ú|)ÚnotÚandÚorc @ s e Zd ZU dZded< ddddddd Zd
|
d d d Zdd ddZedd Z edd ddZ
|
edd Z edd ddZ edd ddZ dS ) ÚOpz.
|
Hold an operator of arbitrary arity.
|
r& ÚopNzIterable[Term | Op]r6 )rt Úoperandsr8 c C s t ||¡| _|| _|| _d S r$ )Ú _bool_op_mapÚgetrt ru r. )r> rt ru r. r2 r2 r3 r? Õ s z Op.__init__r r7 c C s
|
t | jS r$ )Úiterru rB r2 r2 r3 Ú__iter__Ú s z Op.__iter__c C s( dd | j D }td| j d |¡S )zW
|
Print a generic n-ary operator and its operands using infix notation.
|
c s s | ]}d t | dV qdS )ú(rY N)r )Ú.0Zoprr2 r2 r3 Ú <genexpr>â s z Op.__repr__.<locals>.<genexpr>ú )ru r rt Újoin)r> Zparenedr2 r2 r3 rD Ý s z Op.__repr__c C s, | j tt krtjS tdd t | ¡D S )Nc s s | ] }|j V qd S r$ ©rN ©r{ Útermr2 r2 r3 r| ê s z!Op.return_type.<locals>.<genexpr>)rt Ú CMP_OPS_SYMSÚ BOOL_OPS_SYMSr^ Úbool_r ÚcomÚflattenrB r2 r2 r3 rj å s zOp.return_typer4 c C s( | j }tt d¡g}| jtko&|| S )NÚobject)Ú operand_typesÚ frozensetr^ rW rj r )r> ÚtypesZ obj_dtype_setr2 r2 r3 Úhas_invalid_return_typeì s zOp.has_invalid_return_typec C s t dd t | ¡D S )Nc s s | ] }|j V qd S r$ r r r2 r2 r3 r| ô s z#Op.operand_types.<locals>.<genexpr>)r r
r rB r2 r2 r3 r ò s zOp.operand_typesc C s t dd | jD S )Nc s s | ] }|j V qd S r$ )r )r{ Úoperandr2 r2 r3 r| ø s z Op.is_scalar.<locals>.<genexpr>)Úallru rB r2 r2 r3 r ö s z Op.is_scalarc C s8 z | j j}W n tk
|
r& | j }Y nX t|ttjfS r$ )rj rN rX r] r r^ r_ r` r2 r2 r3 rb ú s
|
zOp.is_datetime)N)rZ rf rg Ú__doc__rh r? ry rD ri rj r r r rb r2 r2 r2 r3 rs Î s
|
|
|
rs c C s\ z | |¡W S tk
|
rV t| rJz| | ¡W Y S tk
|
rH Y nX | |k Y S X dS )z`
|
Compute the vectorized membership of ``x in y`` if possible, otherwise
|
use Python.
|
N©ÚisinrX r
|
©ÚxÚyr2 r2 r3 Ú_in s r c C s` z| |¡ W S tk
|
rZ t| rNz| | ¡ W Y S tk
|
rL Y nX | |k Y S X dS )zd
|
Compute the vectorized membership of ``x not in y`` if possible,
|
otherwise use Python.
|
Nr r r2 r2 r3 Ú_not_in s r )ú>ú<z>=z<=z==z!=Úinznot in)rn ro rq rr )ú+ú-Ú*ú/ú**ú//ú%)r r r r6 r7 c C s` t |¡}| D ]L}|j|kr qz|j |¡}W n tk
|
rN | |j¡}Y nX | |¡ 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^ rW rN rF ZastyperX rP )ZtermsÚacceptable_dtypesrW Údtr rT r2 r2 r3 Ú _cast_inplaceM s
|
|
r¢ r4 c C s
|
t | tS r$ )r% r# )Úobjr2 r2 r3 Úis_termf s r¤ c sT e Zd ZdZddd fdd Zdd Zdd d
|
d Zdd d dZdd Z Z S )ÚBinOpz
|
Hold a binary operator and its operands.
|
|
Parameters
|
----------
|
op : str
|
lhs : Term or Op
|
rhs : Term or Op
|
r& r6 ©rt r8 c
|
s t |||f¡ || _|| _| ¡ | ¡ zt| | _W nH tk
|
r } z*t t
|
¡ }t dt | d| |W 5 d }~X Y nX d S )NzInvalid binary operator ú, valid operators are ) r( r? ÚlhsÚrhsÚ _disallow_scalar_only_bool_opsÚconvert_valuesÚ_binary_ops_dictÚfuncÚKeyErrorÚlistÚkeysÚ
|
ValueErrorr[ )r> rt r¨ r© Úerrr° r0 r2 r3 r? u s ÿþzBinOp.__init__c C s | |¡}| |¡}| ||¡S )zã
|
Recursively evaluate an expression in Python space.
|
|
Parameters
|
----------
|
env : Scope
|
|
Returns
|
-------
|
object
|
The result of an evaluated expression.
|
)r¨ r© r )r> r, ÚleftÚrightr2 r2 r3 rJ s
|
|
zBinOp.__call__)Úenginec C s |dkr| |}nd| j j|||||d}| jj|||||d}| j|krZ| |j|j¡}n ddlm} | | |||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)rµ ÚparserÚ term_typeÚeval_in_pythonr )Úeval)Z
|
local_dictrµ r· ©r, ) r¨ rK r© rt r rF Z pandas.core.computation.evalrº Zadd_tmp) r> r, rµ r· r¸ r¹ rS r³ r´ rº r+ r2 r2 r3 rK s,
|
ûû
|
|
zBinOp.evaluater7 c sæ fdd} j j }}t|r~|jr~t|r~|jr~|j}t|ttfrR||}t t
|
|}|j dk rr| d¡} j |¡ t|râ|jrât|râ|jrâ|j}t|ttfr¶||}t t
|
|}|j dk rÖ| d¡} j |¡ dS )zK
|
Convert datetimes to a comparable value in an expression.
|
c s& j d k rtt j d}nt}|| S )N)r. )r. r r r )rF ÚencoderrB r2 r3 Ú stringifyÐ s
|
z'BinOp.convert_values.<locals>.stringifyNÚUTC)r¨ r© r¤ rb r rF r% re Úfloatr r ÚtzZ
|
tz_convertrP )r> r½ r¨ r© Úvr2 rB r3 r« Ë s$
|
|
|
|
zBinOp.convert_valuesc C sr | j }| j}|j}t|d|}|j}t|d|}|js<|jrn| jtkrnt|tt j
|
frft|tt j
|
fsnt dd S )NrN z$cannot evaluate scalar only bool ops) r© r¨ rj Úgetattrr rt Ú_bool_ops_dictr] r4 r^ r rR )r> r© r¨ Zrhs_rtZlhs_rtr2 r2 r3 rª ì s" ÿÿþûú
|
z$BinOp._disallow_scalar_only_bool_ops)
|
rZ rf rg r r? rJ rK r« rª rl r2 r2 r0 r3 r¥ j s
|
1!r¥ c C s t t | ¡jtjS r$ )r] r^ rW rN Únumber)rW r2 r2 r3 Ú isnumeric s rÅ c s& e Zd ZdZdd fdd Z ZS )ÚDivz
|
Div operator to special case casting.
|
|
Parameters
|
----------
|
lhs, rhs : Term or Op
|
The Terms or Ops in the ``/`` expression.
|
r6 r7 c sj t d||¡ t|jr$t|jsFtd| j d|j d|j dtjtjg}t t
|
| ¡|tj d S )Nr z unsupported operand type(s) for z: 'z' and 'ú') r( r? rÅ rj Ú TypeErrorrt r^ Úfloat32Zfloat_r¢ r
r )r> r¨ r© r r0 r2 r3 r? s ÿ z Div.__init__)rZ rf rg r r? rl r2 r2 r0 r3 rÆ s rÆ )r r rm rp c sV e Zd ZdZddd fdd Zddd d
|
Zd dd d ZeddddZ Z 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.
|
z Literal[('+', '-', '~', 'not')]r6 r¦ c
|
sf t ||f¡ || _zt| | _W n< tk
|
r` } z tdt| dt |W 5 d }~X Y nX d S )NzInvalid unary operator r§ ) r( r? r Ú_unary_ops_dictr r® r± r[ ÚUNARY_OPS_SYMS)r> rt r r² r0 r2 r3 r? 4 s ÿýzUnaryOp.__init__ÚMathCallr7 c C s | |¡}| |¡S r$ )r r )r> r, r r2 r2 r3 rJ @ s
|
zUnaryOp.__call__r& c C s t | j d| j dS )Nrz rY )r rt r rB r2 r2 r3 rD E s zUnaryOp.__repr__znp.dtypec C sR | j }|jt d¡kr t d¡S t|trH|jtks>|jtkrHt d¡S t d¡S )Nr4 re ) r rj r^ rW r% rs rt Ú _cmp_ops_dictrà )r> r r2 r2 r3 rj H s
|
|
ÿÿ
|
zUnaryOp.return_type)
|
rZ rf rg r r? rJ rD ri rj rl r2 r2 r0 r3 rÊ # s rÊ c s8 e Zd Zdd fdd Zdd Zdddd Z ZS )
|
rÍ r6 r7 c s t |j|¡ || _d S r$ )r( r? r+ r )r> r rH r0 r2 r3 r? U s zMathCall.__init__c
|
sD fdd| j D }tjdd | jj| W 5 Q R £ S Q R X d S )Nc s g | ] }| qS r2 r2 )r{ rt r» r2 r3 Ú
|
<listcomp>[ s z%MathCall.__call__.<locals>.<listcomp>Úignore)r )ru r^ Zerrstater )r> r, ru r2 r» r3 rJ Y s zMathCall.__call__r& c C s( t t| j}t| j dd |¡ dS )Nrz ú,rY )Úmapr& ru r rt r~ )r> ru r2 r2 r3 rD _ s zMathCall.__repr__)rZ rf rg r? rJ rD rl r2 r2 r0 r3 rÍ T s rÍ c @ s$ e Zd ZdddddZdd ZdS ) ÚFuncNoder& r6 )r+ r8 c C s. |t krtd| d|| _tt|| _d S )Nú"z " is not a supported function)ÚMATHOPSr± r+ r r^ r )r> r+ r2 r2 r3 r? e s zFuncNode.__init__c G s
|
t | |S r$ )rÍ )r> rH r2 r2 r3 rJ k s zFuncNode.__call__N)rZ rf rg r? rJ r2 r2 r2 r3 rÓ d s rÓ )Xr Ú
|
__future__r r Ú functoolsr ÚoperatorÚtypingr r r r Únumpyr^ Zpandas._libs.tslibsr Zpandas.core.dtypes.commonr
|
r Zpandas.core.commonÚcoreÚcommonr
Z pandas.core.computation.commonr r Z pandas.core.computation.scoper Zpandas.io.formats.printingr r Z
|
REDUCTIONSZ_unary_math_opsZ_binary_math_opsrÕ r; r# r' rv rs r r r ÚgtÚltÚgeÚleÚeqÚneZ_cmp_ops_funcsÚdictÚziprÎ r Úand_Úor_Z_bool_ops_funcsrà ZARITH_OPS_SYMSÚaddÚsubÚmulÚtruedivÚpowÚfloordivÚmodZ_arith_ops_funcsZ_arith_ops_dictZSPECIAL_CASE_ARITH_OPS_SYMSZ _special_case_arith_ops_funcsZ _special_case_arith_ops_dictr¬ ÚdrP r¢ r¤ r¥ rÅ rÆ rÌ ÚposÚnegÚinvertZ_unary_ops_funcsrË rÊ rÍ rÓ r2 r2 r2 r3 Ú<module> s t 6ø
|
ù ÿ 1
|
