Файловый менеджер - Редактировать - /home/digitalm/venv/lib/python3.7/site-packages/pandas/core/computation/__pycache__/pytables.cpython-37.pyc
Назад
B �5�g_L � @ s� d Z ddlmZ ddlZddlmZ ddlmZ ddlZ ddl m Z m Z ddl mZ ddlmZ ddlm mZ dd lmZmZmZ dd lmZ dd l m Z dd l m Z m Z dd l!m"Z" ddl#m$Z$ ddl%m&Z&m'Z' G dd� dej(�Z)G dd� dej*�Z*G dd� de*�Z+G dd� dej,�Z,G dd� de,�Z-G dd� de-�Z.G d d � d e,�Z/G d d � d e/�Z0G d d!� d!ej1�Z1G d"d#� d#e �Z2d$d%� Z3G d&d'� d'ej4�Z5G d(d)� d)�Z6d*d+�d,d-�Z7dS ).z1 manage PyTables query interface via Expressions � )� annotationsN)�partial)�Any)� Timedelta� Timestamp)� DeepChainMap)� is_list_like)�expr�ops�scope)�ensure_decoded)�BaseExprVisitor)�UndefinedVariableError�is_term)� extract_array)�Index)� pprint_thing�pprint_thing_encodedc s4 e Zd ZU dZded< d ddd�� fdd � Z� ZS ) � PyTablesScope)� queryableszdict[str, Any]r N�intzdict[str, Any] | None)�levelr c s$ t � j|d ||d� |p i | _d S )N� )� global_dict� local_dict)�super�__init__r )�selfr r r r )� __class__� �V/home/digitalm-up/venv/lib/python3.7/site-packages/pandas/core/computation/pytables.pyr , s zPyTablesScope.__init__)NNN)�__name__� __module__� __qualname__� __slots__�__annotations__r � __classcell__r r )r r r ' s r c sL e Zd ZU ded< d dd�Zddd�� fdd� Zd d � Zed d � �Z� Z S )�Termr �envNc C s t |t�r| }nt}t�|�S )N)� isinstance�str�Constant�object�__new__)�cls�namer( �side�encoding�klassr r r r- : s z Term.__new__)r( c s t � j||||d� d S )N)r0 r1 )r r )r r/ r( r0 r1 )r r r r A s z Term.__init__c C sb | j dkr4| j| jjkr.tdt| j�� d���| jS y| jj| jdd�S tk r\ | jS X d S )N�leftzname z is not definedF)�is_local)r0 r/ r( r � NameError�repr�resolver )r r r r � _resolve_nameD s zTerm._resolve_namec C s | j S )N)�_value)r r r r �valueS s z Term.value)NN)NN) r! r" r# r% r- r r8 �propertyr: r& r r )r r r' 7 s r' c s, e Zd Zddd�� fdd� Zdd� Z� ZS ) r+ Nr )r( c s. t |t�stt|���t� j||||d� d S )N)r0 r1 )r) r �AssertionError�typer r )r r: r( r0 r1 )r r r r Y s zConstant.__init__c C s | j S )N)�_name)r r r r r8 ] s zConstant._resolve_name)NN)r! r" r# r r8 r& r r )r r r+ X s r+ c s� e Zd ZU dZded< ded< ded< ddd�� fd d � Zd d � Zd d� Zdd� Ze dd�dd��Z e dd�dd��Z e dd� �Z e dd� �Z e dd � �Zdd�d d �Zd d�d d!�Zd"d#� Z� ZS )$�BinOp� r* �opzdict[str, Any]r z str | None� condition)rA r c s& t � �|||� || _|| _d | _d S )N)r r r r1 rB )r rA �lhs�rhsr r1 )r r r r i s zBinOp.__init__c C s d S )Nr )r r r r � _disallow_scalar_only_bool_opso s z$BinOp._disallow_scalar_only_bool_opsc s� � �fdd�}�j �j }}t|�r<t|�r<||j|j�}nnt|�s`t|�r`||�� �|j�}nJt|�r�t|�s�||j|�� ��}n&t|�s�t|�s�||�� �|�� ��}|S )Nc s� | dkr |S |dkr| S � }t | t�rTt |t�r6t}q�t | |�rD| S t ||�r�|S n6t | t�r�t |t�rnt}n t | |�r|| S t ||�r�|S |�j| |�j�jd��� S )z5create and return a new specialized BinOp from myselfN)r r1 ) r) �ConditionBinOp�JointConditionBinOp� FilterBinOp�JointFilterBinOprA r r1 �evaluate)r3 �right�k)r2 r r r �prs s* zBinOp.prune.<locals>.pr)rC rD r r: �prune)r r2 rM r3 rK �resr )r2 r r rN r s z BinOp.prunec C s& t |�s|g}t|tj�r"|�� }|S )zinplace conform rhs)r r) �npZndarrayZravel)r rD r r r �conform� s z BinOp.conform�bool)�returnc C s | j | jkS )z$return True if this is a valid field)rC r )r r r r �is_valid� s zBinOp.is_validc C s | j �| j�dk S )zx return True if this is a valid column name for generation (e.g. an actual column in the table) N)r �getrC )r r r r � is_in_table� s zBinOp.is_in_tablec C s t | j�| j�dd�S )zthe kind of my field�kindN)�getattrr rU rC )r r r r rW � s z BinOp.kindc C s t | j�| j�dd�S )zthe meta of my field�metaN)rX r rU rC )r r r r rY � s z BinOp.metac C s t | j�| j�dd�S )zthe metadata of my field�metadataN)rX r rU rC )r r r r rZ � s zBinOp.metadatac C s( |� | j�}d| j� d| j� d|� d�S )z2create and return the op string for this TermValue�(� �))�tostringr1 rC rA )r �v�valr r r �generate� s zBinOp.generate� TermValuec s� � fdd�}t � j�}t � j�}|dks0|dkrxt|ttf�rF||�}t |�}t|�}|jdk rj|�d�}t ||j |�S |dks�|dkr�t|t �r�t |�j }nt |d d �j }t t|�||�S |d k� r�t � jd d �}||kr�d}n|j|dd�}t ||d�S |dk�r tt|��}t |||�S |dk�r<t|�}t |||�S |dk�rxt|t ��rd|�� �� dk}nt|�}t |||�S t|t ��r�t |||�d�S td|� dt|�� d|� d���dS )zn convert the expression that is in the term to something that is accepted by pytables c s � j d k rt| � j d�S t| �S )N)r1 )r1 r r )r: )r r r � stringify� s z&BinOp.convert_value.<locals>.stringifyZ datetime64�datetimeN�UTCZ timedelta64� timedelta�s)�unit�categoryT)Z extract_numpy���r3 )r0 �integer�floatrR ) �false�f�no�n�none�0z[]z{}� �stringzCannot compare z of type z to z column)r rW rY r) r rl r �tzZ tz_convertrb r: r* r r rZ Z searchsorted�strip�lowerrR � TypeErrorr= )r r_ rc rW rY rZ �resultr )r r � convert_value� sH zBinOp.convert_valuec C s d S )Nr )r r r r �convert_values s zBinOp.convert_values)r! r" r# �_max_selectorsr% r rE rN rQ r; rT rV rW rY rZ ra rz r{ r& r r )r r r? a s * >r? c @ sP e Zd ZU dZded< dd�dd�Zdd � Zd d � Zd d � Zddd�dd�Z dS )rH Nz tuple[Any, Any, Index] | None�filterr* )rS c C s0 | j d krdS td| j d � d| j d � d��S )NzFilter: Not Initializedz [Filter : [r z] -> [r �])r} r )r r r r �__repr__ s zFilterBinOp.__repr__c C s. | j dk r*| j d | jdd�| j d f| _ | S )zinvert the filterNr T)�invert� )r} �generate_filter_op)r r r r r� s zFilterBinOp.invertc C s | j gS )z return the actual filter format)r} )r r r r �format s zFilterBinOp.formatc C s� | j std| � d���| �| j�}t|�}| jrj| jdkrft|�| jkrf| � � }| j |t |�f| _ | S d S | jdkr�| � � }| j |t |�f| _ nt d| � d���| S )Nzquery term is not valid [r~ )z==z!=z7passing a filterable condition to a non-table indexer [)rT � ValueErrorrQ rD �listrV rA �lenr| r� rC r r} rx )r rD �valuesZ filter_opr r r rJ s zFilterBinOp.evaluateFrR )r� c C s0 | j dkr|r | j dkr$|r$dd� S dd� S d S )Nz!=z==c S s | � |� S )N)�isin)�axis�valsr r r �<lambda>? � z0FilterBinOp.generate_filter_op.<locals>.<lambda>c S s | � |�S )N)r� )r� r� r r r r� A r� )rA )r r� r r r r� = s z FilterBinOp.generate_filter_op)F) r! r" r# r} r% r r� r� rJ r� r r r r rH s rH c @ s e Zd Zdd� Zdd� ZdS )rI c C s t d��d S )Nz unable to collapse Joint Filters)�NotImplementedError)r r r r r� E s zJointFilterBinOp.formatc C s | S )Nr )r r r r rJ H s zJointFilterBinOp.evaluateN)r! r" r# r� rJ r r r r rI D s rI c @ s2 e Zd Zdd�dd�Zdd� Zdd� Zd d � Zd S ) rF r* )rS c C s t d| j� d��S )Nz[Condition : [z]])r rB )r r r r r M s zConditionBinOp.__repr__c C s t d��dS )zinvert the conditionz6cannot use an invert condition when passing to numexprN)r� )r r r r r� P s zConditionBinOp.invertc C s | j S )zreturn the actual ne format)rB )r r r r r� Y s zConditionBinOp.formatc s� � j std� � d���� js d S � �� j�}� fdd�|D �}� jdkr�t|�� jkr~� fdd�|D �}dd�|�� d �� _ q�d S n� � |d �� _ � S ) Nzquery term is not valid [r~ c s g | ]}� � |��qS r )rz )�.0r_ )r r r � <listcomp>g s z+ConditionBinOp.evaluate.<locals>.<listcomp>)z==z!=c s g | ]}� � |��qS r )ra )r� r_ )r r r r� n s r[ z | r] r ) rT r� rV rQ rD rA r� r| �joinrB ra )r rD r� �vsr )r r rJ ] s zConditionBinOp.evaluateN)r! r" r# r r� r� rJ r r r r rF L s rF c @ s e Zd Zdd� ZdS )rG c C s( d| j j� d| j� d| jj� d�| _| S )Nr[ r\ r] )rC rB rA rD )r r r r rJ { s $z JointConditionBinOp.evaluateN)r! r" r# rJ r r r r rG z s rG c @ s e Zd Zdd� ZdS )�UnaryOpc C sh | j dkrtd��| j}|�|�}|d k rdt|t�r>|jd k s\t|t�sdt|t�rd|jd k rd|� � S d S )N�~z$UnaryOp only support invert type ops) rA r� �operandrN � issubclassrF rB rH r} r� )r r2 r� r r r rN � s z UnaryOp.pruneN)r! r" r# rN r r r r r� � s r� c s\ e Zd ZeZeZ� fdd�Zdd� Zdd� Z dd� Z d d � Z d d � Z d d� Z dd� Z� ZS )�PyTablesExprVisitorc sL t � �|||� x6| jD ],}| j| }t| d|� �|f� fdd� � qW d S )N�visit_c s t t|f� �S )N)r r? )�node�bin_op)�kwargsr r r� � r� z.PyTablesExprVisitor.__init__.<locals>.<lambda>)r r � binary_opsZbinary_op_nodes_map�setattr)r r( �engine�parserr� r� Zbin_node)r )r� r r � s z PyTablesExprVisitor.__init__c K sh t |jtjtjf�r&td| �|j��S t |jtj�rN| � | �|j�j | j �S t |jtj �rdt d��d S )Nr� z Unary addition not supported)r) rA �ast�Not�Invertr� �visitr� �USub� const_typer: r( �UAddr� )r r� r� r r r � visit_UnaryOp� s z!PyTablesExprVisitor.visit_UnaryOpc K s | � |j�jS )N)r� r: )r r� r� r r r � visit_Index� s z PyTablesExprVisitor.visit_Indexc K s* t jt �� g|jd |jgd�}| �|�S )Nr )r r3 Z comparators)r� �Compare�Eq�targetsr: r� )r r� r� Zcmprr r r � visit_Assign� s z PyTablesExprVisitor.visit_Assignc K s� | � |j�}| � |j�}y |j}W n tk r6 Y nX t|t�rH|j}y| �|| | j�S tk r� } z"t dt |�� dt |�� ��|�W d d }~X Y nX d S )Nzcannot subscript z with ) r� r: �slice�AttributeErrorr) r' r� r( rx r� r6 )r r� r� r: Zslobj�errr r r �visit_Subscript� s z#PyTablesExprVisitor.visit_Subscriptc K s� |j }|j}t|j�}|tjkr�| �|�}y |j}W n tk rH Y nX y| �t ||�| j �S tk r� t |tj �r�|j |kr�|S Y nX td|j� ���d S )NzInvalid Attribute context )�attrr: r= �ctxr� �Loadr� r� � term_typerX r( r) �Name�idr� r! )r r� r� r� r: r� �resolvedr r r �visit_Attribute� s z#PyTablesExprVisitor.visit_Attributec C s t |tj�rt�� S |S )N)r) r� �Inr� )r rA r r r � translate_In� s z PyTablesExprVisitor.translate_Inc C s | � |j�|j||fS )N)r� rA )r r� r3 rK r r r �_rewrite_membership_op� s z*PyTablesExprVisitor._rewrite_membership_op)r! r" r# r+ r� r' r� r r� r� r� r� r� r� r� r& r r )r r r� � s r� c C s" t | ttf�s t| �s td��| S )a� Validate that the where statement is of the right type. The type may either be String, Expr, or list-like of Exprs. Parameters ---------- w : String term expression, Expr, or list-like of Exprs. Returns ------- where : The original where clause if the check was successful. Raises ------ TypeError : An invalid data type was passed in for w (e.g. dict). zLwhere must be passed as a string, PyTablesExpr, or list-like of PyTablesExpr)r) � PyTablesExprr* r rx )�wr r r �_validate_where� s r� c @ sR e Zd ZU dZded< ded< ded< dd d d �d d�Zdd�dd�Zdd� ZdS )r� a� Hold a pytables-like expression, comprised of possibly multiple 'terms'. Parameters ---------- where : string term expression, PyTablesExpr, or list-like of PyTablesExprs queryables : a "kinds" map (dict of column name -> kind), or None if column is non-indexable encoding : an encoding that will encode the query terms Returns ------- a PyTablesExpr object Examples -------- 'index>=date' "columns=['A', 'D']" 'columns=A' 'columns==A' "~(columns=['A','B'])" 'index>df.index[3] & string="bar"' '(index>df.index[3] & index<=df.index[6]) | string="bar"' "ts>=Timestamp('2012-02-01')" "major_axis>=20130101" zPyTablesExprVisitor | None�_visitorr r( r* r Nr zdict[str, Any] | Noner )r � scope_levelc C s t |�}|| _d | _d | _d | _d | _t� }t|t�rF|j j }|j }njt |�r�t |�}x8t|�D ],\}}t|t�r||j j }q`t |�}|||< q`W d�dd� t�|�D ��}n|}|| _ t|d |d�| _ |d k �rt| j t��r| j j�|� t| j |dd|d�| _| �� | _d S )Nz & c s s | ]}d |� d�V qdS )r[ r] Nr )r� r� r r r � <genexpr>B s z(PyTablesExpr.__init__.<locals>.<genexpr>r )r Zpytables)r r� r� r1 )r� r1 rB r} �termsr� r r) r� r( r r r r� � enumerater� �com�flattenr r* r �updater� �parse) r �wherer r1 r� r Z_where�idxr� r r r r # s< zPyTablesExpr.__init__)rS c C s | j d k rt| j �S t| j�S )N)r� r r )r r r r r U s zPyTablesExpr.__repr__c C s� y| j �t�| _W n< tk rN } z td| j� d| � d��|�W dd}~X Y nX y| j �t�| _W n< tk r� } z td| j� d| � d��|�W dd}~X Y nX | j| jfS )z2create and return the numexpr condition and filterzcannot process expression [z], [z] is not a valid conditionNz] is not a valid filter) r� rN rF rB r� r� r rH r} )r r� r r r rJ Z s zPyTablesExpr.evaluate)NNr )r! r" r# �__doc__r% r r rJ r r r r r� s -r� c @ s, e Zd ZdZdd�dd�Zdd�dd�Zd S ) rb z<hold a term value the we use to construct a condition/filterr* )rW c C s( t |t�st|��|| _|| _|| _d S )N)r) r* r< r: � convertedrW )r r: r� rW r r r r q s zTermValue.__init__)rS c C sH | j dkr*|dk r t| j�S d| j� d�S | j dkr>t| j�S t| j�S )z6quote the string if not encoded else encode and returnrt N�"rl )rW r* r� r6 )r r1 r r r r^ w s zTermValue.tostringN)r! r" r# r� r r^ r r r r rb n s rb rR )rS c s4 t � t�sdS tjtj d }t� fdd�|D ��S )z7loose checking if s is a pytables-acceptable expressionF)�=c 3 s | ]}|� kV qd S )Nr )r� rA )rg r r r� � s z#maybe_expression.<locals>.<genexpr>)r) r* r� r� Z unary_ops�any)rg r r )rg r �maybe_expression� s r� )8r� � __future__r r� � functoolsr �typingr �numpyrP Zpandas._libs.tslibsr r Zpandas.compat.chainmapr Zpandas.core.dtypes.commonr Zpandas.core.common�core�commonr� Zpandas.core.computationr r r Z_scopeZ pandas.core.computation.commonr Z pandas.core.computation.exprr Zpandas.core.computation.opsr r Zpandas.core.constructionr Zpandas.core.indexes.baser Zpandas.io.formats.printingr r ZScoper r' r+ r? rH rI rF rG r� r� r� �Exprr� rb r� r r r r �<module> s> ! '=.Tk
| ver. 1.4 |
Github
|
.
| PHP 8.3.23 | Генерация страницы: 0.06 |
proxy
|
phpinfo
|
Настройка