U f/eL@sddlmZddlmZmZddlmZddlZddlZddl m Z m Z ddl mZmZmZmZmZddlmZmZddlmZmZdd lmZdd lmZdd lmZmZm Z m!Z!dd l"m#Z#dd l$m%Z%ddl&m'Z'm(Z(m)Z)m*Z*ddl+m,m-Z.ddl/m,m0m1Z2ddl/m3Z3ddl4m5Z5ddl6m7Z7m8Z8ddl9m:Z:ddl;mZ>e>?ddiddiZ@ddZAe:dddge'jBe'dde:dd ge'Gd!d"d"e5ZCd'd#d"d$d%d&ZDdS)() annotations)datetime timedelta)HashableN)indexlib) BaseOffsetNaTPeriod ResolutionTick)DateParseErrorparse_time_string)DtypeDtypeObj)InvalidIndexError)doc)is_datetime64_any_dtype is_integer is_scalar pandas_dtype) PeriodDtype)is_valid_na_for_dtype) PeriodArray period_arrayraise_on_incompatiblevalidate_dtype_freq)maybe_extract_name)DatetimeIndexOpsMixin) DatetimeIndexIndex) inherit_names) Int64IndexZ target_klasszPeriodIndex or list of PeriodsklassrcKsJ|d}|jdkr:|dd}t||d}|j|f|S||f|SdS)Ndataint64freqr&)popdtyper _simple_new)clsdvaluesr&r.>/tmp/pip-unpacked-wheel-tiezk1ph/pandas/core/indexes/period.py_new_PeriodIndexEs     r0strftime start_timeZend_timeT)wrapZ is_leap_yearZ_format_native_typescseZdZUdZdZdgZded<ded<eZe j Z dZ e ejfd dd edJd ddddZe ejdKddddZee ejjddddZee ejjddddZee ejjddddZdLddd dd!d"d#Zed$dd%d&Zd'd(Zd)dd*d+d,Zd-d$d$d.fd/d0 Ze ejdejfdd1fd2d3 Zeddd4d5Zed dd6d7Z fd8d9Z!dMd:d;Z"ejfd d<d=d>Z#d?d@dAdBdCZ$d?dDdEdFZ%d dGdHdIZ&Z'S)N PeriodIndexa Immutable ndarray holding ordinal values indicating regular periods in time. Index keys are boxed to Period objects which carries the metadata (eg, frequency information). Parameters ---------- data : array-like (1d int np.ndarray or PeriodArray), optional Optional period-like data to construct index with. copy : bool Make a copy of input ndarray. freq : str or period object, optional One of pandas period strings or corresponding objects. year : int, array, or Series, default None month : int, array, or Series, default None quarter : int, array, or Series, default None day : int, array, or Series, default None hour : int, array, or Series, default None minute : int, array, or Series, default None second : int, array, or Series, default None dtype : str or PeriodDtype, default None Attributes ---------- day dayofweek day_of_week dayofyear day_of_year days_in_month daysinmonth end_time freq freqstr hour is_leap_year minute month quarter qyear second start_time week weekday weekofyear year Methods ------- asfreq strftime to_timestamp See Also -------- Index : The base pandas Index type. Period : Represents a period of time. DatetimeIndex : Index with datetime64 data. TimedeltaIndex : Index of timedelta64 data. period_range : Create a fixed-frequency PeriodIndex. Examples -------- >>> idx = pd.PeriodIndex(year=[2000, 2002], quarter=[1, 3]) >>> idx PeriodIndex(['2000Q1', '2002Q3'], dtype='period[Q-DEC]') Z periodindexnamer_datarr&Tzpandas.arrays.PeriodArray)otherZ other_nameNEstr)howreturncCs"|j||}t|j||jdSNr5)r6asfreqtyper*r5selfr&r:Zarrr.r.r/r>szPeriodIndex.asfreqstartr)r;cCs|j||}tj||jdSr<)r6 to_timestamprr*r5r@r.r.r/rCszPeriodIndex.to_timestampr"cCst|jj|jdSr<)r"r6hourr5rAr.r.r/rDszPeriodIndex.hourcCst|jj|jdSr<)r"r6minuter5rEr.r.r/rFszPeriodIndex.minutecCst|jj|jdSr<)r"r6secondr5rEr.r.r/rGszPeriodIndex.secondFz Dtype | Noneboolr)r)copyr5r;c Ksdddddddh}t||sBtt||d} td | t|||}|dkr|dkrtddd||\}} | }t||d }nft||}|rt||r|j |kr| |}|dkr|dk rt j |t j d }t||d }n t||d }|r|}|j||d S)NyearmonthdayZquarterrDrFrGrz-__new__() got an unexpected keyword argument r'r))r$r&r=)setissubsetlist TypeErrorrr_generate_ranger isinstancer&r>npasarrayr%rrIr*) r+r$Zordinalr&r)rIr5fieldsZvalid_field_setargumentZfreq2r.r.r/__new__s4      zPeriodIndex.__new__z np.ndarraycCstj|tdS)NrM)rTrUobjectrEr.r.r/r-szPeriodIndex.valuescCsxt|ttjttjfr4t|jtrj|j|}|Sn6t|t r^|j |jj krR|j St ||n t |rj|St |ddS)a Convert timedelta-like input to an integer multiple of self.freq Parameters ---------- other : timedelta, np.timedelta64, DateOffset, int, np.ndarray Returns ------- converted : int, np.ndarray[int64] Raises ------ IncompatibleFrequency : if the input cannot be written as a multiple of self.freq. Note IncompatibleFrequency subclasses ValueError. N)rSrrTZ timedelta64r Zndarrayr&r6Z _check_timedeltalike_freq_compatrbasenrr)rAr7deltar.r.r/_maybe_convert_timedeltas    z$PeriodIndex._maybe_convert_timedeltar)r)r;cCst|tsdS|j|jkS)zF Can we compare values of the given dtype to our own? F)rSrr&)rAr)r.r.r/_is_comparable_dtype7s z PeriodIndex._is_comparable_dtyper )wheremaskr;cs<t|trt|j|jd}nt|ts.tdt||S)z where : array of timestamps mask : np.ndarray[bool] Array of booleans where data is not NA. r'z6asof_locs `where` must be DatetimeIndex or PeriodIndex)rSrr4Z_valuesr&rQsuper asof_locs)rAr_r` __class__r.r/rbBs   zPeriodIndex.asof_locsrIcs^t|}|tjk r$tjdtddnd}t|rNt|dd}|j|d |St j ||dS)NzThe 'how' keyword in PeriodIndex.astype is deprecated and will be removed in a future version. Use index.to_timestamp(how=how) instead) stacklevelrBtzr:re) rr no_defaultwarningswarn FutureWarningrgetattrrCZ tz_localizeraastype)rAr)rIr:rhrcr.r/roOs  zPeriodIndex.astypecCsDt|dkrdS|jstd|j}|dd|dddkS)z Returns True if this PeriodIndex is range-like in that all Periods between start and end are present, in order. rTzIndex is not monotonicNrf)lenZis_monotonic_increasing ValueErrorZasi8all)rAr-r.r.r/is_fullfs  zPeriodIndex.is_fullcCsdS)NZperiodr.rEr.r.r/ inferred_typesszPeriodIndex.inferred_typecs(t||}|j|jkr$||}|SN)ra_convert_tolerancer)r])rA tolerancetargetrcr.r/rx|s  zPeriodIndex._convert_tolerancec CsP|}t|st|t||jr(t}nt|tr2z||}|WStt fk r\YnXzt ||j \}}Wn8t t fk r}zt d|d|W5d}~XYnXt|} | jj} |jj} | | kst| | ks| tjkr|jj jdkrt||j d}|j|||d}|S|dkr,t |n|}nt|trp|j } |j } | j| jkrf| j| jkst |nXt|trzt||j d}Wn.t k r}zt ||W5d}~XYnXnt |zt||j d}Wn.t k r}zt ||W5d}~XYnXzt||||WSt k rJ}zt ||W5d}~XYnXdS)a Get integer location for requested label. Parameters ---------- key : Period, NaT, str, or datetime String or datetime key must be parsable as Period. Returns ------- loc : int or ndarray[int64] Raises ------ KeyError Key is not present in the index. TypeError If key is listlike or otherwise not hashable. zCannot interpret 'z ' as periodNBr')methodry)rrrr)r rSr9_get_string_slicerQrsrr&r KeyErrorr from_attrname freq_groupvaluefreq_group_codeAssertionErrorZRESO_DAYr5r get_locr[Z_period_dtype_coderr )rAkeyr|ryZorig_keylocZasdtreso_strerrresogrpfreqnZsfreqZkfreqr.r.r/rsh   $          zPeriodIndex.get_loc)sidec Cs|dddtjfkst||ddt|trr&)rArrrZivr.r.r/rsz$PeriodIndex._parsed_string_to_bounds)rcCs:t|tstt||f|j}|jj}|j|ks6tdSrw) rSr rr?rr)rrrs)rArrrr.r.r/_validate_partial_date_slices  z(PeriodIndex._validate_partial_date_slice)rc CsZt||j\}}t|}z|||WStk rT}zt||W5d}~XYnXdSrw)rr&r rZ_partial_date_slicer~)rArrrrrr.r.r/r}s  zPeriodIndex._get_string_slice)Nr8)NrB)NNNNFN)NN)(__name__ __module__ __qualname____doc__Z_typ _attributes__annotations__rZ _data_clslibindexZ PeriodEngineZ _engine_typeZ!_supports_partial_string_indexingrr>_shared_doc_kwargsrCpropertyrDfgetrFrGrXr-r]r^rbr rorrjrurvrxrrrrr} __classcell__r.r.rcr/r4Psh E   >$    ^( r4z int | None)periodsr;cCsjt|||dkrtd|dkr:t|ts:t|ts:d}tj||||id\}}t||d}t||dS)a Return a fixed frequency PeriodIndex. The day (calendar) is the default frequency. Parameters ---------- start : str or period-like, default None Left bound for generating periods. end : str or period-like, default None Right bound for generating periods. periods : int, default None Number of periods to generate. freq : str or DateOffset, optional Frequency alias. By default the freq is taken from `start` or `end` if those are Period objects. Otherwise, the default is ``"D"`` for daily frequency. name : str, default None Name of the resulting PeriodIndex. Returns ------- PeriodIndex Notes ----- Of the three parameters: ``start``, ``end``, and ``periods``, exactly two must be specified. To learn more about the frequency strings, please see `this link `__. Examples -------- >>> pd.period_range(start='2017-01-01', end='2018-01-01', freq='M') PeriodIndex(['2017-01', '2017-02', '2017-03', '2017-04', '2017-05', '2017-06', '2017-07', '2017-08', '2017-09', '2017-10', '2017-11', '2017-12', '2018-01'], dtype='period[M]') If ``start`` or ``end`` are ``Period`` objects, they will be used as anchor endpoints for a ``PeriodIndex`` with frequency matching that of the ``period_range`` constructor. >>> pd.period_range(start=pd.Period('2017Q1', freq='Q'), ... end=pd.Period('2017Q2', freq='Q'), freq='M') PeriodIndex(['2017-03', '2017-04', '2017-05', '2017-06'], dtype='period[M]') rfzOOf the three parameters: start, end, and periods, exactly two must be specifiedND)rVr'r=)comZcount_not_nonersrSr rrRr4)rBrrr&r5r$r.r.r/ period_range(s4 r)NNNNN)E __future__rrrtypingrrkZnumpyrTZ pandas._libsrrrZpandas._libs.tslibsrr r r r Zpandas._libs.tslibs.parsingr rZpandas._typingrrZ pandas.errorsrZpandas.util._decoratorsrZpandas.core.dtypes.commonrrrrZpandas.core.dtypes.dtypesrZpandas.core.dtypes.missingrZpandas.core.arrays.periodrrrrZpandas.core.commoncorecommonrZpandas.core.indexes.baseZindexesrZZibaserZ pandas.core.indexes.datetimelikerZpandas.core.indexes.datetimesrr Zpandas.core.indexes.extensionr!Zpandas.core.indexes.numericr"dictZ_index_doc_kwargsupdaterr0Z _field_opsr4rr.r.r.r/sV             V