3
���h_4 � @ s� U d Z ddlmZ ddlmZ ddlmZmZmZm Z ddl
m Z m
Z
ddlmZmZ ddlmZ ddlmZmZmZ G d d
� d
�Zed d �Zeed
ddd�Zeedddd�Zdd� Z dd� Z dd� Z e� Z d dae d< de d< d e d < ee �ed �k�rd e d< ee d ddd �Z e� Z!d !dbe!d< d e!d < ee!d ddd �Z"d!d"� Z#ed d �Z$ee%ef $ee$d#dd$d�Z&d%d&� Z'e� Z(d'(d e(d(< d e(d)< ee(ddd*�Z)ee(d+ddd�Z*d,d-� Z+e� Z,d.,d e,d(< d e,d)< d/e,d0< ee,d1ddd�Z-ee,d2ddd�Z.ed d/d3�Z/ee/d4d5�Z0ed d d/d6�Z1ee1d7ddd�Z2ee1d8ddd�Z3ed9d:�Z4ee%e%f 4ee4d;ddd�Z5ed d<�Z6ee%ef 6ee6d=ddd�Z7ed d �Z8ee%ef 8ee8d>ddd�Z9e� Z:d:dce:d< de:d< d e:d < ee:d?d4d@�Z;e� Z<dA<d e<d(< d e<d)< e<j=� Z>d e>d< d/e>d0< d e>dB< e<j=� Z?d e?d< d/e?d0< d e?dB< e<j=� Z@d/e@dC< d/e@d0< d/e<d0< ee<d4d5�ZAee>dDddd�ZBee?dEddd�ZCee<dFddd�ZDee@dGddd�ZEe� ZFd.Fd eFd(< d eFd)< d/eFd0< eeFd4d5�ZGe� ZHdHHd eHd)< dIeHdJ< eeHdKd4d@�ZIdLdM� ZJed dN�ZKeeKdOddd�ZLd dP�dQdR�ZMd dP�dSdT�ZNd dP�dUdV�ZOddd dP�dWdX�ZPdeZQe%d d[�d\d]�ZReeS d d^�d_d`�ZTd S )fa�
For compatibility with numpy libraries, pandas functions or
methods have to accept '*args' and '**kwargs' parameters to
accommodate numpy arguments that are not actually used or
respected in the pandas implementation.
To ensure that users do not abuse these parameters, validation
is performed in 'validators.py' to make sure that any extra
parameters passed correspond ONLY to those in the numpy signature.
Part of that validation includes whether or not the user attempted
to pass in non-default values for these extraneous parameters. As we
want to discourage users from relying on these parameters when calling
the pandas implementation, we want them only to pass in the default values
for these parameters.
This module provides a set of commonly used default arguments for functions
and methods that are spread throughout the codebase. This module will make it
easier to adjust to future upstream changes in the analogous numpy signatures.
� )� OrderedDict)� LooseVersion)�Any�Dict�Optional�Union)� __version__�ndarray)�is_bool�
is_integer)�UnsupportedFunctionCall)�
validate_args�validate_args_and_kwargs�validate_kwargsc @ s6 e Zd Zdee d�dd�Zd ee dd�dd�ZdS )
�CompatValidatorN)�methodc C s || _ || _|| _|| _d S )N)�fnamer �defaults�max_fname_arg_count)�selfr r r r � r �>/tmp/pip-build-5_djhm0z/pandas/pandas/compat/numpy/function.py�__init__$ s zCompatValidator.__init__)r �returnc C s� |s|r�|d kr| j n|}|d kr(| jn|}|d kr:| jn|}|dkrXt|||| j� nD|dkrpt||| j� n,|dkr�t||||| j� ntd|� d���d S )N�args�kwargs�bothzinvalid validation method '�')r r r r
r r r �
ValueError)r r r r r r r r r �__call__0 s zCompatValidator.__call__)NNN)NNN)�__name__�
__module__� __qualname__r �strr r r r r r r # s r N)�outZargminr � )r r r Zargmaxc C s( t | t�s| d kr | f| }d} | |fS )NT)�
isinstancer )�skipnar r r r �process_skipnaV s
r( c C s t | |�\} }t||� | S )a!
If 'Series.argmin' is called via the 'numpy' library,
the third parameter in its signature is 'out', which
takes either an ndarray or 'None', so check if the
'skipna' parameter is either an instance of ndarray or
is None, since 'skipna' itself should be a boolean
)r( �validate_argmin)r' r r r r r �validate_argmin_with_skipna^ s
r* c C s t | |�\} }t||� | S )a!
If 'Series.argmax' is called via the 'numpy' library,
the third parameter in its signature is 'out', which
takes either an ndarray or 'None', so check if the
'skipna' parameter is either an instance of ndarray or
is None, since 'skipna' itself should be a boolean
)r( �validate_argmax)r' r r r r r �validate_argmax_with_skipnak s
r, z+OrderedDict[str, Optional[Union[int, str]]]�axisZ quicksort�kind�orderz1.17.0Zargsort)r r r z OrderedDict[str, Optional[int]]c C s0 t | �s| dkr | f| }d} t||dd� | S )a%
If 'Categorical.argsort' is called via the 'numpy' library, the
first parameter in its signature is 'axis', which takes either
an integer or 'None', so check if the 'ascending' parameter has
either integer type or is None, since 'ascending' itself should
be a boolean
NT� )r )r �validate_argsort_kind)Z ascendingr r r r r � validate_argsort_with_ascending� s
r2 Zclipr0 c C s& t | t�r| f| }d} t||� | S )a
If 'NDFrame.clip' is called via the numpy library, the third
parameter in its signature is 'out', which can takes an ndarray,
so check if the 'axis' parameter is an instance of ndarray, since
'axis' itself should either be an integer or None
N)r& r �
validate_clip)r- r r r r r �validate_clip_with_axis� s
r4 zOrderedDict[str, Any]�dtyper$ )r r Zcumsumc C s( t | �s| f| }d} t|||d� | S )z�
If this function is called via the 'numpy' library, the third
parameter in its signature is 'dtype', which takes either a
'numpy' dtype or 'None', so check if the 'skipna' parameter is
a boolean or not
T)r )r
�validate_cum_func)r' r r �namer r r � validate_cum_func_with_skipna� s
r8 z OrderedDict[str, Optional[bool]]F�keepdims�all�any)r$ r9 r )r )r- r$ r9 �min�max�C)r/ Zreshape)r- �repeat�round�sort)r r z OrderedDict[str, Optional[Any]]�initialZoverwrite_input�sum�prod�meanZmedianz OrderedDict[str, Optional[str]]�raise�modeZtakec C s4 t | t�s| dkr | f| }d} t||ddd� | S )z�
If this function is called via the 'numpy' library, the third
parameter in its signature is 'axis', which takes either an
ndarray or 'None', so check if the 'convert' parameter is either
an instance of ndarray or is None
NTr0 r )r r )r& r �
validate_take)�convertr r r r r �validate_take_with_convert( s
rJ )ZaxesZ transpose)r c C sF d}d| � d�}t |�dkr$t|��x |D ]}||kr*t|��q*W d S )Nr- r5 r$ z9numpy operations are not valid with window objects. Use .z() directly instead r )r- r5 r$ )�lenr )r7 r r �
numpy_args�msg�argr r r �validate_window_func= s
rO c C sF d}d| � d�}t |�dkr$t|��x |D ]}||kr*t|��q*W d S )Nr- r5 r$ zFnumpy operations are not valid with window objects. Use .rolling(...).z () instead r )r- r5 r$ )rK r )r7 r r rL rM rN r r r �validate_rolling_funcL s
rP c C sF d}d| � d�}t |�dkr$t|��x |D ]}||kr*t|��q*W d S )Nr- r5 r$ zHnumpy operations are not valid with window objects. Use .expanding(...).z () instead r )r- r5 r$ )rK r )r7 r r rL rM rN r r r �validate_expanding_func[ s
rQ c C sD |dkr g }t |�t |� }t|�t|� dkr@td| � d���dS )z�
'args' and 'kwargs' should be empty, except for allowed
kwargs because all of
their necessary parameters are explicitly listed in
the function signature
Nr z?numpy operations are not valid with groupby. Use .groupby(...).z
() instead)�setrK r )r7 r r �allowedr r r �validate_groupby_funcj s rT �std�var)r r c C s: t |�t |� dkr6| tkr.td| � d���ntd��dS )z�
'args' and 'kwargs' should be empty because all of
their necessary parameters are explicitly listed in
the function signature
r zAnumpy operations are not valid with resample. Use .resample(...).z
() insteadz too many arguments passed inN)rK �RESAMPLER_NUMPY_OPSr � TypeError)r r r r r r �validate_resampler_func� s
rY )r- r c C s@ d}| dkrdS | |ks,| dk r<|| dk r<t d|� d���dS )z�
Ensure that the axis argument passed to min, max, argmin, or argmax is
zero or None, as otherwise it will be incorrectly ignored.
Parameters
----------
axis : int or None
Raises
------
ValueError
r% Nr z4`axis` must be fewer than the number of dimensions (�))r )r- �ndimr r r �validate_minmax_axis� s
r\ ���r] r] )N)r< r= rC rD rE rU rV )U�__doc__� collectionsr Zdistutils.versionr Ztypingr r r r Znumpyr Z _np_versionr Zpandas._libs.libr
r Z
pandas.errorsr Zpandas.util._validatorsr
r r r �dictZARGMINMAX_DEFAULTSr) r+ r( r* r, ZARGSORT_DEFAULTSZvalidate_argsortZARGSORT_DEFAULTS_KINDr1 r2 Z
CLIP_DEFAULTSr# r3 r4 ZCUM_FUNC_DEFAULTSr6 Zvalidate_cumsumr8 ZALLANY_DEFAULTSZ validate_allZ validate_anyZLOGICAL_FUNC_DEFAULTSZvalidate_logical_funcZMINMAX_DEFAULTSZ validate_minZ validate_maxZRESHAPE_DEFAULTSZvalidate_reshapeZREPEAT_DEFAULTSZvalidate_repeatZROUND_DEFAULTSZvalidate_roundZ
SORT_DEFAULTSZ
validate_sortZSTAT_FUNC_DEFAULTS�copyZ SUM_DEFAULTSZ
PROD_DEFAULTSZMEDIAN_DEFAULTSZvalidate_stat_funcZ validate_sumZ
validate_prodZ
validate_meanZvalidate_medianZSTAT_DDOF_FUNC_DEFAULTSZvalidate_stat_ddof_funcZ
TAKE_DEFAULTSrH rJ ZTRANSPOSE_DEFAULTSZvalidate_transposerO rP rQ rT rW rY �intr\ r r r r �<module> s� *
|