3
���hN � @ s� d dl mZmZmZ d dlZd dlmZ erDd dlm Z d dl
m Z dddd�dd �Z ed
d �d d
�Z
d
d
eeeeef d�dd�ZdS )� )�
TYPE_CHECKING�List�TupleN)� ArrayLike)�Block)� BlockManagerr )�left�right�returnc C s� g }x�t | j�D ]�\}}|j}|j}t|tj� }|j|jdd�} x~t | �D ]r\}
} t| jtj� } t || || �\}
}||
|�}|r�| r�t
|d�r�|j dd�}| j |�}t
||� |j|� qLW qW t|�||jdd�}|S )NT)Z
only_slice�reshape� F)�axesZdo_integrity_check���)� enumerate�blocks�mgr_locs�values�
isinstance�npZndarrayZ_slice_take_blocks_ax0�indexer�_get_same_shape_values�hasattrr Z_split_op_result�_reset_block_mgr_locs�extend�typer
)r r Zarray_opZres_blks�nZblk�locsZblk_vals�left_eaZrblks�k�rblk�right_ea�lvals�rvalsZ
res_values�nbsZnew_mgr� r$ �;/tmp/pip-build-5_djhm0z/pandas/pandas/core/internals/ops.py�operate_blockwise s"
r& r )r# c C s& x | D ]}|j |jj }||_qW dS )zA
Reset mgr_locs to correspond to our original DataFrame.
N)Zas_arrayr r )r# r �nbZnblocsr$ r$ r% r @ s
r )�lblkr r r r
c C s� | j }|j }|jjs t|j��|p$|sX||jjdd�f }|j|jks�t|j|jf��n�|r~|r~|j|jks�t|j|jf��nj|r�||jjdd�f }|jd dks�t|j��|ddd�f }n(|jd dks�t|j��|ddd�f }||fS )zH
Slice lblk.values to align with rblk. Squeeze if we have EAs.
Nr r )r r Z
is_slice_like�AssertionErrorr �shape)r( r r r r! r" r$ r$ r% r L s r )Ztypingr r r Znumpyr Zpandas._typingr Z pandas.core.internals.blocksr Z pandas.core.internals.managersr r& r �boolr r$ r$ r$ r% �<module> s 2
|