3
���h�( � @ s0 d Z ddlmZmZ ddlmZmZ ddlm Z m
Z
ddl m Z ddl
mZ ddlmZmZmZmZmZmZmZ ddlmZ dd lmZ dd
l j j Z dd l m!Z!m"Z" dd l#m$Z$m%Z% dd
l&m'Z' e'dddddddgej( edd�e'dddddd d d d d d!d"g e�G d#d$� d$e%���Z)e)j*� d(e)d%�d&d'�Z+d
S ))z implement the TimedeltaIndex � )�index�lib)� Timedelta� to_offset)�DtypeObj�Label)�InvalidIndexError)�doc)� TD64NS_DTYPE�is_float�
is_integer� is_scalar�is_timedelta64_dtype�is_timedelta64_ns_dtype� pandas_dtype)� datetimelike)�TimedeltaArrayN)�Index�maybe_extract_name)�DatetimeIndexOpsMixin�DatetimeTimedeltaMixin)�
inherit_names�__neg__�__pos__�__abs__�
total_seconds�round�floor�ceilT)�wrapZ _bool_opsZ _object_ops�
_field_opsZ_datetimelike_opsZ_datetimelike_methodsZ
_other_ops�
componentsZto_pytimedelta�sumZstdZmedianZ_format_native_typesc @ s� e Zd ZU dZdZejZddgZddgZ dZ
e dde
jdeddfdd �Zed e ed
�d d ��Zed
d� �Zeej�d dd��Zeed�dd�Zd dd�Z e d�dd�Z ed�dd�Z ee d�d d ��Z dS )!�TimedeltaIndexa�
Immutable ndarray of timedelta64 data, represented internally as int64, and
which can be boxed to timedelta objects.
Parameters
----------
data : array-like (1-dimensional), optional
Optional timedelta-like data to construct index with.
unit : unit of the arg (D,h,m,s,ms,us,ns) denote the unit, optional
Which is an integer/float number.
freq : str or pandas offset object, optional
One of pandas date offset strings or corresponding objects. The string
'infer' can be passed in order to set the frequency of the index as the
inferred frequency upon creation.
copy : bool
Make a copy of input ndarray.
name : object
Name to be stored in the index.
Attributes
----------
days
seconds
microseconds
nanoseconds
components
inferred_freq
Methods
-------
to_pytimedelta
to_series
round
floor
ceil
to_frame
mean
See Also
--------
Index : The base pandas Index type.
Timedelta : Represents a duration between two dates or times.
DatetimeIndex : Index of datetime64 data.
PeriodIndex : Index of Period data.
timedelta_range : Create a fixed-frequency TimedeltaIndex.
Notes
-----
To learn more about the frequency strings, please see `this link
<https://pandas.pydata.org/pandas-docs/stable/user_guide/timeseries.html#offset-aliases>`__.
Ztimedeltaindex�name�freqTNFc C s� t ||| �}t|�r.t| j� dt|�� d���|d kr>td��t|t�rl|tj krl|r^|j
� }| j ||d�S t|t �r�|tj kr�|d kr�|r�|j
� S |j
� S tj|||||d�}| j ||d�S )
Nz2() must be called with a collection of some kind, z was passed�Y�y�MzqUnits 'M', 'Y', and 'y' are no longer supported, as they do not represent unambiguous timedelta values durations.)r$ )r% �unit�dtype�copy> r' r& r( )r r
� TypeError�__name__�repr�
ValueError�
isinstancer r �
no_defaultr+ � _simple_newr# Z
_shallow_copyZ_from_sequence) �cls�datar) r% �closedr* r+ r$ �tdarr� r7 �@/tmp/pip-build-5_djhm0z/pandas/pandas/core/indexes/timedeltas.py�__new__w s$
zTimedeltaIndex.__new__)�valuesr$ c C s> t |t�st�tj| �}||_||_i |_|j|_|j � |S )N)
r0 r �AssertionError�objectr9 �_data�_name�_cacheZ _index_dataZ_reset_identity)r3 r: r$ �resultr7 r7 r8 r2 � s
zTimedeltaIndex._simple_newc C s ddl m} || dd�S )Nr )�_get_format_timedelta64T)Zbox)Zpandas.io.formats.formatrA )�selfrA r7 r7 r8 �_formatter_func� s z TimedeltaIndex._formatter_funcc C sb t |�}t|�rRt|� rR| jj||d�}| jr>t|| jd�S t|jd�| jd�S tj| ||d�S )N)r+ )r$ �i8) r r r r= �astypeZhasnansr r$ r )rB r* r+ r@ r7 r7 r8 rE � s zTimedeltaIndex.astype)r* �returnc C s t |�S )zF
Can we compare values of the given dtype to our own?
)r )rB r* r7 r7 r8 �_is_comparable_dtype� s z#TimedeltaIndex._is_comparable_dtypec C sb t |�st|��y| jj|dd�}W n, tk
rP } zt|�|�W Y dd}~X nX tj| |||�S )z�
Get integer location for requested label
Returns
-------
loc : int, slice, or ndarray[int]
T)Zcast_strN)r
r r= Z_validate_scalarr, �KeyErrorr �get_loc)rB �key�methodZ tolerance�errr7 r7 r8 rI � s zTimedeltaIndex.get_loc)�sidec C sp |dks t �t|t�rPt|�}|j|j�}|dkr6|S |t|j� tdd� S n t|�s`t|�rl| j d|� |S ) a
If label is a string, cast it to timedelta according to resolution.
Parameters
----------
label : object
side : {'left', 'right'}
kind : {'loc', 'getitem'} or None
Returns
-------
label : object
�loc�getitemN�left� �ns�slice)rN rO N)
r; r0 �strr r Zresolution_stringr r r Z_invalid_indexer)rB �labelrM �kind�parsedZlboundr7 r7 r8 �_maybe_cast_slice_bound� s
z&TimedeltaIndex._maybe_cast_slice_bound)rF c C s || j kp|dkS )N� timedelta)�
inferred_type)rB �typr7 r7 r8 �is_type_compatible� s z!TimedeltaIndex.is_type_compatiblec C s dS )NZ timedelta64r7 )rB r7 r7 r8 rZ � s z TimedeltaIndex.inferred_type)N)T)NN)!r- �
__module__� __qualname__�__doc__Z_typ�libindexZTimedeltaEngineZ _engine_typeZ _comparables� _attributesZ_is_numeric_dtyper r= r r1 r
r9 � classmethodr r2 �propertyrC r r rE r �boolrG rI rT rX r\ rZ r7 r7 r7 r8 r# s2
J "
r# )rF c C sJ |dkrt j|| |�rd}tj|�\}}tj| ||||d�}tj||d�S )a�
Return a fixed frequency TimedeltaIndex, with day as the default
frequency.
Parameters
----------
start : str or timedelta-like, default None
Left bound for generating timedeltas.
end : str or timedelta-like, default None
Right bound for generating timedeltas.
periods : int, default None
Number of periods to generate.
freq : str or DateOffset, default 'D'
Frequency strings can have multiples, e.g. '5H'.
name : str, default None
Name of the resulting TimedeltaIndex.
closed : str, default None
Make the interval closed with respect to the given frequency to
the 'left', 'right', or both sides (None).
Returns
-------
rng : TimedeltaIndex
Notes
-----
Of the four parameters ``start``, ``end``, ``periods``, and ``freq``,
exactly three must be specified. If ``freq`` is omitted, the resulting
``TimedeltaIndex`` will have ``periods`` linearly spaced elements between
``start`` and ``end`` (closed on both sides).
To learn more about the frequency strings, please see `this link
<https://pandas.pydata.org/pandas-docs/stable/user_guide/timeseries.html#offset-aliases>`__.
Examples
--------
>>> pd.timedelta_range(start='1 day', periods=4)
TimedeltaIndex(['1 days', '2 days', '3 days', '4 days'],
dtype='timedelta64[ns]', freq='D')
The ``closed`` parameter specifies which endpoint is included. The default
behavior is to include both endpoints.
>>> pd.timedelta_range(start='1 day', periods=4, closed='right')
TimedeltaIndex(['2 days', '3 days', '4 days'],
dtype='timedelta64[ns]', freq='D')
The ``freq`` parameter specifies the frequency of the TimedeltaIndex.
Only fixed frequencies can be passed, non-fixed frequencies such as
'M' (month end) will raise.
>>> pd.timedelta_range(start='1 day', end='2 days', freq='6H')
TimedeltaIndex(['1 days 00:00:00', '1 days 06:00:00', '1 days 12:00:00',
'1 days 18:00:00', '2 days 00:00:00'],
dtype='timedelta64[ns]', freq='6H')
Specify ``start``, ``end``, and ``periods``; the frequency is generated
automatically (linearly spaced).
>>> pd.timedelta_range(start='1 day', end='5 days', periods=4)
TimedeltaIndex(['1 days 00:00:00', '2 days 08:00:00', '3 days 16:00:00',
'5 days 00:00:00'],
dtype='timedelta64[ns]', freq=None)
N�D)r5 )r$ )�comZany_none�dtlZmaybe_infer_freqr Z_generate_ranger# r2 )�start�endZperiodsr% r$ r5 �_r6 r7 r7 r8 �timedelta_range s
Crk )NNNNNN),r_ Z pandas._libsr r` r Zpandas._libs.tslibsr r Zpandas._typingr r Z
pandas.errorsr Zpandas.util._decoratorsr Zpandas.core.dtypes.commonr
r r r
r r r Zpandas.core.arraysr rg Z pandas.core.arrays.timedeltasr Zpandas.core.common�core�commonrf Zpandas.core.indexes.baser r Z pandas.core.indexes.datetimeliker r Z pandas.core.indexes.extensionr r r# Z _add_logical_methods_disabledrk r7 r7 r7 r8 �<module> sH $
O |