U
|
®ý°dG ã @ s d dl mZ d dlZd dlmZ d dlmZmZm Z m
|
Z
|
d dl m Z m Z mZmZmZ d dlmZmZ d dlmZmZmZ d dlmZ d d lmZ d d
|
l m Z d d l m Z m Z m!Z!m"Z" d d l#m$Z$m%Z% d d l&m'Z' ddddddddddddddddddddddddddddZ(d d d d d d!d"d#d$g Z)e)D ],Z*eD ] Z+e* d%e+ Z,e(e* e(e,< q.q&d&D ](Z*eD ] Z+e* d%e+ Z-e-e(e-< q`qXe D ]Z.d'e. e(d'e. < qd(d)d*d+d,Z/d)d-d.d/Z0G d0d1 d1Z1G d2d3 d3e1Z2d4d5d6d7d8Z3d(d9d(d:d;d<Z4d5d-d=d>Z5d5d-d?d@Z6d(d-dAdBZ7d(d(d5dCdDdEZ8d(d5dFdGdHZ9d(d5dFdIdJZ:d(d5dFdKdLZ;d(d5dFdMdNZ<dOd,d/d>d@dPgZ=dS )Qé )Ú annotationsN)Ú unique_deltas)Ú TimestampÚget_unit_from_dtypeÚperiods_per_dayÚtz_convert_from_utc)ÚDAYSÚ MONTH_ALIASESÚ MONTH_NUMBERSÚMONTHSÚint_to_weekday)Úbuild_field_sarrayÚmonth_position_check)Ú
|
DateOffsetÚDayÚ to_offset)Úget_rule_month)Únpt)Úcache_readonly)Úis_datetime64_dtypeÚis_numeric_dtypeÚis_period_dtypeÚis_timedelta64_dtype)ÚABCIndexÚ ABCSeries)ÚuniqueÚ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* r3 r. r2 r- r4 r/ ú-)r r úW-Ústrú
|
str | None)Ú
|
offset_strÚreturnc C s t | d¡S )z4
|
Alias to closest period strings BQ->Q etc.
|
N)Ú_offset_to_period_mapÚget)r9 © r= úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/tseries/frequencies.pyÚget_period_alias^ s r? ©r: c C sÜ ddl m}m} t| trN| j}t|sJt|sJ|jt ksJt
|
d| j |} t | dsZn.t | jrnt
|
dnt| jrt | }| ¡ S t| |rºt| |sºt| r´t
|
d| j | j} t| |sÌ|| } t| }| ¡ S )aC
|
Infer the most likely frequency given the input index.
|
|
Parameters
|
----------
|
index : DatetimeIndex or TimedeltaIndex
|
If passed a Series will use the values of the series (NOT THE INDEX).
|
|
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.
|
|
Examples
|
--------
|
>>> idx = pd.date_range(start='2020/12/01', end='2020/12/30', periods=30)
|
>>> pd.infer_freq(idx)
|
'D'
|
r )Ú DatetimeIndexÚIndexz>cannot infer freq from a non-convertible dtype on a Series of ÚdtypezJPeriodIndex given. Check the `freq` attribute instead of using infer_freq.z8cannot infer freq from a non-convertible index of dtype )Zpandas.core.apirA rB Ú
|
isinstancer Z_valuesr r rC ÚobjectÚ TypeErrorÚhasattrr Ú_TimedeltaFrequencyInfererÚget_freqr Ú_FrequencyInferer)ÚindexrA rB ÚvaluesZinfererr= r= r> Ú
|
infer_freqi s@
|
ÿþý
|
ÿ
|
|
ÿ
|
|
ÿ
|
rM c @ sP e Zd ZdZddddZeddddZeddd d
|
Zed dd d Zed dddZ ddddZ
|
eddddZ eddddZ eddddZ eddd d Zddd d Zeddd d!Zeddd"d#Zddd$d%Zddd&d'Zddd(d)Zddd*d+Zddd,d-Zd dd.d/Zddd0d1Zd2S )3rJ z8
|
Not sure if I can avoid the state machine here
|
ÚNoner@ c C s || _ |j| _t|tr*t|jjj| _ nt|jj| _ t
|
|drb|j d k rbt | j|j | j d| _t |dk rvtd| j jp| j j| _d S )NÚtz©Zresoé z(Need at least 3 dates to infer frequency)rK Úasi8Úi8valuesrD r r Ú_dataZ_ndarrayrC Ú_cresorG rO r ÚlenÚ
|
ValueErrorZ_is_monotonic_increasingZ_is_monotonic_decreasingÚ is_monotonic)ÚselfrK r= r= r> Ú__init__· s$
|
ÿ
|
|
ÿ ÿz_FrequencyInferer.__init__znpt.NDArray[np.int64]c C s
|
t | jS ©N)r rS ©rY r= r= r> ÚdeltasÖ s z_FrequencyInferer.deltasc C s t | jjS r[ )r rK rR r\ r= r= r> Ú deltas_asi8Ú s z _FrequencyInferer.deltas_asi8Úboolc C s t | jdkS ©Né )rV r] r\ r= r= r> Ú is_uniqueà s z_FrequencyInferer.is_uniquec C s t | jdkS r` )rV r^ r\ r= r= r> Úis_unique_asi8ä s z _FrequencyInferer.is_unique_asi8r8 c C s" | j r| jjsdS | jd }t| j}|r<t||r<| ¡ S | jddgddgdddgfkr^dS | j shdS | j
|
d }|d }|d }|d }t||r¢t d || S t||rºt d
|
|| S t||rÒt d || S t||d ròt d ||d S t||d rt d||d S t d|S dS )z¨
|
Find the appropriate frequency string to describe the inferred
|
frequency of self.i8values
|
|
Returns
|
-------
|
str or None
|
Nr ra é éA ZBHé é< r' r" r# iè r$ i@B r% r& ) rX rK Z
|
_is_uniquer] r rU Ú _is_multipleÚ_infer_daily_ruleÚ hour_deltasrc r^ Ú_maybe_add_count)rY ÚdeltaÚppdÚpphÚppmZppsr= r= r> rI è s2
|
|
|
|
|
|
z_FrequencyInferer.get_freqz list[int]c s t | j fdd| jD S )Nc s g | ] }| qS r= r= ©Ú.0Úx©rm r= r> Ú
|
<listcomp> s z0_FrequencyInferer.day_deltas.<locals>.<listcomp>©r rU r] r\ r= rs r> Ú
|
day_deltas s
|
z _FrequencyInferer.day_deltasc s" t | jd fdd| jD S )Nrf c s g | ] }| qS r= r= rp ©rn r= r> rt " s z1_FrequencyInferer.hour_deltas.<locals>.<listcomp>ru r\ r= rw r> rj s z _FrequencyInferer.hour_deltasz
|
np.ndarrayc C s t | j| jdS )NrP )r rS rU r\ r= r= r> Úfields$ s z_FrequencyInferer.fieldsr c C s t | jd S ©Nr )r rS r\ r= r= r> Ú rep_stamp( s z_FrequencyInferer.rep_stampc C s t | j| jjS r[ )r rx rK Z dayofweekr\ r= r= r> r , s z&_FrequencyInferer.month_position_checkc C s& | j d d | j d }t| d¡S )Nr1 é r Úi8)rx r Úastype)rY Znmonthsr= r= r> Úmdiffs/ s z_FrequencyInferer.mdiffsc C s t | jd d¡S )Nr1 r| )r rx r} r\ r= r= r> Úydiffs4 s z_FrequencyInferer.ydiffsc
|
C sÖ | ¡ }|r:| jd }t| jj }| d| }t||S | ¡ }|r| jd d }dddd}t|| jjd }| d| }t||S | ¡ }|r¨t|| jd S | j r¶|
|
¡ S | ¡ rÂdS | ¡ } | rÒ| S d S ) Nr r5 rQ r{ é é
|
)r é ra r! ) Ú_get_annual_ruler r rz Úmonthrk Ú_get_quarterly_ruler~ Ú_get_monthly_rulerb Ú_get_daily_ruleÚ_is_business_dailyÚ _get_wom_rule)
|
rY Z annual_ruleZnyearsr ÚaliasZquarterly_ruleZ nquartersÚmod_dictZ monthly_ruleZwom_ruler= r= r> ri 8 s0
|
|
|
z#_FrequencyInferer._infer_daily_rulec C sX t | j}| jd | }|d dkrJt| j ¡ }d| }t||d S td|S d S )Nr é r6 r )r rU r] r rz Úweekdayrk )rY rm ÚdaysÚwdr r= r= r> r X s
|
|
z!_FrequencyInferer._get_daily_rulec C sX t | jdkrd S t t| jd dkr,d S | ¡ }|d kr@d S ddddd |¡S d S )Nra r r. r/ r r- ©ÚcsÚbsZceÚbe)rV r r rx r r< ©rY Z pos_checkr= r= r> r c s z"_FrequencyInferer._get_annual_rulec C sT t | jdkrd S | jd d dks(d S | ¡ }|d kr<d S ddddd |¡S d S ) Nra r rQ r+ r* r r, r ©rV r~ r r< r r= r= r> r
q s z%_FrequencyInferer._get_quarterly_rulec C s> t | jdkrd S | ¡ }|d kr&d S ddddd |¡S d S )Nra r0 ZBMSr r) r r r r= r= r> r s z#_FrequencyInferer._get_monthly_rulec C s | j ddgkrdS | jd ¡ }t | j¡}t| j}t ||¡}t |t
|
|¡ d¡}t t |dk|dk@ |dk|dk@ |dk@ B ¡S )Nra rQ Fr r é ) rv rK r ÚnpZdiffrS r rU Z floor_divideÚmodZcumsumr_ Úall)rY Z first_weekdayZshiftsrm Úweekdaysr= r= r> r 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 )Nra r r r zWOM-)r rK r rV Údayr )rY r Zweek_of_monthsÚweekr r= r= r> r s z _FrequencyInferer._get_wom_ruleN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__rZ r r] r^ rb rc rI rv rj rx rz r r~ r ri r r r
r r r r= r= r= r> rJ ² s> 2
|
rJ c @ s e Zd Zdd ZdS )rH c C s | j r| ¡ S d S r[ )rb r r\ r= r= r> ri ¯ s z,_TimedeltaFrequencyInferer._infer_daily_ruleN)r r r ri r= r= r= r> rH ® s rH Úintr_ )Úmultr: c C s | | dkS ry r= )Úusr¢ r= r= r> rh ´ s rh Úfloat)ÚbaseÚcountr: c C s4 |dkr,|t |kstt |}| | S | S d S r` )r¡ ÚAssertionError)r¥ r¦ r= r= r> rk ¸ s
|
rk c C s2 |dks| dkrdS t | } t |}t|rNt| rFtt| t|S | dkS t|r^| dkS t|rn| dkS t|r| |ddddd d
|
d d d h
|
kS |dkr¢| dkS |dkr²| dkS |dkrÂ| dkS |dkrÒ| dkS |d krâ| dkS |d
|
k rô| dkS |d kr| dkS |d kr| dkS |d kr*| 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
|
NF¾
|
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Útargetr= r= r> Ú is_subperiodÅ sH ÿ
|
|
|
|
r¸ c C sV |dks| dkrdS t | } t |}t| rnt|rDt| t|kS t|rft| }t|}t||S |dkS t| r~|dkS t| r|dkS t| r²|| ddddd d
|
d d d h
|
kS | dkrÂ|dkS | dkrÒ|dkS | dkrâ|dkS | dk rô|dkS | d kr|dkS | d
|
kr|dkS | d kr*|dkS | d kr<|dkS | d krN|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· ZsmonthZtmonthr= r= r> Úis_superperiodý sJ
|
|
|
|
|
|
|
r¹ c C s$ | dk s t t| tr | j} | ¡ S )zÈwe might need to coerce a code to a rule_code
|
and uppercase it
|
|
Parameters
|
----------
|
source : str or DateOffset
|
Frequency converting from
|
|
Returns
|
-------
|
str
|
N)r§ rD r Z rule_codeÚupper)Úcoder= r= r> r° 7 s
|
r° )r¶ r· r: c C s t | }t | }|d |d kS )NrQ )r
|
)r¶ r· ZsnumZtnumr= r= r> r³ J s r³ )Úruler: c C s | ¡ } | dkp| d¡S )Nr zA-©rº Ú
|
startswith©r¼ r= r= r> r± P s r± c C s$ | ¡ } | dkp"| d¡p"| d¡S )Nr zQ-r, r½ r¿ r= r= r> r² U s r² c C s | ¡ } | dkS )N)r r) )rº r¿ r= r= r> r´ Z s r´ c C s | ¡ } | dkp| d¡S )Nr( r6 r½ r¿ r= r= r> rµ _ s rµ r r )>Ú
|
__future__r Únumpyr Zpandas._libs.algosr Zpandas._libs.tslibsr r r r Z pandas._libs.tslibs.ccalendarr r r
|
r r Zpandas._libs.tslibs.fieldsr r Zpandas._libs.tslibs.offsetsr r r Zpandas._libs.tslibs.parsingr Zpandas._typingr Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr r r r Zpandas.core.dtypes.genericr r Zpandas.core.algorithmsr r; Z _need_suffixÚ_prefixÚ_mÚkeyÚ_aliasZ_dr? rM rJ rH rh rk r¸ r¹ r° r³ r± r² r´ rµ Ú__all__r= r= r= r> Ú<module> s å I } 8:ú
|
