dg0ddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZddlZddlZddlmZddlmZmZdd lmZdd lmZmZmZdd l m!Z!m"Z"dd l#m$Z$m%Z%dd l&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.e r ddl/m0Z0ddl1m2Z2dHdZ3GddZ4Gdde4Z5Gdde4Z6Gdd e4Z7dId!Z8 dJdKd'Z9 dLdMd3Z:d4Z;dNd6Zd9Z? dPdQd<Z@d=ZAd>ZBd?ZCd@ZDdRdAZEdRdBZFdCZGdDZHd"dEdFZIdGZJdS)S) annotationsN)HashableSequence)partial)Number) TYPE_CHECKINGAnyCallableget_args)utils)_contains_datetime_like_objects ones_like) apply_ufunc)datetime_to_numericpushtimedelta_to_numeric)OPTIONS_get_keep_attrs)get_chunked_array_typeis_chunked_array)Interp1dOptions InterpOptions) OrderedSet is_scalar)Variablebroadcast_variables) DataArray)DatasetobjDataset | DataArray | Variabledimrindexrc:t|g|}t||z}|}||}|||d}||d|||i|||d|d| |gz }|S)zv Return an object where each NaN element in 'obj' is replaced by the length of the gap the element is in. )r!rupper)r!label) rrnotnullwhereffillfillnadiffreindexbfillmax)rr!r"arangevalid valid_arangecumulative_nansnan_block_lengthss 3lib/python3.11/site-packages/xarray/core/missing.py_get_nan_block_lengthsr5s cUE " "Es^^e #F KKMME<<&&L"((S(1188qBBO G44 #s3x ! ! u 3 vq   b L,,#,777 8 8  cHeZdZUdZded<ded<ded<ded<d Zd Zd S) BaseInterpolatorz@Generic interpolator class for normalizing interpolation methodszdict[str, Any] cons_kwargs call_kwargsr fstrmethodc(|j|fi|jSN)r;r:selfxs r4__call__zBaseInterpolator.__call__Astva,,4+,,,r6cNd|jj|jS)Nz{type}: method={method})typer=)format __class____name__r=)rAs r4__repr__zBaseInterpolator.__repr__Ds+(//(0   r6N)rH __module__ __qualname____doc____annotations__rCrIr6r4r8r89sdJJKKKKKK---     r6r8c eZdZdZddZdZdS)NumpyInterpolatorzVOne-dimensional linear interpolation. See Also -------- numpy.interp linearNc|dkrtd||_tj|_i|_d|i|_||_||_|j j dkr tj ntj tj dzz}|||_ ||_ dSt|tr/t!|dkr|d|_ |d|_ dSt#|r||_ ||_ dSt|d ) NrQz7only method `linear` is valid for the NumpyInterpolatorperiodc?rz is not a valid fill_value) ValueErrorr=npinterpr;r9r:_xi_yidtypekindnan_left_right isinstancerlenr)rAxiyir= fill_valuerSr_s r4__init__zNumpyInterpolator.__init__Rs X  VWW W $f- ,,bff"&26B;2F  DJDKKK  H - - H#j//Q2F2F#ADJ$Q-DKKK z " " H#DJ$DKKK FFFGG Gr6cZ|j||j|jf|j|jd|jS)N)leftright)r;r[r\r`rar:r@s r4rCzNumpyInterpolator.__call__lsHtv H H +       r6)rQNN)rHrJrKrLrgrCrNr6r4rPrPJsEHHHH4     r6rPc&eZdZdZ ddZdS)ScipyInterpolatorznInterpolate a 1-D function using Scipy interp1d See Also -------- scipy.interpolate.interp1d NTFc `ddlm} |td|dkr|td|}||_| |_i|_|jjdkr tj ntj tj dzz} | |dkr| | f}n|| }| ||f|j||||d |j|_ dS) Nr)interp1dzPmethod is a required argument, please supply a valid scipy.inter1d method (kind) polynomialz(order is required when method=polynomialrTrUrQ)r^rf bounds_error assume_sortedcopy) scipy.interpolaternrXr=r9r:r]r^rYr_r;) rArdrer=rfrqrrrporderkwargsrnr_s r4rgzScipyInterpolator.__init__s /..... >4  \ ! !} !KLLLF ! ,,bff"&26B;2F  &H"4"4cJJ  J    !%'      r6)NNTFFNrHrJrKrLrgrNr6r4rlrlwsF . . . . . . r6rlc$eZdZdZ ddZdS)SplineInterpolatorzOne-dimensional smoothing spline fit to a given set of data points. See Also -------- scipy.interpolate.UnivariateSpline splineNrc ddlm} |dkrtd||_||_||d|_|td| ||fd|i|j|_dS)Nr)UnivariateSplineryz8only method `spline` is valid for the SplineInterpolator)nuextz.SplineInterpolator does not support fill_valuek)rsr|rXr=r9r:r;) rArdrer=rfrtr}r~rur|s r4rgzSplineInterpolator.__init__s 766666 X  WXX X !"$S11  !MNN N!!"bFFEFT5EFFr6)ryNrzrNrvrNr6r4rxrxsI GGGGGGr6rxc t||j|j}|jD]"\}}||jvr||fd|i|||<|||<#|S)zWrapper for datasets)coordsattrsr!)rErr data_varsitemsdims)funcrAr!rudsnamevars r4_apply_over_vars_with_dimrs~ d4;dj 9 9 9B^))++ c #(??tC33S3F33BtHHBtHH Ir6Tuse_coordinate str | boolstrictboolc ddlm}|dur@||}tj|j|tjS|dur||}nG|j|}|j dkrtd|d|j d | }t|tjr||_|r>|jstd |jd |jstd |jd t||tjfrYt'|dd dd}t||r|j}t+t-||d|f} |jtj}nB#t0tf$r.t1d |jdt'|jdwxYw|S)aSReturn index to use for x values in interpolation or curve fitting. Parameters ---------- arr : DataArray Array to interpolate or fit to a curve. dim : str Name of dimension along which to fit. use_coordinate : str or bool If use_coordinate is True, the coordinate that shares the name of the dimension along which interpolation is being performed will be used as the x values. If False, the x values are set as an equally spaced sequence. strict : bool Whether to raise errors if the index is either non-unique or non-monotonic (default). Returns ------- Variable Numerical values for the x-coordinates. Notes ----- If indexing is along the time dimension, datetime coordinates are converted to time deltas with respect to 1970-01-01. r CFTimeIndexF)r]TrWz/Coordinates used for interpolation must be 1D, z is zD.zIndex z! must be monotonically increasingz has duplicate valuesins)offset datetime_unit)datarzL must be castable to float64 to support interpolation or curve fitting, got .)xarray.coding.cftimeindexr get_axis_numrYr/shapefloat64 get_indexrndimrXto_indexrbpd MultiIndexris_monotonic_increasing is_unique DatetimeIndexrEvaluesrrastype TypeErrorrH)arrr!rrraxisr"rs r4get_clean_interp_indexrsP<655555$$y4 ;;;; c"" >* :??6!66',z666   %''  K, WUejUUUVV V KIejIIIJJ J%+r'7899 eAha++ e[ ) ) !LE$U6NNN      ##BJ// z "    KUZ K K37;;3G K K K     Ls $F((?G'rQHashable | None bool | strr=rlimit int | Nonemax_gap@int | float | str | pd.Timedelta | np.timedelta64 | dt.timedelta keep_attrs bool | Nonec <ddlm}|td|t|||} |t |j} t |std||jvrJt|j| tj |fr|rt|}|s4t|ttjfst#d| dt%|||} t'|fi|\} }t)t*| fi|} |t-d }t/j5t/jd d t4t/jd d t4t7| || |g|gg|gg|jgdd | j|j}dddn #1swxYwY||| }|B||j vrtdtC||| }|||k}|S)z2Interpolate values according to different methods.rrNzdim is a required argumentzmax_gap must be a scalar.zPExpected integer or floating point max_gap since use_coordinate=False. Received r)rT)defaultignoreoverflowz invalid value parallelized)input_core_dimsoutput_core_dims output_dtypesdask vectorizerz6max_gap not implemented for unlabeled coordinates yet.)"rrNotImplementedError_get_valid_fill_maskrErHrrX_indexesrbto_pandas_indexrrrrrYnumberrr_get_interpolatorrfunc_interpolate_narwarningscatch_warningsfilterwarningsRuntimeWarningrr] transposerr(rr5)rAr!rr=rrrrurvalidsmax_typer" interp_class interpolatorrr3s r4 interp_nar4s655555 {!">??? %dC77==)!! :899 9 4=  c"2244r7G6U ! !+733G g ':;; rgorrr #4^ L L LE,V>>v>>L&. GGGGL$T222  " "   *nEEE/>JJJ k   !USEN"eW:,!     TY                   ii dk ! !%H 34eDDii)W455 JsA(F66F:=F:c |}tj|}|}|}|dks|t |dz kr|S|||||fi|}|||||<|S)z8helper function to apply interpolation along 1 dimensionrrW)rrrisnullsumrc) ryrBruoutnansnonansn_nansr;s r4rrs &&((C 9Q<K|]}t|D]}|VdSr?)r ).0vvvs r4 z$_get_interpolator..s6TTQR TT""TTTTTTTr6rQrfN extrapolater=z< is not a vectorizeable interpolator. Available methods are barycentricBarycentricInterpolatorkrogKroghInterpolatorpchipPchipInterpolatorryakimaAkima1DInterpolatorz" is not a valid scipy interpolatorz is not a valid interpolator) r rtuplergetupdaterPrlrXrrx)r=rruinterp1d_methods valid_methodsrs r4rrs 00TTh}&=&=TTTTTM ( <..-??"@  V $$$( = % % % MMM ( ( (,LL  L<<)9<< } $ $./H&QQLL v  ./BFKKLL w  ./BFKKLL x   MMM ( ( (-LL w  ./DfMMLLJJJKK KF@@@AAA  r6c ddg}||vr'||td|}nt|d||fS)rrQnearestrinterpnzH is not a valid interpolator for interpolating over multiple dimensions.)rrrX)r=rurrs r4_get_interpolator_ndr so y)M  V $$$*9f==  ( ( (    r6c||dzi}tj|jd}|jdddi||d|d|kS)zQhelper function to determine values that can be filled when limit is not NonerW_window min_periodsF)rf)skipnarN)r get_temp_dimnamerrrolling constructr)rr!rkwnew_dims r4rrs uqy B$SXy99G   & & &"$ & & 7u - - WU # #   r6ci}|D]\}\}}tj|j}tj|j}|}||gd} ||gd} tt| dz d| dz||<||||f||<|j di||fS)zwSpeed up for linear and nearest neighbor method. Only consider a subspace that is needed for the interpolation rrrVrrN) rrYnanminrnanmaxr get_indexeritemslicer.isel) rindexes_coordsindexesr!rBnew_xminvalmaxvalr"iminimaxs r4 _localizer,sG)//1177Za5<((5<((   &) <<AACC  &) <<AACCS1--tax88  6s 38  g   ..r6ct|}t|}tt|D]}t||rs||j}|||tj||<|||tj||<||fS)zrMake x and new_x float. This is particularly useful for datetime dtype. x, new_x: tuple of np.ndarray )rr]) listrangercr rmin _to_numericrYr)rBrixmins r4 _floatize_xr%<s QA KKE 3q66]] K K *1Q4 0 0 K Q4;??$$DQ4##4rz#BBAaDQx++4rz+JJE!H e8Or6c s|S|dd|d<|}tD]|}tt fdD\}}t |}fd|jD}|z} |t|djz} t|j| j ||||} t| | |j d}t} |jD]B} | vr'| | dj-| | Ct| dkr |j| }|S) aMake an interpolation of Variable Parameters ---------- var : Variable indexes_coords Mapping from dimension name to a pair of original and new coordinates. Original coordinates should be sorted in strictly ascending order. Note that all the coordinates should be Variable objects. method : string One of {'linear', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic'}. For multidimensional interpolation, only {'linear', 'nearest'} can be used. **kwargs keyword arguments to be passed to scipy.interpolate Returns ------- Interpolated Variable See Also -------- DataArray.interp Dataset.interp rpFc g|] }| SrNrNrdrs r4 zinterp..ws999q*999r6cg|]}|v| SrNrN)rr)rs r4r*zinterp..{s???$!r6rT)rfastpathrW)rrrdecompose_interprziprr interp_funcrrrrrraddrc)rrr=ruresultrBr destinationbroadcast_dims original_dimsnew_dimsinterpedout_dimsr)rs ` @r4rZrZPs4 xxzz$ZZ>>F> F*>::11N##9999D999:5)51 @???SX???&- !DQ)<$=$== CM= ) .;  (HCIMMM *||  ADyyq 1! 4 9:::: Q x==1  %V%x0F Mr6c |s|St|dkrt|fddi|\}}nt|fi|\}}t |rt |}|jt|z tttt|djzz}fdt|D}d|D}fd|D} d| D} |tg|| R} |j | \} } td t| | dd d D} | dt| dz d zzd }|dj |djd ufd t|djD} |d vo}t|jjt$js t$j}n|j}|j}|jt.|g| R|||d|| |ddSt1|||||S)a~ multi-dimensional interpolation for array-like. Interpolated axes should be located in the last position. Parameters ---------- var : np.ndarray or dask.array.Array Array to be interpolated. The final dimension is interpolated. x : a list of 1d array. Original coordinates. Should not contain NaN. new_x : a list of 1d array New coordinates. Should not contain NaN. method : string {'linear', 'nearest', 'zero', 'slinear', 'quadratic', 'cubic'} for 1-dimensional interpolation. {'linear', 'nearest'} for multidimensional interpolation **kwargs Optional keyword arguments to be passed to scipy.interpolator Returns ------- interpolated: array Interpolated array Notes ----- This requiers scipy installed. See Also -------- scipy.interpolate.interp1d rWrTrc&g|] \}}||zfgSrNrN)rr"_xnconsts r4r*zinterp_func..s)KKKb6E>+,KKKr6cg|] }|D]}| SrNrNrpairrs r4r*zinterp_func..s%AAAdDAADTAAAAr6cTg|]$}|fdt|jDg%S)cg|]}|zSrNrN)rr"rs r4r*z*interp_func...s;;;5$,;;;r6)r r)rr:rs r4r*zinterp_func..sH   ACR;;;;E"'NN;;; <   r6cg|] }|D]}| SrNrNr=s r4r*zinterp_func..s%III$DIIDIIIIr6c3$K|] }|D]}|V dSr?rN)rr>elems r4rzinterp_func..s/RRdTRRTTRRRRRRRr6NrVc<i|]}|zr|n|SrNrN)rr#r new_x0_chunksnew_x0_chunks_is_not_none new_x0_shapes r4 zinterp_func..sD    1H*CXmA&&VW   r6)rQrF)r/ interp_kwargslocalize concatenater]new_axesmeta align_arrays)rrrcrrrrrrr  enumerate unify_chunksrr.chunksr issubclassr]rErYinexactfloat__meta blockwise_chunked_aware_interpnd _interpnd)rrBrr=rur chunkmanagerout_ind x_arginds new_x_argindsargs_ rechunkedrLrJr]rMr;rrErFrGs @@@@@r4r/r/sB xxzz 1vv{{(SSDSFSS ff+F==f== f8 -c22 xAuV}}%%U4a 9M-N-N(O(OOLKKKillKKK AAiAAA     GL   JI-III U4[[=9=}==0|0$7 9RRc)T!$Q$Z&@&@RRRRR!s9~~1a7799:a Qx~ $1$=!       58=))    33R9R#)."*55 IEEIEy%|% #            S!UD& 1 11r6c|d|d}}|||fddi|tj|}|jdkr*||jdd|jzS|jdkr|dS|S)NrrqTrWr&).r&)rYravelrreshaper)rrBrrrurslts r4 _interp1drdstU1XuA 5443 5 5d 5f 5 5bhuoo F FD zA~~||CIcrcNU[8999 zQG} Kr6c t||\}}t|dkrt|||||S|t t| |jt|z }t jd|Dd}||||fi|}|t |j dzd}||j dd|dj zS)NrWc@g|]}|jSrN)rra)rx1s r4r*z_interpnd.. s$55529??$$555r6r&rr) r%rcrdrr rrYstackrbr)rrBrrrurdrcs r4rXrXs1e$$HAu 1vv{{af555 --s1vvgsx#a&&'899 : :C 55u555B ? ? ?B 43 % %f % %D >>% Q22 3 3D << 3B3%(.8 9 99r6)rJc t|dz}tj|z fdt|d|D}fd||dD}|rtdt tjDdt ||Dt \t fdD\}}jt||||S)zWrapper for `_interpnd` through `blockwise` for chunked arrays. The first half arrays in `coords` are original coordinates, the other half are destination coordinates rVcDg|]\}}td|zg|Sdim_)r)rr!r:r;s r4r*z+_chunked_aware_interpnd..s7VVV73(&3,(()2 . .VVVr6Nc g|]=}tfdtt|jD|>S)cDg|]}dtj|zSrk)rcr)rr!rs r4r*z6_chunked_aware_interpnd...s/PPPC/SY#-//PPPr6)rr rcr)rr:rs r4r*z+_chunked_aware_interpnd..sY     PPPP5RX;O;OPPPRTUU   r6cg|]}d|SrkrN)rr!s r4r*z+_chunked_aware_interpnd..#sFFF s FFFr6c4i|]\}}|jd||fS)r)r)rr:_new_xs r4rHz+_chunked_aware_interpnd..%s(SSSzr6"'!*r6lSSSr6c g|] }| SrNrNr(s r4r*z+_chunked_aware_interpnd..)sCCCq*CCCr6) rcrrOrr r.rrrX) rr/rIrJrn_xrBrrr;s ` @@r4rWrWs1 f++ C ^^c !F WVVVitPSt >U>UVVVA    ,   E  FFc#)nn0E0EFFFLLSSSE]]SSS(^<<^CCCCNCCCD5h S!UK ? ??r6c#Kd|D}g}i}t|D]\}}||g|t|dz krnO|||gz }||dzd}d|D}d|D}||s|Vg}i}|VdS)z\Decompose the interpolation into a succession of independent interpolation keeping the ordercVg|]&\}}|djdkr |djn|g'S)rWr)rr)rr!dests r4r*z$decompose_interp..4sH CQ q((Q ser6rWNch|] }|D]}| SrNrNrrr!s r4 z#decompose_interp..Cs%QQQtDQQSsQQQQr6ch|] }|D]}| SrNrNrxs r4ryz#decompose_interp..Ds%CCCdCCsCCCCr6)rrOrrc intersection) r dest_dimspartial_dest_dimspartial_indexes_coordsr# index_coords other_dimss_partial_dest_dims s_other_dimss r4r-r-1s$'--//I$^%9%9%;%;<<((<%%|n555 I" " " Eil^+q1uww' QQ/@QQQCC CCC "// == () ( ( ( " %' "      r6)rr r!rr"rr?)TT)r!rrrrr)NTrQNNN) r!rrrr=rrrrrrr)Nr&)NN)F)r=rrr)r=r)K __future__rdatetimedtrcollections.abcrr functoolsrnumbersrtypingrr r r numpyrYpandasr xarray.corer xarray.core.commonr rxarray.core.computationrxarray.core.duck_array_opsrrrxarray.core.optionsrrxarray.core.parallelcompatrrxarray.core.typesrrxarray.core.utilsrrxarray.core.variablerrxarray.core.dataarrayrxarray.core.datasetrr5r8rPrlrxrrrrrr)r-rrrrrr%rZr/rdrXrWr-rNr6r4rs""""""........999999999999IIIIIIII//////VVVVVVVVVV88888888OOOOOOOO<<<<<<<<33333333>>>>>>>>,//////++++++:        "* * * * * (* * * Z6 6 6 6 6 (6 6 6 rGGGGG)GGGD    KORRRRRn !%$PT"JJJJJZ(####..XXX7<0 0 0 0 0 f   &    / / / (====@c2c2c2c2L : : : PT@@@@@@!!!!!r6