B W0dL@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)Ndataint64freq)r&)popdtyper _simple_new)clsdvaluesr&r-L/var/www/html/venv/lib/python3.7/site-packages/pandas/core/indexes/period.py_new_PeriodIndexEs     r/strftime start_timeend_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||jdS)N)r5)r6asfreqtyper)r5)selfr&r:arrr-r-r.r<szPeriodIndex.asfreqstartr)r;cCs|j||}tj||jdS)N)r5)r6 to_timestamprr)r5)r>r&r:r?r-r-r.rAszPeriodIndex.to_timestampr"cCst|jj|jdS)N)r5)r"r6hourr5)r>r-r-r.rBszPeriodIndex.hourcCst|jj|jdS)N)r5)r"r6minuter5)r>r-r-r.rCszPeriodIndex.minutecCst|jj|jdS)N)r5)r"r6secondr5)r>r-r-r.rDszPeriodIndex.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)NyearmonthdayZquarterrBrCrDrz-__new__() got an unexpected keyword argument )r&)r()r$r&)r5)setissubsetlist TypeErrorrr_generate_ranger isinstancer&r<npasarrayr%rrFr)) r*r$Zordinalr&r(rFr5fieldsZvalid_field_setargumentZfreq2r-r-r.__new__s2     zPeriodIndex.__new__z np.ndarraycCstj|tdS)N)r()rPrQobject)r>r-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)rOrrPZ timedelta64r Zndarrayr&r6Z _check_timedeltalike_freq_compatrbasenrr)r>r7deltar-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)rOrr&)r>r(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)rOrr4Z_valuesr&rMsuper asof_locs)r>r[r\) __class__r-r.r^Bs   zPeriodIndex.asof_locs)rFcs^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) stacklevelr@tz)r:)rF) rr no_defaultwarningswarn FutureWarningrgetattrrAZ tz_localizer]astype)r>r(rFr:rb)r_r-r.rhOs   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 monotonicNr`)lenZis_monotonic_increasing ValueErrorZasi8all)r>r,r-r-r.is_fullfs  zPeriodIndex.is_fullcCsdS)NZperiodr-)r>r-r-r. inferred_typesszPeriodIndex.inferred_typecs(t||}|j|jkr$||}|S)N)r]_convert_tolerancer(rY)r> tolerancetarget)r_r-r.rp|s  zPeriodIndex._convert_tolerancec CsL|}t|st|t||jr(t}nt|tr0y||}|Stt fk rZYnXyt ||j \}}Wn8t t fk r}zt d|d|Wdd}~XYnXt|} | jj} |jj} | | kst| | ks| tjkr|jj jdkrt||j d}|j|||d}|S|dkr*t |n|}nt|trn|j } |j } | j| jkrd| j| jkst |nXt|tryt||j d}Wn.t k r}zt ||Wdd}~XYnXnt |yt||j d}Wn.t k r}zt ||Wdd}~XYnXyt||||St k rF}zt ||Wdd}~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 periodNB)r&)methodrq)rrrr(r rOr9_get_string_slicerMrlrr&r KeyErrorr from_attrname freq_groupvaluefreq_group_codeAssertionErrorZRESO_DAYr5r get_locrWZ_period_dtype_coderr )r>keyrtrqorig_keylocZasdtreso_strerrresogrpfreqnZsfreqZkfreqr-r-r.r|s^   $       zPeriodIndex.get_loc)sidec Cs|dddtjgkst||ddt|trlabelrrparsedrrrlowerupperr-r-r.rs     z#PeriodIndex._maybe_cast_slice_boundr r)rrcCs4|j}t||jd}|j|jdd|j|jddfS)N)r&r@)r:end)rxr ryr<r&)r>rrrZivr-r-r.rsz$PeriodIndex._parsed_string_to_bounds)rcCs:t|tstt||f|j}|jj}|j|ks6tdS)N) rOr r{r=rxr(rzryrl)r>rrrr-r-r._validate_partial_date_slices  z(PeriodIndex._validate_partial_date_slice)r}c CsXt||j\}}t|}y |||Stk rR}zt||Wdd}~XYnXdS)N)rr&r rwZ_partial_date_slicerv)r>r}rrrrr-r-r.rus   zPeriodIndex._get_string_slice)Nr8)Nr@)NNNNFN)NN)(__name__ __module__ __qualname____doc__Z_typ _attributes__annotations__rZ _data_clslibindexZ PeriodEngineZ _engine_typeZ!_supports_partial_string_indexingrr<_shared_doc_kwargsrApropertyrBfgetrCrDrTr,rYrZr^r rhrrcrnrorpr|rrrru __classcell__r-r-)r_r.r4PsT J 7$    ^( 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]') r`zOOf the three parameters: start, end, and periods, exactly two must be specifiedND)rR)r&)r5)comZcount_not_nonerlrOr rrNr4)r@rrr&r5r$r-r-r. period_range(s4 r)NNNNN)E __future__rrrtypingrrdnumpyrPZ 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.baseZindexesrVZibaserZ pandas.core.indexes.datetimelikerZpandas.core.indexes.datetimesrr Zpandas.core.indexes.extensionr!Zpandas.core.indexes.numericr"dictZ_index_doc_kwargsupdaterr/Z _field_opsr4rr-r-r-r.sF             V