a Of"J@sddlmZddlmZmZddlmZddlZddlZddl m Z ddl m Z mZmZmZmZddlmZddlmZmZdd lmZdd lmZdd lmZdd lmZdd lm Z ddl!m"Z"m#Z#m$Z$m%Z%ddl&m'm(Z)ddl*m'm+m,Z-ddl*m.Z.ddl/m0Z0ddl1m2Z2m3Z3ddl4m5Z5erXddl6m7Z7ddl8m9Z9m:Z:m;Z;mZ>e>?ddiddiZ@ddZAe5gde"jBe"dde5dge"Gdd d e0ZCd&d!d"d d#d$d%ZDdS)') annotations)datetime timedelta) TYPE_CHECKINGN)index) BaseOffsetNaTPeriod ResolutionTick)OFFSET_TO_PERIOD_FREQSTR)cache_readonlydoc)find_stack_level) is_integer) PeriodDtype) ABCSeries)is_valid_na_for_dtype) PeriodArray period_arrayraise_on_incompatiblevalidate_dtype_freq)maybe_extract_name)DatetimeIndexOpsMixin) DatetimeIndexIndex) inherit_names)Hashable)DtypeDtypeObjSelfnptZ target_klasszPeriodIndex or list of PeriodsklassrcKsZ|d}|jdkrF|dd}t|}t||d}|j|fi|S||fi|SdS)Ndataint64freqdtype)popr'rr _simple_new)clsdvaluesr%r'r-S/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/indexes/period.py_new_PeriodIndexDs    r/)strftime start_timeZend_timeT)wrapZ is_leap_yearc s8eZdZUdZdZded<ded<ded<eZd Ze d d d d Z e dd ddZ e ejfddded]dddddZe ejd^dddddZe e ejjdd dd Ze e ejjdd d!d"Ze e ejjdd d#d$Zd_d&d'd(dd)d*d+Zeddddddddd,dd d-d.Zedd/dd d0d1Ze d2d d3d4Zd5d d6d7Zd8d'd9d:d;Zddfd?d@ Ze d'd dAdBZe dd dCdDZfdEdFZdGdHZ dIdJdKdLdMZ!dNdIdOdPdQZ"e e#j$ddRfdSdT Z$ddNdUdVdWZ%e e#j&d`dYddZd[d\Z&Z'S)a 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 .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. month : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. quarter : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. day : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. hour : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. minute : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. second : int, array, or Series, default None .. deprecated:: 2.2.0 Use PeriodIndex.from_fields instead. 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 from_fields from_ordinals 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.from_fields(year=[2000, 2002], quarter=[1, 3]) >>> idx PeriodIndex(['2000Q1', '2002Q3'], dtype='period[Q-DEC]') Z periodindexr_datarr%rr'Tztype[libindex.PeriodEngine])returncCstjSN)libindexZ PeriodEngineselfr-r-r. _engine_typeszPeriodIndex._engine_typer cCs|jjSr6)r'_resolution_objr8r-r-r.r;szPeriodIndex._resolution_objzpandas.arrays.PeriodArray)otherZ other_nameNEstrr )howr5cCs"|j||}t|j||jdSNname)r4asfreqtyper)rBr9r%r?arrr-r-r.rCszPeriodIndex.asfreqstartrcCs|j||}tj||jdSr@)r4 to_timestamprr)rBrEr-r-r.rHszPeriodIndex.to_timestamprcCst|jj|jdSr@)rr4hourrBr8r-r-r.rIszPeriodIndex.hourcCst|jj|jdSr@)rr4minuterBr8r-r-r.rJszPeriodIndex.minutecCst|jj|jdSr@)rr4secondrBr8r-r-r.rKszPeriodIndex.secondFz Dtype | NoneboolHashable | None)r'copyrBr5c Kshd}d} |s$t|ttfr$|j} t||sVttt||} td| nt |rpt j dt t d|durt j dt t dt|||}|dur|dur|s|d|jfi|d|ij}d}n|r|durtdtd nt||}|r$t||r$|j|kr$||}|dur^|dur^tj|tjd }t|}t||d }n*|dur||dur|td n t||d }|r|}|j||| d S)N>rIyearrKmonthquarterrJdayz-__new__() got an unexpected keyword argument zXConstructing PeriodIndex from fields is deprecated. Use PeriodIndex.from_fields instead.) stacklevelzThe 'ordinal' keyword in PeriodIndex is deprecated and will be removed in a future version. Use PeriodIndex.from_ordinals instead.r%Fz Cannot pass both data and fieldsz#Cannot pass both ordinal and fieldsr&z!Cannot pass both data and ordinal)r#r%)rBrefs) isinstancerrZ _referencessetissubsetnextiter TypeErrorlenwarningswarn FutureWarningrrZ_raise_scalar_data_error from_fieldsr4 ValueErrorrr%rCnpasarrayr$rrrrNr)) r*r#Zordinalr%r'rNrBfieldsZvalid_field_setrTargumentr-r-r.__new__sT         zPeriodIndex.__new__)rOrQrPrRrIrJrKr%c Cs>|||||||d} dd| D} tj| |d} || S)N)rOrQrPrRrIrJrKcSsi|]\}}|dur||qSr6r-).0keyvaluer-r-r. Wz+PeriodIndex.from_fields..)rcr%)itemsrZ _from_fieldsr)) r*rOrQrPrRrIrJrKr%rcrFr-r-r.r_As zPeriodIndex.from_fieldsrAcCs4tj|tjd}t|}tj||d}|j||dS)Nr&rA)rarbr$rrr))r*Zordinalsr%rBr'r#r-r-r. from_ordinals[szPeriodIndex.from_ordinalsznpt.NDArray[np.object_]cCstj|tdS)Nr&)rarbobjectr8r-r-r.r,eszPeriodIndex.valueszint | npt.NDArray[np.int64]cCst|ttjttjfr4t|jtrx|j|}|SnDt|t r^|j |jj krR|j St ||nt |rxt|tstJ|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)rUrraZ timedelta64r Zndarrayr%r4Z _check_timedeltalike_freq_compatrbasenrrint)r9r<deltar-r-r._maybe_convert_timedeltais    z$PeriodIndex._maybe_convert_timedeltar)r'r5cCs |j|kS)zF Can we compare values of the given dtype to our own? r&)r9r'r-r-r._is_comparable_dtypesz PeriodIndex._is_comparable_dtypeznpt.NDArray[np.bool_]z np.ndarray)wheremaskr5cs<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)rUrr3Z_valuesr%rZsuper asof_locs)r9rtru __class__r-r.rxs   zPeriodIndex.asof_locscCsHt|dkrdS|jstd|j}t|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 monotonicN)r[Zis_monotonic_increasingr`Zasi8rLall)r9r,r-r-r.is_fulls  zPeriodIndex.is_fullcCsdS)Nperiodr-r8r-r-r. inferred_typeszPeriodIndex.inferred_typecs(t||}|j|jkr$||}|Sr6)rw_convert_tolerancer'rr)r9Z tolerancetargetryr-r.rs  zPeriodIndex._convert_tolerancec CsV|}||t||jr t}nt|trz||\}}Wn6tyr}ztd|d|WYd}~n d}~00| |rz| ||WSty}zt||WYd}~n d}~00||j kr| |}nt|n6t|t r||n t|tr | |}nt|zt||WStyP}zt||WYd}~n d}~00dS)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 periodN)Z_check_indexing_errorrr'rrUr>Z_parse_with_resor`KeyErrorZ_can_partial_date_sliceZ_partial_date_slicer;_cast_partial_indexing_scalarr _disallow_mismatched_indexingrrget_loc)r9rgorig_keyparsedresoerrr-r-r.rs4   (         zPeriodIndex.get_locr None)rgr5cCs|j|jkrt|dSr6)Z_dtyper'r)r9rgr-r-r.rs z)PeriodIndex._disallow_mismatched_indexingr)labelr5c CsFzt||jd}Wn.ty@}zt||WYd}~n d}~00|S)Nrv)r r%r`r)r9rrrr-r-r.rs  z)PeriodIndex._cast_partial_indexing_scalar)sidecs"t|tr||}t||Sr6)rUrrrw_maybe_cast_slice_bound)r9rrryr-r.rs  z#PeriodIndex._maybe_cast_slice_bound)rrcCs<t|j|j}t||d}|j|jdd|j|jddfS)NrvrG)r?end)r getZ attr_abbrevr rCr%)r9rrr%Zivr-r-r._parsed_string_to_boundss z$PeriodIndex._parsed_string_to_boundsr{rp)periodsr5cCs&|durtdt|jd||S)Nz%`freq` argument is not supported for z.shift)rZrD__name__)r9rr%r-r-r.shifts zPeriodIndex.shift)Nr=)NrG)NNNNFN)r{N)(r __module__ __qualname____doc__Z_typ__annotations__rZ _data_clsZ!_supports_partial_string_indexingpropertyr:r r;rrC_shared_doc_kwargsrHrIfgetrJrKre classmethodr_rlr,rrrsrxrrrrrrrrrr __classcell__r-r-ryr.r3Ps \   W "    >r3z int | NonerM)rrBr5cCsnt|||dkrtd|dur:t|ts:t|ts:d}t||||\}}t|}t||d}t||dS)a Return a fixed frequency PeriodIndex. The day (calendar) is the default frequency. Parameters ---------- start : str, datetime, date, pandas.Timestamp, or period-like, default None Left bound for generating periods. end : str, datetime, date, pandas.Timestamp, 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 specifiedNDr&rA) comZcount_not_noner`rUr rZ_generate_rangerr3)rGrrr%rBr#r'r-r-r. period_range#s8 r)NNNNN)E __future__rrrtypingrr\numpyraZ pandas._libsrr7Zpandas._libs.tslibsrrr r r Zpandas._libs.tslibs.dtypesr Zpandas.util._decoratorsr rZpandas.util._exceptionsrZpandas.core.dtypes.commonrZpandas.core.dtypes.dtypesrZpandas.core.dtypes.genericrZpandas.core.dtypes.missingrZpandas.core.arrays.periodrrrrZpandas.core.commoncorecommonrZpandas.core.indexes.baseZindexesrnZibaserZ pandas.core.indexes.datetimelikerZpandas.core.indexes.datetimesrrZpandas.core.indexes.extensionrcollections.abcrZpandas._typingrrr r!dictZ_index_doc_kwargsupdaterr/Z _field_opsr3rr-r-r-r.sX                 Q