tf*ddlmZddlZddlmZddlmZddlmZddl Z ddl m Z ddl mZddlmZdd lmZdd lmZdd lmZee j"dd Zd ZdZddZddZy)) annotationsN)Iterator)wraps)Number)merge)Array) NUMPY_GE_122) percentile)tokenize)HighLevelGraphcNt|}t|sd|fSt|tr t|}|jj dk(rmt |jj||}ddl }|jj||jj|jj|fSt|jjdk(r2ddl }t||j |j"fr |j$}t'j(|jt&j*r|}t|jdvr,ddlm}|r|j1d}n#|j3d}n|j3d}t |||j1|j}|ddk(r t5|d|j5|d<||fSt'j(|jt&j6sd}t ||||fS) Ncategory)methodrDatetimeTZDtype)SeriesIndex) PANDAS_GE_200i8nearest)len isinstancerlistdtypename np_percentilecatcodespandas Categorical from_codes categoriesorderedtype__name__rrvaluesnp issubdtype datetime64dask.dataframe._compatrastypeviewminnumber) aqrnresultpdr%ra2s _/var/www/html/software/conda/envs/higlass/lib/python3.12/site-packages/dask/array/percentile.py _percentiler5s AA q6Qw!X Gww||z!quu{{Af=~~((1C1CQWW__UWXXX AGG}!22 a"))RXX. /A }}QWWbmm, 7  2 2 <]]4([[&T"Br1V4;;FLLI Q419F1Ivzz|4F1Iqy ==")) , Af -q 00cBddlm}|}|j||S)NrTDigest)crickr9update)r.r9ts r4_tdigest_chunkr=9s AHHQK Hr6cddlm}|}|j|tj|j |dz S)Nrr8gY@)r:r9rr&arrayquantile)qsdigestsr9r<s r4_percentiles_from_tdigestrCBs6 A AGGW 88AJJrEz* ++r6c ddlm}ddlm}m}gd}||vrt j dt|}d|vr1trt j dt|jd}|rtd|j|jd k(s td t|tr|g}|||| }t!|||} |j"} t%j&| t$j(r|g| || d z j"} ||| } ||vrt+d||dk(r|dk(rt%j&| t$j,s$t%j&| t$j(rdddlm} | ddd| z} t3|j5Dcic]\}}| |ft6|f}}}d| z}|dft8|t;|fi}nt%j<|d d}d|d<d| z} t3|j5Dcic]\}}| |f||||f}}}d| z}|dft>||gtA|jBdzt;||fi}tE||}tGjH|||g}tK||tA|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) array_safemeta_from_array)defaultdasktdigestzJIn Dask 2022.1.0, the `method=` argument was renamed to `internal_method=` interpolationzXIn Dask 2022.1.0, the `interpolation=` argument to percentile was renamed to `method= ` z0percentile() got an unexpected keyword argument z+Percentiles only implemented for 1-d arrayslike)rrN?)rz"`internal_method=` must be one of rJlinear)import_requiredr:z=crick is a required dependency for using the t-digest method.zpercentile_tdigest_chunk-zpercentile_tdigest-constant)modedzpercentile_chunk-z percentile-) dependencies)chunksmeta)&dask.array.dispatchrEdask.array.utilsrFrGwarningswarn FutureWarningr pop TypeErrorkeysndimNotImplementedErrorrrr rr&r'integer ValueErrorfloating dask.utilsrQ enumerate __dask_keys__r=rCsortedpadmerge_percentilesrrWrr from_collectionsr)r.r/rinternal_methodkwargsr5rFrGallowed_internal_methodstokenrrXrQrikeydskname2dsk2calc_qgraphs r4r r KsRE<= )) X  !&  MM  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 <J:.Kc >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)rFrMNzNo non-trivial arrays foundrrz3qs, vals, and Ns parameters must be the same length)shaperLrPleft)siderightlowerhighermidpointrOzVinterpolation method can only be 'linear', 'lower', 'higher', 'midpoint', or 'nearest')$rZrFrrrmapziprdr&fullrnanrrrkrrrr r!r"r'r- empty_likediffappend concatenateargsorttakecumsumsuminterp searchsortedminimummaximumabs)finalqrAvalsrNs raise_on_nanrFr/valNLvr1r2countscount combined_valscombined_counts sort_order combined_q desired_qrvr{r}r~upperlower_residualupper_residualmaskindexs r4rkrks:,&(#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/6 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 )rP)rPrH)r~NT) __future__rr[collections.abcr functoolsrnumbersrnumpyr&tlzrdask.array.corerdask.array.numpy_compatr r r dask.baser dask.highlevelgraphr r5r=rCrkr6r4rs^"$!0?.r}}#1#1L ,A?Hqr6