U
|
¬ý°dÞ ã @ s d Z ddlmZ ddlZddlmZ ddlmZm Z m
|
Z
|
m Z m Z ddl mZ dddd d
|
d d d ddZddddd ddddZdddd ddddZdS )zj
|
Helper functions to generate range-like data for DatetimeArray
|
(and possibly TimedeltaArray/PeriodArray)
|
é )Ú annotationsN)Úi8max)Ú
|
BaseOffsetÚOutOfBoundsDatetimeÚ TimedeltaÚ TimestampÚiNaT)ÚnptÚnsz Timestamp | Timedelta | Nonez
|
int | Noner Ústrznpt.NDArray[np.intp])ÚstartÚendÚperiodsÚfreqÚunitÚreturnc
|
C s² | dk r| j nd}|dk r |j nd}|j t|}z|j|dd}W n: tk
|
r~ } z td| d| d|W 5 d}~X Y nX t|j } |dkrÈ|dk rÈ|dk rÈ|}
|
|
|
||
|
| | | d d } n\|dk rî|dk rî|}
|
t|
|
|| d d
|
} n6|dk r |dk r || } t| || d d
|
}
|
ntd tjd dx ztj |
|
| | tj
|
d} W nX t k
|
r¢ |
|
g} | d | kr| | d | ¡ qbtj | dd
tj
|
d} Y nX W 5 Q R X | S )a
|
Generate a range of dates or timestamps with the spans between dates
|
described by the given `freq` DateOffset.
|
|
Parameters
|
----------
|
start : Timedelta, Timestamp or None
|
First point of produced date range.
|
end : Timedelta, Timestamp or None
|
Last point of produced date range.
|
periods : int or None
|
Number of periods in produced date range.
|
freq : Tick
|
Describes space between dates in produced date range.
|
unit : str, default "ns"
|
The resolution the output is meant to represent.
|
|
Returns
|
-------
|
ndarray[np.int64]
|
Representing the given resolution.
|
NF)Zround_okzfreq=z is incompatible with unit=z,. Use a lower freq or a higher unit instead.é é r )Úsider zEat least 'start' or 'end' should be specified if a 'period' is given.Úraise©Zover)Zdtypeéÿÿÿÿ)Ú_valueZnanosr Zas_unitÚ
|
ValueErrorÚintÚ _generate_range_overflow_safeÚnpÚerrstateZarangeÚint64ÚFloatingPointErrorÚappendÚarray)r r r r r ÚistartZiendÚtdÚerrÚstrideÚbÚeÚvaluesZxdr© r) úQd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/arrays/_ranges.pyÚgenerate_regular_range sJ ÿ
|
ÿý
|
"ÿ(r+ r r )Úendpointr r% r r c C s |dks t t t¡}d| d| d| }tjddR z t |¡t t |¡¡ }W n, tk
|
r } zt||W 5 d}~X Y nX W 5 Q R X t |¡|kr¬t| |||S | dkrÄ|d krÄ|dksä| d k rØ|k rîn n|d
|
krît|n@|d
|
kr.| | | kr| k r.n nt | | |d ||S |d }|| } d| k rV|k shn t | || |ft | |||}
|
t |
|
| ||S ) a[
|
Calculate the second endpoint for passing to np.arange, checking
|
to avoid an integer overflow. Catch OverflowError and re-raise
|
as OutOfBoundsDatetime.
|
|
Parameters
|
----------
|
endpoint : int
|
nanosecond timestamp of the known endpoint of the desired range
|
periods : int
|
number of periods in the desired range
|
stride : int
|
nanoseconds between periods in the desired range
|
side : {'start', 'end'}
|
which end of the range `endpoint` refers to
|
|
Returns
|
-------
|
other_end : int
|
|
Raises
|
------
|
OutOfBoundsDatetime
|
©r r úCannot generate range with ú=ú and periods=r r Nr r r r r )
|
ÚAssertionErrorr Úuint64r r Úabsr r Ú$_generate_range_overflow_safe_signedr ) r, r r% r Úi64maxÚmsgÚaddendr$ Z mid_periodsÚ remainingZmidpointr) r) r* r ] s>
|
& ÿÿ
|
ÿ
|
* ÿ*r c
|
C s( |dks t |dkr |d9 }tjddÜ t |¡t |¡ }z,t | ¡| }|tkrZt|W W 5 Q R £ S ttfk
|
r Y nX |dkr| dks¨|dk r¤| dks¨t |dkr t | ¡t |¡ }t t¡}||ksÜt ||t |¡ kr |W 5 Q R £ S W 5 Q R X t d| d| d | d
|
S ) z
|
A special case for _generate_range_overflow_safe where `periods * stride`
|
can be calculated without overflowing int64 bounds.
|
r- r r r r r r. r/ r0 N)
|
r1 r r r r Ú OverflowErrorr r2 r r )r, r r% r r7 Úresultr5 r) r) r* r4 s, $
|
|
ÿr4 )r
|
)r )Ú__doc__Ú
|
__future__r Únumpyr Zpandas._libs.libr Zpandas._libs.tslibsr r r r r Zpandas._typingr r+ r r4 r) r) r) r* Ú<module> s ûJ ÿC
|
