U Cv ft@sUdZddlmZddlZddlZddlZddlmZddlZ ddl Z ddl m Z ddlmZmZmZddlmZddlmZdd lmZz ddlZWnek rdZYnXd Zd Zd Zd ed<ze jje fZ!Wne"k re fZ!YnXddZ#ddZ$ddZ%dGddZ&e&dddZ'e&Z(e&ddZ)e'e(e)gZ*ddZ+d d!Z,d"d#Z-d$d%Z.dHd'd(Z/d)d*Z0d+d,Z1dId/d0Z2dJd1d2Z3dKd3d4Z4Gd5d6d6e j5Z6d7d8Z7d9d:Z8d;d<Z9d=d>Z:d?d@Z;dAdBZdS)Lz0DatetimeIndex analog for cftime.datetime objects) annotationsN timedelta)Version)_STANDARD_CALENDARScftime_to_nptimeinfer_calendar_name)_contains_cftime_datetimes)OPTIONS) is_scalard ztuple[type[Exception], ...]OUT_OF_BOUNDS_TIMEDELTA_ERRORScCsd|d|dS)Nz(?P<>))namepatternrr=/tmp/pip-unpacked-wheel-h316xyqg/xarray/coding/cftimeindex.pynamedPsrcCs d|dS)Nz(?:z)?r)xrrroptionalTsrcCs$|sdS|dtt|ddS)Nr)rtrailing_optional)xsrrrrXsr\-T\:cCsnd|ddf|ddf|ddf|ddf|ddfg}g}|D]&\}}}||rL|ndt||q6d t|d S) N)Nyearz\d{4}monthz\d{2}dayhourminutesecondr^$)appendrr)date_sep datetime_septime_seppiecesZ pattern_listseprZ sub_patternrrr build_pattern^sr.r)r)r+ )r*cCs8tD] }t||}|r|Sqtd|dS)Nz4no ISO-8601 or cftime-string-like match for string: ) _PATTERNSrematch groupdict ValueError)Zdatetime_stringrr2rrrparse_iso8601_likets r5cCsftdkrtd|ddd}t|}i}dD](}||d}|dk r,t|||<|}q,|jf||fS)NNo module named 'cftime'r)r r!r"r#r$r%)cftimeModuleNotFoundErrorr5getintreplace) date_typeZtimestrdefaultresultr;attrvalue resolutionrrr_parse_iso8601_with_reso~s   rBcCsp|dkr2||jdd||jdddtddfS|dkr|jdkrb||jdddtdd}n||j|jddtdd}||j|jd|fS|dkr||j|j|j}||tdddfS|d kr||j|j|j|j}||tddd fS|d kr,||j|j|j|j|j}||tddd fS|d krh||j|j|j|j|j|j}||tdddfStdS)zGeneralization of pandas.tseries.index.DatetimeIndex._parsed_string_to_bounds for use with non-standard calendars and cftime.datetime objects. r r microsecondsr! r")daysrDr#)hoursrDr$)minutesrDr%)secondsrDN)r rr!r"r#r$r%KeyError)r<rAparsedendstartrrr_parsed_string_to_boundssJ    rOcstfdd|DS)z(Adapted from pandas.tslib.get_date_fieldcsg|]}t|qSr)getattr.0datefieldrr sz"get_date_field..)nparray)Z datetimesrUrrTrget_date_fieldsrY0.0cs$|ffdd }|_||_t|S)z1Adapted from pandas.tseries.index._field_accessorcsRtdkrtdttjt|kr.t|jStddd|dtjddS)Nr6zThe z!rz2 accessor requires a minimum version of cftime of z . Found an installed version of .)r7r8r __version__rY_data ImportError)selfmin_cftime_versionrrrfs z_field_accessor..f)__name____doc__property)rZ docstringr`rbrrar_field_accessors rfcCs|jjrt|jdSdSdS)Nr)r]sizetyper_rrr get_date_typesrjcsntdkrtdt|dkrj|d}t|t|tjsFtdtfdd|Dsjtd|dS)Nr6rz?CFTimeIndex requires cftime.datetime objects. Got object of {}.c3s|]}t|VqdSN) isinstancerRr@r<rr sz-assert_all_valid_date_type..zJCFTimeIndex requires using datetime objects of all the same type. Got {}.) r7r8lenrhrldatetime TypeErrorformatall)datasamplerrnrassert_all_valid_date_types"  rw, , cCs|d|tt||S)z&Format a single row from format_times.r/)joinmapstr)timesindent separatorrow_endrrr format_rowsrcCst|tt|d}tt||}d} t|D]T} | dkrD|n|} | |dkrX|n|} || || d|} | t| | || d7} q4| S)z)Format values of cftimeindex as pd.Index.rrr)r~rr)maxCFTIME_REPR_LENGTHrpmathceilranger)index max_widthoffsetrfirst_row_offsetZintermediate_row_end last_row_endZ n_per_rowZn_rowsZrepresentationrowr~rZ times_for_rowrrr format_timess   rcCsdd|jdt|d|jdt|dkr:d|jdndd}dd|D}||}|S)z.Format attributes of CFTimeIndex for __repr__.'N)dtypelengthcalendarfreqcSsg|]\}}|d|qS)=r)rRkvrrrrVsz format_attrs..)rrprritemsrz)rrattrs attrs_strrrr format_attrss   rcsleZdZdZeddZeddZeddZedd Zed d Z ed d Z eddZ edddZ edddZ edddZeeZdOddZddZddZdd Zd!d"Zd#d$Zfd%d&ZdPd'd(Zd)d*Zd+d,Zd-d.Zd/d0d1d2d3Zd4d5Zd6d7Zd8d9Z d:d;Z!dQd=d>Z"d?d@Z#edAdBZ$edCdDZ%edEdFZ&dGdHZ'dIdJZ(dKdLZ)dMdNZ*Z+S)R CFTimeIndexafCustom Index for working with CF calendars and dates All elements of a CFTimeIndex must be cftime.datetime objects. Parameters ---------- data : array or CFTimeIndex Sequence of cftime.datetime objects to use in index name : str, default: None Name of the resulting index See Also -------- cftime_range r zThe year of the datetimer!zThe month of the datetimer"zThe days of the datetimer#zThe hours of the datetimer$zThe minutes of the datetimer%zThe seconds of the datetime microsecondz The microseconds of the datetimeZdayofyrz'The ordinal day of year of the datetimez1.0.2.1ZdayofwkzThe day of week of the datetimeZ daysinmonthz/The number of days in the month of the datetimez1.1.0.0NcKsJt||dkr t|dr |j}t|}tj|dd|_||_i|_|S)NrOr) rwhasattrrobject__new__rWrXr]_cache)clsrurkwargsr>rrrr=s zCFTimeIndex.__new__c Cs t|j}td}t|d}t|tkr>t|j||dd}nRt|jdt||ddd}t|jt d|||d}d|d |d |g}t |}|d |d |d }t||krt|||kr| ddd |d}|d |dd |d|d }|S)zA Return a string representation for this object. display_widthr)rrN,)rrr r/z...z([z], rryz], r) rhrcr rpITEMS_IN_REPR_MAX_ELSE_ELLIPSISrvalues"REPR_ELLIPSIS_SHOW_ITEMS_FRONT_ENDrzrr;) r_ klass_namerrZdatastrZ front_strZend_strrZ full_repr_strrrr__repr__Hs>    "zCFTimeIndex.__repr__c Cst|j||\}}|j}|jrt|rZ||dkr>||dksV||dkrZ||dkrZt|j|dd}|j|dd}t||S||k}||k} t || @S)a5Adapted from pandas.tseries.index.DatetimeIndex._partial_date_slice Note that when using a CFTimeIndex, if a partial-date selection returns a single element, it will never be converted to a scalar coordinate; this is in slight contrast to the behavior when using a DatetimeIndex, which sometimes will return a DataArray with a scalar coordinate depending on the resolution of the datetimes used in defining the index. For example: >>> from cftime import DatetimeNoLeap >>> da = xr.DataArray( ... [1, 2], ... coords=[[DatetimeNoLeap(2001, 1, 1), DatetimeNoLeap(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array([1]) Coordinates: * time (time) object 2001-01-01 00:00:00 >>> da = xr.DataArray( ... [1, 2], ... coords=[[pd.Timestamp(2001, 1, 1), pd.Timestamp(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array(1) Coordinates: time datetime64[ns] 2001-01-01 >>> da = xr.DataArray( ... [1, 2], ... coords=[[pd.Timestamp(2001, 1, 1, 1), pd.Timestamp(2001, 2, 1)]], ... dims=["time"], ... ) >>> da.sel(time="2001-01-01") array([1]) Coordinates: * time (time) datetime64[ns] 2001-01-01T01:00:00 rrFleft)sideright) rOr<r]is_monotonic_increasingrprKZ searchsortedslicerWZ flatnonzero) r_rArLrNrMr}rrZlhs_maskZrhs_maskrrr_partial_date_sliceos&+     zCFTimeIndex._partial_date_slicecCsBt|j|\}}z|||}Wntk r<t|YnX|S)zAAdapted from pandas.tseries.index.DatetimeIndex._get_string_slice)rBr<rrK)r_keyrLrAlocrrr_get_string_slices zCFTimeIndex._get_string_slicec Cs|j|d|d}|j|d|d}t|j||j}t|j||j}|jr`||k|dkB}n||k|dkB}t|||} |dk r||| |} | S)z.Adapted from pandas.Index._get_nearest_indexerpad)limitZbackfillrFN)Z get_indexerabsrrrWwhere_filter_indexer_tolerance) r_targetr toleranceZ left_indexerZ right_indexerZleft_distancesZright_distances conditionindexerrrr_get_nearest_indexersz CFTimeIndex._get_nearest_indexercCsJt|tjr"t|j||j}nt|j||}t||k|d}|S)z3Adapted from pandas.Index._filter_indexer_tolerancerF)rlpdIndexrrrWr)r_rrrZdistancerrrrs  z%CFTimeIndex._filter_indexer_tolerancecs$t|tr||St|SdS)z7Adapted from pandas.tseries.index.DatetimeIndex.get_locN)rlr|rsuperget_locr_r __class__rrrs  zCFTimeIndex.get_loccCsbt|ts|St|j|\}}t|j||\}}|jrRt|dkrR|dkrN|S|S|dkr^|S|S)aJAdapted from pandas.tseries.index.DatetimeIndex._maybe_cast_slice_bound Note that we have never used the kind argument in CFTimeIndex and it is deprecated as of pandas version 1.3.0. It exists only for compatibility reasons. We can remove it when our minimum version of pandas is 1.3.0. rr)rlr|rBr<rOZis_monotonic_decreasingrp)r_labelrkindrLrArNrMrrr_maybe_cast_slice_bounds z#CFTimeIndex._maybe_cast_slice_boundcCsXt|jttkr |j|St|trD|j||j|j |j S|j| |SdS)z9Adapted from pandas.tseries.index.DatetimeIndex.get_valueN) rWZasarrayrboolZilocrlrZ slice_indexerrNstopstepr)r_Zseriesrrrr get_values   zCFTimeIndex.get_valuec CsTz2||}t|p0t|tkp0t|tjo0|jWStt t fk rNYdSXdS)zKAdapted from pandas.tseries.base.DatetimeIndexOpsMixin.__contains__FN) rr rhrrlrWZndarrayrgrKrrr4)r_rr>rrr __contains__s  zCFTimeIndex.__contains__cCs ||S)z-Needed for .loc based partial-string indexing)rrrrrcontainsszCFTimeIndex.containsz int | floatzstr | timedelta)nrcCsNt|tr|||St|tr>> index = xr.cftime_range("2000", periods=1, freq="M") >>> index CFTimeIndex([2000-01-31 00:00:00], dtype='object', length=1, calendar='standard', freq=None) >>> index.shift(1, "M") CFTimeIndex([2000-02-29 00:00:00], dtype='object', length=1, calendar='standard', freq=None) >>> index.shift(1.5, "D") CFTimeIndex([2000-02-01 12:00:00], dtype='object', length=1, calendar='standard', freq=None) r) to_offsetz9'freq' must be of type str or datetime.timedelta, got {}.N)rlrr|xarray.coding.cftime_offsetsrrrrs)r_rrrrrrshifts"    zCFTimeIndex.shiftcCs&t|tjr|}tt||SrkrlrTimedeltaIndexto_pytimedeltarrWrXr_otherrrr__add__4s zCFTimeIndex.__add__cCs&t|tjr|}t|t|Srkrrrrr__radd__9s zCFTimeIndex.__radd__cCst|rtt||St|tjrs  zCFTimeIndex.__sub__cCs8zt|t|WStk r2tdYnXdSr)rrrWrXrr4rrrr__rsub__Ns zCFTimeIndex.__rsub__FcCs<t|}t|}|tkr2|s2tjd|tddt|S)aIf possible, convert this index to a pandas.DatetimeIndex. Parameters ---------- unsafe : bool Flag to turn off warning when converting from a CFTimeIndex with a non-standard calendar to a DatetimeIndex (default ``False``). Returns ------- pandas.DatetimeIndex Raises ------ ValueError If the CFTimeIndex contains dates that are not possible in the standard calendar or outside the pandas.Timestamp-valid range. Warns ----- RuntimeWarning If converting from a non-standard calendar to a DatetimeIndex. Warnings -------- Note that for non-standard calendars, this will change the calendar type of the index. In that case the result of this method should be used with caution. Examples -------- >>> times = xr.cftime_range("2000", periods=2, calendar="gregorian") >>> times CFTimeIndex([2000-01-01 00:00:00, 2000-01-02 00:00:00], dtype='object', length=2, calendar='standard', freq=None) >>> times.to_datetimeindex() DatetimeIndex(['2000-01-01', '2000-01-02'], dtype='datetime64[ns]', freq=None) zConverting a CFTimeIndex with dates from a non-standard calendar, {!r}, to a pandas.DatetimeIndex, which uses dates from the standard calendar. This may lead to subtle errors in operations that depend on the length of time between dates.r) stacklevel) rrrwarningswarnrsRuntimeWarningrZ DatetimeIndex)r_ZunsafeZnptimesrrrrto_datetimeindexWs'  zCFTimeIndex.to_datetimeindexcstfdd|jDS)a Return an Index of formatted strings specified by date_format, which supports the same string format as the python standard library. Details of the string format can be found in `python string format doc `__ Parameters ---------- date_format : str Date format string (e.g. "%Y-%m-%d") Returns ------- pandas.Index Index of formatted strings Examples -------- >>> rng = xr.cftime_range( ... start="2000", periods=5, freq="2MS", calendar="noleap" ... ) >>> rng.strftime("%B %d, %Y, %r") Index(['January 01, 2000, 12:00:00 AM', 'March 01, 2000, 12:00:00 AM', 'May 01, 2000, 12:00:00 AM', 'July 01, 2000, 12:00:00 AM', 'September 01, 2000, 12:00:00 AM'], dtype='object') csg|]}|qSr)strftimerQ date_formatrrrVsz(CFTimeIndex.strftime..)rrr])r_rrrrrszCFTimeIndex.strftimecs<ddlm|dddtjfdd|jDtjdS)z@Convert to integers with units of microseconds since 1970-01-01.r) exact_cftime_datetime_differencercsg|]}t|qSr)_total_microsecondsrQepochrrrrVsz$CFTimeIndex.asi8..r)Zxarray.core.resample_cftimerr<rWrXrZint64rirrrasi8s  zCFTimeIndex.asi8cCsddlm}||S)z0The calendar used by the datetimes in the index.r)r)xarray.coding.timesr)r_rrrrrs zCFTimeIndex.calendarcCsddlm}||S)z-The frequency used by the dates in the index.r) infer_freq)Zxarray.coding.frequenciesr)r_rrrrrs zCFTimeIndex.freqc Cs\ddlm}m}||}t||s0t|dt|}|j}|||}t||j |j S)z%Round dates using a specified method.r) CFTIME_TICKSrz is a non-fixed frequency) rrrrlr4rZ as_timedeltar_cftimeindex_from_i8r<r) r_rmethodrrrunitrZroundedrrr_round_via_methods   zCFTimeIndex._round_via_methodcCs ||tS)aRound dates down to fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )r _floor_intr_rrrrfloorszCFTimeIndex.floorcCs ||tS)aRound dates up to fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )r _ceil_intrrrrrszCFTimeIndex.ceilcCs ||tS)aRound dates to a fixed frequency. Parameters ---------- freq : str The frequency level to round the index to. Must be a fixed frequency like 'S' (second) not 'ME' (month end). See `frequency aliases `_ for a list of possible values. Returns ------- CFTimeIndex )r_round_to_nearest_half_evenrrrrroundszCFTimeIndex.round)N)N)F),rc __module__ __qualname__rdrfr r!r"r#r$r%rZ dayofyearZ dayofweekZ days_in_monthrerjr<rrrrrrrrrrrrrrrrrrrrrrrrr __classcell__rrrrrs`         '@     . 5    rcCst||\}}|Srk)rB)r<Z datetime_strrS_rrr_parse_iso8601_without_resosrcs$tfdd|D|jS)aCreate a numpy array from an array of strings. For use in generating dates from strings for use with interp. Assumes the array is either 0-dimensional or 1-dimensional. Parameters ---------- strings : array of strings Strings to convert to dates date_type : cftime.datetime type Calendar type to use for dates Returns ------- np.array csg|]}t|qSr)r)rRsrnrrrVsz2_parse_array_of_cftime_strings..)rWrXZravelZreshapeshape)stringsr<rrnr_parse_array_of_cftime_strings s rcCst|}t|dtS)z.ra)rWrXr)rr<rdatesrrrr(s rcCs|tddS)zCompute the total number of microseconds of a datetime.timedelta. Parameters ---------- delta : datetime.timedelta Input timedelta. Returns ------- int rrCr)deltarrrr=s rcCs|t||SCopied from pandas.rW remainderrrrrrrLsrcCs|t| |Srr r rrrrQsrcCsj|drt||d|St||\}}t||dkt||dk|d}||d7<||S)r rr)rrWdivmod logical_or logical_and)rrZquotientr maskrrrrVs r)rrr)NrZ)rrxry)rxrryr)rx)?rd __future__rrr1rrqrZnumpyrWZpandasrZpackaging.versionrrrrrZxarray.core.commonr Zxarray.core.optionsr Zxarray.core.utilsr r7r^rrr__annotations__errorsZOutOfBoundsTimedelta OverflowErrorrAttributeErrorrrrr.Z_BASIC_PATTERNZ_EXTENDED_PATTERNZ_CFTIME_PATTERNr0r5rBrOrYrfrjrwrrrrrrrrrrrrrrrrrst(              ,    n