3
���h�B � @ s� d dl mZ d dlZd dlZd dlmZ d dlmZm Z d dl
m Z m Z m
Z
mZmZ d dlmZmZ d dlmZmZmZmZ d dlmZ d d lmZ d d
l m Z m Z m Z d d l m!Z! d d l"m#Z# d
Z$e$d
Z%e%d
Z&de& Z'de' Z(de( Z)dddddddddddddddddddddd dddddd �Z*d d d d!d"d#d$d%d&g Z+x8e+D ]0Z,x(eD ] Z-e,� d'e-� �Z.e*e, e*e.< �qNW �qDW x4dPD ],Z,x$eD ] Z-e,� d'e-� �Z/e/e*e/< �q�W �q~W x"e D ]Z0d(e0� �e*d(e0� �< �q�W e1ee1 d)�d*d+�Z2e1ed,�d-d.�Z3dQe4ee1 d0�d1d2�Z5G d3d4� d4�Z6G d5d6� d6e6�Z7e8e4d7�d8d9�Z9e1e:e1d:�d;d<�Z;e4d=�d>d?�Z<e4d=�d@dA�Z=e1d=�dBdC�Z>e1e1e4dD�dEdF�Z?e1e4dG�dHdI�Z@e1e4dG�dJdK�ZAe1e4dG�dLdM�ZBe1e4dG�dNdO�ZCdS )R� )�OptionalN)�
unique_deltas)� Timestamp� tzconversion)�DAYS�
MONTH_ALIASES�
MONTH_NUMBERS�MONTHS�int_to_weekday)�build_field_sarray�month_position_check)�
DateOffset�Day� _get_offset� to_offset)�get_rule_month)�cache_readonly)�is_datetime64_dtype�is_period_dtype�is_timedelta64_dtype)� ABCSeries)�uniquei� �< � �D�M�Q�A�C�B�T�S�L�U�N�H�W)ZWEEKDAYZEOM�BM�BQS�QS�BQ�BA�AS�BAS�MSr r r r r! r" r# r$ r% r r r& r �Y�BY�YS�BYSr) r* r( r1 r, r0 r+ r2 r- �-zW-)�
offset_str�returnc C s t j| d�S )z4
Alias to closest period strings BQ->Q etc.
N)�_offset_to_period_map�get)r4 � r8 �</tmp/pip-build-5_djhm0z/pandas/pandas/tseries/frequencies.py�get_period_aliasZ s r: )�namer5 c C s t jdtdd� t| �S )z�
Return DateOffset object associated with rule name.
.. deprecated:: 1.0.0
Examples
--------
get_offset('EOM') --> BMonthEnd(1)
zWget_offset is deprecated and will be removed in a future version, use to_offset instead� )�
stacklevel)�warnings�warn�
FutureWarningr )r; r8 r8 r9 �
get_offseta s
rA T)r? r5 c C s� ddl }t| t�rF| j}t|�p0t|�p0|jtksBtd| j� ���|} t | d�sRn2t
| j�rftd��n t| j�r�t | |d�}|j � S t| |j
�r�t| |j� r�t| |j|jf�r�tdt| �� ���| j} t| |j�s�|j| �} t| |d�}|j � S )a
Infer the most likely frequency given the input index. If the frequency is
uncertain, a warning will be printed.
Parameters
----------
index : DatetimeIndex or TimedeltaIndex
If passed a Series will use the values of the series (NOT THE INDEX).
warn : bool, default True
Returns
-------
str or None
None if no discernible frequency.
Raises
------
TypeError
If the index is not datetime-like.
ValueError
If there are fewer than three values.
r Nz>cannot infer freq from a non-convertible dtype on a Series of �dtypezJPeriodIndex given. Check the `freq` attribute instead of using infer_freq.)r? z4cannot infer freq from a non-convertible index type )Zpandas�
isinstancer Z_valuesr r rB �object� TypeError�hasattrr �_TimedeltaFrequencyInferer�get_freq�IndexZ
DatetimeIndexZ
Int64IndexZ Float64Index�type�_FrequencyInferer)�indexr? �pd�valuesZinfererr8 r8 r9 �
infer_freqx s4
rO c @ s& e Zd ZdZd,ed�dd�Zedd� �Zedd � �Zeed
�d d ��Z eed
�d
d��Z
e e d
�dd�Z
edd� �Zedd� �Zedd� �Zedd� �Zdd� Zedd � �Zed d � �Ze e d
�d d �Ze e d
�d!d"�Ze e d
�d#d$�Ze e d
�d%d&�Zed
�d'd(�Ze e d
�d)d*�Zd+S )-rK z8
Not sure if I can avoid the state machine here
T)r? c C sd || _ |j| _t|d�r4|jd k r4tj| j|j�| _|| _t|�dk rNt d��| j j
p\| j j | _ d S )N�tz� z(Need at least 3 dates to infer frequency)
rL �asi8�i8valuesrF rP r Ztz_convert_from_utcr? �len�
ValueErrorZ_is_monotonic_increasingZ_is_monotonic_decreasing� is_monotonic)�selfrL r? r8 r8 r9 �__init__� s
z_FrequencyInferer.__init__c C s
t | j�S )N)r rS )rW r8 r8 r9 �deltas� s z_FrequencyInferer.deltasc C s t | jj�S )N)r rL rR )rW r8 r8 r9 � deltas_asi8� s z _FrequencyInferer.deltas_asi8)r5 c C s t | j�dkS )N� )rT rY )rW r8 r8 r9 � is_unique� s z_FrequencyInferer.is_uniquec C s t | j�dkS )Nr[ )rT rZ )rW r8 r8 r9 �is_unique_asi8� s z _FrequencyInferer.is_unique_asi8c C s� | j s| jj rdS | jd }t|t�r2| j� S | jddgddgdddgfkrTdS | js^dS | j d }t|t
�r�t d|t
�S t|t �r�t d|t �S t|t
�r�t d |t
�S t|t�r�t d
|t �S t|t�r�t d |t �S t d |�S dS )
z�
Find the appropriate frequency string to describe the inferred
frequency of self.i8values
Returns
-------
str or None
Nr r[ � �A ZBHr% r r! r" r# r$ )rV rL Z
_is_uniquerY � _is_multiple�_ONE_DAY�_infer_daily_rule� hour_deltasr] rZ � _ONE_HOUR�_maybe_add_count� _ONE_MINUTE� _ONE_SECOND�
_ONE_MILLI�
_ONE_MICRO)rW �deltar8 r8 r9 rH � s*
z_FrequencyInferer.get_freqc C s dd� | j D �S )Nc S s g | ] }|t �qS r8 )ra )�.0�xr8 r8 r9 �
<listcomp> s z0_FrequencyInferer.day_deltas.<locals>.<listcomp>)rY )rW r8 r8 r9 �
day_deltas s z _FrequencyInferer.day_deltasc C s dd� | j D �S )Nc S s g | ] }|t �qS r8 )rd )rk rl r8 r8 r9 rm s z1_FrequencyInferer.hour_deltas.<locals>.<listcomp>)rY )rW r8 r8 r9 rc s z _FrequencyInferer.hour_deltasc C s
t | j�S )N)r rS )rW r8 r8 r9 �fields s z_FrequencyInferer.fieldsc C s t | jd �S )Nr )r rS )rW r8 r8 r9 � rep_stamp s z_FrequencyInferer.rep_stampc C s t | j| jj�S )N)r ro rL Z dayofweek)rW r8 r8 r9 r # s z&_FrequencyInferer.month_position_checkc C s& | j d d | j d }t|jd��S )Nr/ � r �i8)ro r �astype)rW Znmonthsr8 r8 r9 �mdiffs& s z_FrequencyInferer.mdiffsc C s t | jd jd��S )Nr/ rr )r ro rs )rW r8 r8 r9 �ydiffs+ s z_FrequencyInferer.ydiffsc C s | j � }|r:| jd }t| jj }|� d|� �}t||�S | j� }|r�| jd d }dddd�}t|| jjd }|� d|� �}t||�S | j� }|r�t|| jd �S | j r�| j
d t } | d dkr�t | jj
� }
td |
� �| d �S td
| �S | j� �rd S | j� } | �r| S d S ) Nr r3 rQ rq � �
)r r< r[ � zW-r r )�_get_annual_ruleru r rp �monthre �_get_quarterly_rulert �_get_monthly_ruler\ rY ra r
�weekday�_is_business_daily�
_get_wom_rule) rW Z annual_ruleZnyearsrz �aliasZquarterly_ruleZ nquartersZmod_dictZ monthly_rule�days�dayZwom_ruler8 r8 r9 rb / s8
z#_FrequencyInferer._infer_daily_rulec C sH t | j�dkrd S t t| jd ��dkr,d S | j� }ddddd�j|�S )Nr[ r r, r- r r+ )�cs�bs�ce�be)rT ru r ro r r7 )rW � pos_checkr8 r8 r9 ry U s z"_FrequencyInferer._get_annual_rulec C sD t | j�dkrd S | jd d dks(d S | j� }ddddd�j|�S ) Nr[ r rQ r) r( r r* )r� r� r� r� )rT rt r r7 )rW r� r8 r8 r9 r{ _ s z%_FrequencyInferer._get_quarterly_rulec C s. t | j�dkrd S | j� }ddddd�j|�S )Nr[ r. ZBMSr r' )r� r� r� r� )rT rt r r7 )rW r� r8 r8 r9 r| i s z#_FrequencyInferer._get_monthly_rulec C s~ | j ddgkrdS | jd j� }tj| jj�}tj|t�}tj|tj |� d�}tj
|dk|dk@ |dk|dk@ |dk@ B �S )Nr[ rQ Fr rx � ) rn rL r} �npZdiffrR Z floor_dividera �modZcumsum�all)rW Z
first_weekdayZshifts�weekdaysr8 r8 r9 r~ o s z$_FrequencyInferer._is_business_dailyc C s~ t | jj�}t|�dkr d S t | jjd d �}||dk }t|�dksTt|�dkrXd S |d d }t|d }d|� |� �S )Nr[ rx r� r zWOM-)r rL r} rT r� r
)rW r� Zweek_of_months�week�wdr8 r8 r9 r ~ s z _FrequencyInferer._get_wom_ruleN)T)�__name__�
__module__� __qualname__�__doc__�boolrX r rY rZ r\ r] r �strrH rn rc ro rp r rt ru rb ry r{ r| r~ r r8 r8 r8 r9 rK � s, - &
rK c @ s e Zd Zdd� ZdS )rG c C sT | j rP| jd t }|d dkrFt| jj� }d|� �}t||d �S td|�S d S )Nr rx zW-r )r\ rY ra r
rp r} re )rW r� r� r� r8 r8 r9 rb � s
z,_TimedeltaFrequencyInferer._infer_daily_ruleN)r� r� r� rb r8 r8 r8 r9 rG � s rG )�multr5 c C s | | dkS )Nr r8 )�usr� r8 r8 r9 r` � s r` )�base�countr5 c C s4 |dkr,|t |�kst�t |�}|� | � �S | S d S )Nr[ )�int�AssertionError)r� r� r8 r8 r9 re � s
re )r5 c C s: |dks| dkrdS t | �} t |�}t|�rNt| �rFtt| �t|��S | d
kS t|�r^| dkS t|�rn| dkS t|�r�| |dddddd d
d d h
kS |dkr�| dkS |dk� r�| dkS |dk� r�| dkS |dk� r�| dkS |dk� r�| dkS |d k� r�| dkS |d
k�r| dkS |d k�r | dkS |d k�r2| dkS dS dS )a
Returns True if downsampling is possible between source and target
frequencies
Parameters
----------
source : str or DateOffset
Frequency converting from
target : str or DateOffset
Frequency converting to
Returns
-------
bool
NFr r r r r% r r! r" r# r$ >
r r r% r r$ r! r r# r" r >
r r r% r r$ r! r r# r" r > r r r% r r$ r! r r# r" > r r r% r$ r! r# r" > r r% r$ r! r r# r" > r r% r r$ r! r# r" > r r% r$ r! r# r" > r r$ r! r# r" > r# r" r$ r! > r# r" r$ > r# r$ > r$ )�_maybe_coerce_freq�
_is_annual�
_is_quarterly�_quarter_months_conformr � _is_monthly�
_is_weekly)�source�targetr8 r8 r9 � is_subperiod� sD
r� c C s\ |dks| dkrdS t | �} t |�}t| �rnt|�rDt| �t|�kS t|�rft| �}t|�}t||�S |d
kS t| �r~|dkS t| �r�|dkS t| �r�|| dddddd d
d d h
kS | dk� r�|dkS | dk� r�|dkS | dk� r�|dkS | dk� r�|dkS | dk�r |dkS | d k�r |dkS | d
k�r0|dkS | d k�rB|dkS | d k�rT|dkS dS dS )a
Returns True if upsampling is possible between source and target
frequencies
Parameters
----------
source : str or DateOffset
Frequency converting from
target : str or DateOffset
Frequency converting to
Returns
-------
bool
NFr r r r r% r r! r" r# r$ >
r r r% r r$ r! r r# r" r >
r r r% r r$ r! r r# r" r > r r r% r r$ r! r r# r" > r r r% r r$ r! r r# r" > r r r% r r$ r! r r# r" > r r r% r r$ r! r r# r" > r r% r$ r! r# r" > r r$ r! r# r" > r# r" r$ r! > r# r" r$ > r# r$ > r$ )r� r� r r� r� r� r� )r� r� ZsmonthZtmonthr8 r8 r9 �is_superperiod� sJ
r� c C s$ | dk s t �t| t�r | j} | j� S )z� we might need to coerce a code to a rule_code
and uppercase it
Parameters
----------
source : string or DateOffset
Frequency converting from
Returns
-------
str
N)r� rC r
Z rule_code�upper)�coder8 r8 r9 r� & s
r� )r� r� r5 c C s t | }t | }|d |d kS )NrQ )r )r� r� ZsnumZtnumr8 r8 r9 r� 9 s r� )�ruler5 c C s | j � } | dkp| jd�S )Nr zA-)r� �
startswith)r� r8 r8 r9 r� ? s r� c C s$ | j � } | dkp"| jd�p"| jd�S )Nr zQ-r* )r� r� )r� r8 r8 r9 r� D s r� c C s | j � } | dkp| dkS )Nr r' )r� )r� r8 r8 r9 r� I s r� c C s | j � } | dkp| jd�S )Nr& zW-)r� r� )r� r8 r8 r9 r� N s r� )r r )T)DZtypingr r> Znumpyr� Zpandas._libs.algosr Zpandas._libs.tslibsr r Z pandas._libs.tslibs.ccalendarr r r r r
Zpandas._libs.tslibs.fieldsr r Zpandas._libs.tslibs.offsetsr
r r r Zpandas._libs.tslibs.parsingr Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr r r Zpandas.core.dtypes.genericr Zpandas.core.algorithmsr ri rh rg rf rd ra r6 Z _need_suffix�_prefix�_m�keyZ_aliasZ_dr� r: rA r� rO rK rG r� r` �floatre r� r� r� r� r� r� r� r� r8 r8 r8 r9 �<module> s�
B ]
8: |