3
���hX6 � @ s d dl Z d dlZd dlmZmZmZmZmZmZm Z m
Z
d dl Z d dl
mZ d dlmZ d dlmZ d dlmZmZmZmZ d dlmZ d dlmZ er�d d l m Z m Z m Z ee ef Z!dd ee"ee# e"d �d
d�Z$G dd� de j%d�Z&G dd� de&�Z'G dd� de&�Z(dS )� N)�
TYPE_CHECKING�Any�Dict�Iterator�Optional�Tuple�Type�Union)�option_context)�Axis)�cache_readonly)� is_dict_like�is_extension_array_dtype� is_list_like� is_sequence)� ABCSeries)�!create_series_with_explicit_dtype)� DataFrame�Index�SeriesFr )�obj�axis�raw� result_type�ignore_failuresc C s: | j |�}|dkrt}n |dkr$t}|| ||||||d�S )z? construct and return a row or column based frame apply object r � )r r r �args�kwds)Z_get_axis_number�
FrameRowApply�FrameColumnApply) r �funcr r r r r r �klass� r" �3/tmp/pip-build-5_djhm0z/pandas/pandas/core/apply.py� frame_apply s
r$ c @ sT e Zd ZU eeejdd�dd���Zeejdd�dd���Z eeje
d d�dd ���Z eje de
d- d �d d
��Zd
eee ed�dd�Zedd�dd��Zedd�dd��Zedd�dd��Zedd� �Zedd�dd��Zedd�dd ��Zd d � Zd d � Zd!d"� Z d
d
d#�d$d%�Z d&d'� Z e e df d�d(d)�Z e de
d. d �d*d+�Z!d,S )/�
FrameApplyr )�returnc C s d S )Nr" )�selfr" r" r# � result_index? s zFrameApply.result_indexc C s d S )Nr" )r' r" r" r# �result_columnsD s zFrameApply.result_columnsr c C s d S )Nr" )r' r" r" r# �series_generatorI s zFrameApply.series_generatorr )�results� res_indexr& c C s d S )Nr" )r' r+ r, r" r" r# �wrap_results_for_axisN s z FrameApply.wrap_results_for_axis)r r r r c sv || _ || _|| _� pf | _�p"i | _|dkr6td��|| _�sD� rht�tj t
f� rh� ��fdd�}n�}|| _ d S )N�reduce� broadcast�expandzUinvalid value for result_type, must be one of {None, 'reduce', 'broadcast', 'expand'}c s �| f� ���S )Nr" )�x)r r r r" r# �fq s z FrameApply.__init__.<locals>.f)Nr. r/ r0 ) r r r r r �
ValueErrorr �
isinstance�np�ufunc�strr2 ) r' r r r r r r r r2 r" )r r r r# �__init__V s
zFrameApply.__init__c C s | j S )N)r) )r' r" r" r# � res_columnsy s zFrameApply.res_columnsc C s | j jS )N)r �columns)r' r" r" r# r: } s zFrameApply.columnsc C s | j jS )N)r �index)r' r" r" r# r; � s zFrameApply.indexc C s | j jS )N)r �values)r' r" r" r# r< � s zFrameApply.valuesc C s | j jS )N)r �dtypes)r' r" r" r# r= � s zFrameApply.dtypesc C s | j j| j�S )N)r Z
_get_agg_axisr )r' r" r" r# �agg_axis� s zFrameApply.agg_axisc C s2 t | j�st| j�r8| jj| jf| j�d| ji| j��S t| j �dkr\t| j
�dkr\| j � S t | jt
�r�t| j| j�}tj|�}d|jkr�| j| jd< || j| j�S t | jtj�r�tjdd�� | jjjd| jd�}W dQ R X | jj|d�S | jd k�r| j| j�S t| jj��s| j � S | j�r*| j � S | j � S )
z compute the results r r �ignore)�all�apply)r N)�datar/ ) r r2 r
r Z aggregater r r �lenr: r; �apply_empty_resultr4 r7 �getattr�inspectZgetfullargspecr5 r6 Zerrstate�_mgrrA � _constructorr �apply_broadcastr@ �shaper � apply_raw�apply_standard)r' r �sigr+ r" r" r# �
get_result� s* $
zFrameApply.get_resultc
C s� | j dkr| jj� S | j dk}ddlm} |shy| j|g tjd��}W n tk
rZ Y nX t ||� }|r�t
| j �r�| j|g tjd��}ntj }| jj
|| j d�S | jj� S dS )z�
we have an empty result; at least 1 axis is 0
we will try to apply the function to an empty
series in order to see if this is a reduction function
r. Nr )r )�dtype)r; )r. N)r r �copy�pandasr r2 r5 �float64� Exceptionr4 rC r> �nan�_constructor_sliced)r' Z
should_reducer �rr" r" r# rD � s
z FrameApply.apply_empty_resultc C sV dd� }t j|| j�| j| j�}|jdkr@| jj|| j| j d�S | jj
|| j d�S dS )z& apply to the values as a numpy array c s � fdd�}|S )z�
Wrap user supplied function to work around numpy issue.
see https://github.com/numpy/numpy/issues/8352
c s& � | |�}t |t�r"tj|td�}|S )N)rO )r4 r7 r5 �array�object)r �kwargs�result)r r" r# �wrapper� s
z<FrameApply.apply_raw.<locals>.wrap_function.<locals>.wrapperr" )r r[ r" )r r# �
wrap_function� s z+FrameApply.apply_raw.<locals>.wrap_function� )r; r: )r; N) r5 Zapply_along_axisr2 r r< �ndimr rH r; r: rU r> )r' r\ rZ r" r" r# rK � s
zFrameApply.apply_raw)�targetr& c C s� t j|j�}|jd }xnt|j�D ]`\}}| j|| �}t j|�j}|dkrVt d��n |dkrr|t
|�krrt d��||d d �|f< q"W | j j ||j
|jd�}|S )Nr r ztoo many dims to broadcastzcannot broadcast result)r; r: )r5 Z
empty_liker< rJ � enumerater: r2 Zasarrayr^ r3 rC r rH r; ) r' r_ Z
result_valuesZresult_compare�i�col�resZaresrZ r" r" r# rI � s
zFrameApply.apply_broadcastc C s | j � \}}| j||�S )N)�apply_series_generator� wrap_results)r' r+ r, r" r" r# rL s zFrameApply.apply_standardc C s� | j }| j}i }| jr~g }xFt|�D ]:\}}y| j|�||< W n tk
rR Y q$X |j|� q$W t|�t|�k r�|j|�}n\t dd ��L xDt|�D ]8\}}| j|�||< t
|| t �r�|| j dd�||< q�W W d Q R X ||fS )Nzmode.chained_assignmentF)�deep)
r* r( r r` r2 rS �appendrC Ztaker
r4 r rP )r' Z
series_genr, r+ Z successesra �vr" r" r# rd s& "z!FrameApply.apply_series_generatorc C sj ddl m} t|�dkr8d|kr8t|d �r8| j||�S | jj}||krXt|tj d�}n||�}||_
|S )Nr )r )Zdtype_if_empty) rQ r rC r r- r rU r r5 rR r; )r' r+ r, r Zconstructor_slicedrZ r" r" r# re 9 s zFrameApply.wrap_resultsN)r r )r r )"�__name__�
__module__� __qualname__�intr �property�abc�abstractmethodr( r) r r* �ResTyper r- �boolr r7 r8 r9 r: r; r r< r= r> rN rD rK rI rL r rd re r" r" r" r# r% 9 sH
*#!r% )� metaclassc sn e Zd ZdZddd�� fdd� Zedd� �Zedd �d
d ��Zedd �d d
��Ze de
d d�dd�Z � Z S )r r r )r_ r& c s t � j|�S )N)�superrI )r' r_ )� __class__r" r# rI V s z FrameRowApply.apply_broadcastc s � fdd�t t� j��D �S )Nc 3 s | ]}� j j|d d�V qdS )r )r N)r �_ixs)�.0ra )r' r" r# � <genexpr>[ s z1FrameRowApply.series_generator.<locals>.<genexpr>)�rangerC r: )r' r" )r' r# r* Y s z FrameRowApply.series_generatorr )r& c C s | j S )N)r: )r' r" r" r# r( ] s zFrameRowApply.result_indexc C s | j S )N)r; )r' r" r" r# r) a s z FrameRowApply.result_columnsr )r+ r, r& c C s� | j dkr | jj|�}||_|S | j dkrLtdd� |j� D ��rL| jj|�S y| jj|d�}W nF tk
r� } z*dt|�kr�| jj|�}||_|S � W Y dd}~X nX t |d t
�s�t |j�t | j �kr�| j |_t |j
�t |�kr�||_
|S )z! return the results for the rows r. Nc s s | ]}t |t�V qd S )N)r4 �dict)rv r1 r" r" r# rw q s z6FrameRowApply.wrap_results_for_axis.<locals>.<genexpr>)rB z arrays must all be same lengthr )r r rU r; r@ r< rH r3 r7 r4 r rC r9 r: )r' r+ r, rc rZ �errr" r" r# r- e s*
z#FrameRowApply.wrap_results_for_axis)r r )
ri rj rk r rI rm r* r( r) rp r r- �
__classcell__r" r" )rt r# r S s r c s� e Zd ZdZddd�� fdd� Zedd� �Zedd �d
d ��Zedd �d d
��Ze de
d d�dd�Z e ddd�dd�Z � Z
S )r r r )r_ r& c s t � j|j�}|jS )N)rs rI �T)r' r_ rZ )rt r" r# rI � s z FrameColumnApply.apply_broadcastc c s� | j }t|�dkst�| jjddd�}|j}|jd }t|j�rn| j}xXt t|��D ]}|j|dd�V qTW n2x0t
|| j �D ] \}}||_||_ ||_ |V q|W d S )Nr )r )
r< rC �AssertionErrorr ru rG �blocksr rO rx �zipr; �name) r' r< ZserZmgrZblkr ra Zarrr� r" r" r# r* � s
z!FrameColumnApply.series_generatorr )r& c C s | j S )N)r; )r' r" r" r# r( � s z FrameColumnApply.result_indexc C s | j S )N)r: )r' r" r" r# r) � s z FrameColumnApply.result_columnsr )r+ r, r& c C sJ | j dkr| j||�}n.t|d t�s:| jj|�}||_n | j||�}|S )z$ return the results for the columns r0 r )r �infer_to_same_shaper4 r r rU r; )r' r+ r, rZ r" r" r# r- � s
z&FrameColumnApply.wrap_results_for_axisc C s& | j j|d�}|j}||_|j� }|S )z9 infer the results to the same shape as the input object )rB )r rH r| r; Z
infer_objects)r' r+ r, rZ r" r" r# r� � s
z$FrameColumnApply.infer_to_same_shape)r r )ri rj rk r rI rm r* r( r) rp r r- r� r{ r" r" )rt r# r � s r )r FNFNN))rn rF Ztypingr r r r r r r r Znumpyr5 Zpandas._configr
Zpandas._typingr Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr
r r r Zpandas.core.dtypes.genericr Zpandas.core.constructionr rQ r r r rl rp rq r7 r$ �ABCMetar% r r r" r" r" r# �<module> s. ( : |