bi+ddlmZddlZddlmZddlmZddlmZddl Z ddl m Z ddl mZddlmZdd lmZdd lmZee j*dd Zd Zd ZddZddZee dZy)) annotationsN)Iterator)wraps)Number)merge)Array)tokenize)HighLevelGraph) derived_fromcVt|}t|sd|fSt|tr t|}|jj dk(rwt j|jj||}ddl }|jj||jj|jj|fSt|jj dk(r2ddl }t||j"|j$fr |j&}t j(|jt j*r|}t|j dvr|j-d}n|j/d}t j|||j-|j}|ddk(r t1|d|j1|d<||fSt j(|jt j2sd}t j||||fS)Ncategory)methodrDatetimeTZDtype)SeriesIndexi8nearest)len isinstancerlistdtypenamenp percentilecatcodespandas Categorical from_codes categoriesorderedtype__name__rrvalues issubdtype datetime64astypeviewminnumber)aqrnresultpdr$a2s P/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/array/percentile.py _percentiler2s AA q6Qw!X Gww||z!quu{{Af=~~((1C1CQWW__UWXXX AGG}!22 a"))RXX. /A }}QWWbmm, 7  2 2t$BT"Br1V4;;FLLI Q419F1Ivzz|4F1Iqy ==")) , ==Af -q 00cBddlm}|}|j||S)NrTDigest)crickr6update)r+r6ts r1_tdigest_chunkr:3s AHHQK Hr3cddlm}|}|j|tj|j |dz S)Nrr5gY@)r7r6rrarrayquantile)qsdigestsr6r9s r1_percentiles_from_tdigestr@<s6 A AGGW 88AJJrEz* ++r3c Vddlm}ddlm}ddlm}m}|jdkDrVtj||jjdk(r|jjdnd}|||fd|i|Sgd } || vrtjd t|}d |vr+tjd t|j!d }|rt#d |j%|jdk(s t'dt)|t*r|g}||||}t-|||} |j} tj.| tj0r|g| ||dz j} ||| } || vrt3d| |dk(r|dk(rtj.| tj4s$tj.| tj0rdddlm} | ddd| z}t;|j=Dcic]\}}||ft>|f}}}d| z}|dft@|tC|fi}ntjD|dd}d|d<d| z}t;|j=Dcic]\}}||f||||f}}}d| z}|dftF||gtI|jJdztC||fi}tM||}tOjP|||g }tS||tI|ff| !Scc}}wcc}}w)"uApproximate percentile of 1-D array Parameters ---------- a : Array q : array_like of float Percentile or sequence of percentiles to compute, which must be between 0 and 100 inclusive. method : {'linear', 'lower', 'higher', 'midpoint', 'nearest'}, optional The interpolation method to use when the desired percentile lies between two data points ``i < j``. Only valid for ``internal_method='dask'``. - 'linear': ``i + (j - i) * fraction``, where ``fraction`` is the fractional part of the index surrounded by ``i`` and ``j``. - 'lower': ``i``. - 'higher': ``j``. - 'nearest': ``i`` or ``j``, whichever is nearest. - 'midpoint': ``(i + j) / 2``. .. versionchanged:: 2022.1.0 This argument was previously called "interpolation" internal_method : {'default', 'dask', 'tdigest'}, optional What internal method to use. By default will use dask's internal custom algorithm (``'dask'``). If set to ``'tdigest'`` will use tdigest for floats and ints and fallback to the ``'dask'`` otherwise. .. versionchanged:: 2022.1.0 This argument was previously called “method”. interpolation : str, optional Deprecated name for the method keyword argument. .. deprecated:: 2022.1.0 See Also -------- numpy.percentile : Numpy's equivalent Percentile function r)percentile_lookup)r=) array_safemeta_from_arrayfdr)defaultdasktdigestz8The `method=` argument was renamed to `internal_method=` interpolationzFThe `interpolation=` argument to percentile was renamed to `method= ` z0percentile() got an unexpected keyword argument z+Percentiles only implemented for 1-d arrayslike)rrM?)rz"`internal_method=` must be one of rJlinear)import_requiredr7z=crick is a required dependency for using the t-digest method.zpercentile_tdigest_chunk-zpercentile_tdigest-constant)modezpercentile_chunk-z percentile-) dependencies)chunksmeta)*dask.array.dispatchrBdask.array.reductionsr=dask.array.utilsrCrDndimr true_dividerkindr"warningswarn FutureWarningpop TypeErrorkeysNotImplementedErrorrrr r%integer ValueErrorfloating dask.utilsrP enumerate __dask_keys__r:r@sortedpadmerge_percentilesrrUrr from_collectionsr)r+r,rinternal_methodkwargsr2r=rCrDallowed_internal_methodstokenrrVrPrikeydskname2dsk2calc_qgraphs r1rrEs'RE.<vvz NN1177<<33Faggll3/C P16V6v66= )) F  !&  T  O, >v{{}o N   66Q;!"OPP!V C1?1-.A Q6 "E GGE }}UBJJ'Be/!2DEKRR 1E *D6601I0J K  9$ h  ]]5"++ ."--rzz2R. T +U2;DQ__EV;W 17CT1I, ,  &- 66#;GH 1:.r "U*$AOO$56 31I S&&9 9  % AJ!3qxx{++s    T C  + +E3aS IE Ayl >>A  s !LL%c @ddlm}t|tr t |}|||}t t t|}t |}| t |\}}t |}t t t |||D cgc]\}}} | s ||| fc} }}} | s@|r tdtjt|ddz tjS| \}}}|djjdk(rct|||D cgc]} | jc} |||} ddl} | j"j%| |dj&|dj(Stj*|djtj,sd}t|t|k7st|t|k7r td g}t ||D]f\}} tj.|t| }tj0|||d|d d|d|d<|| z}|j3|htj4|}|tj4||}tj6|}tj8||}tj8||}tj:|}|||}|t=|z}|d k(rtj>|||}|Stj@||d }tj@||dd z }tjB|t|d z |tjB||}tjD||}|dk(r||}|S|dk(r||}|S|dk(rd||||zz}|S|dk(rLtjF|||z }tjF|||z }||kD}|}||||<||}|Stdcc} }}wcc} w)aCombine several percentile calculations of different data. Parameters ---------- finalq : numpy.array Percentiles to compute (must use same scale as ``qs``). qs : sequence of :class:`numpy.array`s Percentiles calculated on different sets of data. vals : sequence of :class:`numpy.array`s Resulting values associated with percentiles ``qs``. Ns : sequence of integers The number of data elements associated with each data set. method : {'linear', 'lower', 'higher', 'midpoint', 'nearest'} Specify the interpolation method to use to calculate final percentiles. For more information, see :func:`numpy.percentile`. Examples -------- >>> finalq = [10, 20, 30, 40, 50, 60, 70, 80] >>> qs = [[20, 40, 60, 80], [20, 40, 60, 80]] >>> vals = [np.array([1, 2, 3, 4]), np.array([10, 11, 12, 13])] >>> Ns = [100, 100] # Both original arrays had 100 elements >>> merge_percentiles(finalq, qs, vals, Ns=Ns) array([ 1, 2, 3, 4, 10, 11, 12, 13]) r)rCrLNzNo non-trivial arrays foundr rz3qs, vals, and Ns parameters must be the same length)shaperErOleft)sideright)outlowerhighermidpointrNzVinterpolation method can only be 'linear', 'lower', 'higher', 'midpoint', or 'nearest')$rYrCrrrmapziprerfullrnanrrrlrrrrr r!r%r* empty_likediffappend concatenateargsorttakecumsumsuminterp searchsortedminimummaximumabs)finalqr>valsrNs raise_on_nanrCr,valNLvr.r/countscount combined_valscombined_counts sort_order combined_q desired_qrvr|r~rupperlower_residualupper_residualmaskindexs r1rlrls:,&(#f V ,F c$m B :D z:b bB SBb0AGG91c1QAsA;G HIA :; ;wws2a5zA~rvv..LBb Aw}}Z'" B$/Q/\  ~~((a1C1CT!W__UU ==a 2 4yCGs2w#b'1NOOFB 1 fCF3GGJqqt45ab Q4a   e NN4(M !7mLOM*JGGM:6Mggoz:O?+J] 3FR I YYy*m <2 I/z96B IGDqH 4]+a/T: 4' 4' W u%B" I!x u%B Iz ! e,}U/CCDB Iy VVJu$5 $ABNVVJu$5 $ABN!N2DE+E$Ku%B I 5 KH0s PPP c ddlm}tj||jj dk(r|jj dnd}|||fi|S)Nr) nanquantilerFrG)rXrrr[rr\r")r+r,rors r1 nanpercentilerAsE1 qqww||s/B!'',,s+LA q! &v &&r3)rO)rOrH)rNT) __future__rr]collections.abcr functoolsrnumbersrnumpyrtlzrdask.array.corer dask.baser dask.highlevelgraphr rgr rr2r:r@rlrr3r1rsu"$!.#r}}11B ,E?Pqhb''r3