3
���h� � @ s d Z ddlmZmZ ddlmZmZmZmZm Z m
Z
ddl Z ddl Z
ddlmZ ddlmZ ddlmZmZ ddlmZ ddljjZdd lm Z ee
ee e f eee e f f d
�d d �Z e e ee ee e f d
�dd�Z e!d
�dd�Z"dd� Z#e#dd� �Z$dd� Z%dd� Z&dS )z!
Core eval alignment algorithms.
� )�partial�wraps)�Dict�Optional�Sequence�Tuple�Type�UnionN)�
FrameOrSeries)�PerformanceWarning)� ABCDataFrame� ABCSeries)� PandasObject)�result_type_many)�returnc C sR d }t | jtj�r&ttj| jjd�}n$t| j�}t| jd�rJt || jj
�}||fS )N)�dtype�axes) �
isinstance�value�np�ndarrayr Z
asanyarrayr �type�hasattr�_zip_axes_from_typer )�termr �typ� r �?/tmp/pip-build-5_djhm0z/pandas/pandas/core/computation/align.py�_align_core_single_unary_op s
r )r �new_axesr c s � fdd�t | j�D �}|S )Nc s i | ]\}}� | |�qS r r )�.0�i�name)r r r �
<dictcomp>) s z'_zip_axes_from_type.<locals>.<dictcomp>)� enumerateZ _AXIS_ORDERS)r r r r )r r r & s r c C s t dd� | D ��S )zB
Check a sequence of terms for instances of PandasObject.
c s s | ]}t |jt�V qd S )N)r r r )r r r r r � <genexpr>1 s z&_any_pandas_objects.<locals>.<genexpr>)�any)�termsr r r �_any_pandas_objects- s r( c s t � �� fdd��}|S )Nc sB t | �dkrt| d �S dd� | D �}t| �s:t|� d fS � | �S )N� r c s s | ] }|j V qd S )N)r )r r r r r r% ; s z9_filter_special_cases.<locals>.wrapper.<locals>.<genexpr>)�lenr r( r )r' Z term_values)�fr r �wrapper5 s z&_filter_special_cases.<locals>.wrapper)r )r+ r, r )r+ r �_filter_special_cases4 s r- c s, dd� t � �D �}� fdd�|D �}ddlm} |tt||���}� |j� j}|j}|j}t |�}|dk} x�� fdd�|D �D ]r}
t
|
t �} | o�| } xZt |
j�D ]L\}
}| r�|d |
j }}n
|
| }}|| j
|�s�|| j|d d
�||< q�W q�W �x$|j� D �]\}}x�tt|�|�D ]�\}
}� | j}t|d ��r t
|t ��oL|dk}|�r`||d n|}t |j|
�}t |�}tjtdt|| ���}|dk�r�|d k�r�d
|
� dt� | j�� d|d�d�}tj|tdd� t|j ||
dd�}� | j |� � �q W � | j � | jj � �qW |t ||�fS )Nc S s g | ]\}}t |jd �r|�qS )r )r r )r r! r r r r �
<listcomp>H s z _align_core.<locals>.<listcomp>c s g | ]}� | j j�qS r )r �ndim)r r! )r' r r r. I s r )�Seriesr) c 3 s | ]}� | j V qd S )N)r )r r! )r' r r r% V s z _align_core.<locals>.<genexpr>�outer)�how�reindexi' z Alignment difference on axis z. is larger than an order of magnitude on term z, by more than z.4gz; performance may suffer� )�category�
stacklevelF)�axis�copy) r$ Zpandasr0 �dict�zipZidxmaxr Z _constructorr r* r r
�index�is_�join�items�ranger r �log10�max�abs�reprr" �warnings�warnr r r3 �update�valuesr )r' Z
term_indexZ term_dimsr0 ZndimsZbiggestr r ZnaxesZgt_than_one_axisr Z is_seriesZis_series_and_gt_one_axisr7 r> ZaxZitmr! r/ ZtiZ transposeZ reindexerZterm_axis_sizeZreindexer_size�ordm�wr+ r )r' r � _align_coreF sD
$ rJ c C s� yt tj| ��} W nL tk
r^ t| jttf�rNt| j�}|t || jj
�fS t j | j�dfS X t
dd� | D ��r�tdd� | D �� jdfS t| �\}}||fS )z
Align a set of terms.
Nc s s | ] }|j V qd S )N)Z is_scalar)r r r r r r% � s z align_terms.<locals>.<genexpr>c s s | ] }|j V qd S )N)r )r r r r r r% � s )�list�com�flatten� TypeErrorr r r
r r r r r � result_type�allr rJ )r' r r r r r � align_terms� s
rQ c C s� y
| j } W n tk
r Y nX tj|j|�}t| t� rXt| t�rX| |fd|i|��S t |d�r�| tj
kr�|tj
kr�|j |�}nF| |�j |�}t |j
�dkr�t |�dkr�t|tj�s�tj|g�j |�}|S )a�
Reconstruct an object given its type, raw value, and possibly empty
(None) axes.
Parameters
----------
typ : object
A type
obj : object
The value to use in the type constructor
axes : dict
The axes to use to construct the resulting pandas object
Returns
-------
ret : typ
An object of type ``typ`` with the value `obj` and possible axes
`axes`.
r r r) )r �AttributeErrorr rO r r r �
issubclassr r Zbool_Zastyper* �shaper �array)r �objr r Zres_tZ ret_valuer r r �reconstruct_object� s
rW )'�__doc__� functoolsr r Ztypingr r r r r r rD Znumpyr Zpandas._typingr
Z
pandas.errorsr Zpandas.core.dtypes.genericr r
Zpandas.core.baser Zpandas.core.common�core�commonrL Z pandas.core.computation.commonr �str�intr r �boolr( r- rJ rQ rW r r r r �<module> s$ . : |