a ~b^@s`ddlmZmZmZddlmZddlmZddlZddl Z ddl m Z m Z mZmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZmZmZmZmZm Z dd l!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z.m/Z/m0Z0ddl1m2Z2m3Z3m4Z4Gddde&Z5Gddde,Z6Gddde(e)e*Z7Gddde7Z8dS))absolute_importdivisionunicode_literals) defaultdict)partialN) FactorRangeCircleVBarHBar)BokehOverlaySelectionDisplay) NdOverlay) Dimension Dimensioned)sorted_context) basestringdimension_sanitizer wrap_tupleunique_iteratorisfinite is_dask_array is_cupy_array)univariate_kde)dim)AreaPlot)CompositeElementPlot ColorbarPlot LegendPlot) PolygonPlot)base_propertiesfill_propertiesline_properties) LooseVersion bokeh_version decode_bytesc@s@eZdZdZejdddZejdddZejdddZ e Z dS) DistributionPlotzH DistributionPlot visualizes a distribution of values as a KDE. N> The bandwidth of the kernel for the density estimate.defaultdocr D Draw the estimate to cut * bw from the extreme data points.T9 Whether the bivariate contours should be filled.) __name__ __module__ __qualname____doc__paramNumber bandwidthcutBooleanfilledr selection_displayr8r8=lib/python3.9/site-packages/holoviews/plotting/bokeh/stats.pyr&s r&c@sZeZdZdZejdddZejdddZejdddZ ej d e e fd d Z ed d dZdS) BivariatePlotz Bivariate plot visualizes two-dimensional kernel density estimates. Additionally, by enabling the joint option, the marginals distributions can be plotted alongside each axis (does not animate or compose). Nr'r(r r+Fr, zD A list of scalar values used to specify the contour levels.r)class_r*cmapT) color_propZis_cmap)r-r.r/r0r1r2r3r4r5r6 ClassSelectorlistintZlevelsr r7r8r8r8r9r:-s r:cseZdZejdddZejdeefdddZ e Z dd d d d Z d d e eDdd e eeDdd eeDgdZe gdZdZeddZd%fdd ZddZd&ddZd'fdd ZddZfdd Zd!d"Zd#d$ZZS)(BoxWhiskerPlotFz- Whether to show legend for the plot.r(NTzR Deprecated in favor of color style mapping, e.g. `box_color=dim('color')`r)r=Z allow_Noner*ZwhiskerboxZoutlier)segmentvbarhbarcirclecCsg|] }d|qS)Zwhisker_r8.0pr8r8r9 UzBoxWhiskerPlot.cCsg|] }d|qS)box_r8rJr8r8r9rMVrNcCsg|] }d|qS)Zoutlier_r8rJr8r8r9rMWrN) box_widthr>box_cmap)rP whisker_widthr>rQZ box_colorr?combinedcstt||||d|jdS)NZ categoricalr)superrC get_extentsvdims)selfelementrangesZ range_type __class__r8r9rV`s zBoxWhiskerPlot.get_extentscCs|j|jdfS)Nr)kdimsrW)rXrYr8r8r9_get_axis_dimseszBoxWhiskerPlot._get_axis_dimsc Cs@t||d}|jr<|js<|jr.r cs&g|]}tddtj|DqS)css|]\}}||VqdSN pprint_valuerKdkr8r8r9 rNz9BoxWhiskerPlot._get_factors...)tuplezipr]rqrYr8r9rMscSs$g|]}t|dkr|dn|qSr rlenrKfr8r8r9rMrN)r]rdgroupbyrokeysrjsorted invert_axesrXrYrZZxfactorsZyfactorsfactorsr8r|r9 _get_factorszs   zBoxWhiskerPlot._get_factorscs,t|jtttfsdStt|||dSrs)rlglyphrr r rUrC_postprocess_hover)rXZrendererr_r[r8r9rsz!BoxWhiskerPlot._postprocess_hovercsPt}t}t}|r.ddl}|j|j}n|rFddlm}|jntj||sdtṙfddt dddD\}}} | |} t | d| k | } t |d| k |} nd\}}} d\} } | k| kB} |r"| | | | | | | fS|r<|||| | | | S||| | | | fSdS) Nrc3s|]}|dVqdS)qNr8rKr percentilevalsr8r9ryrNz,BoxWhiskerPlot._box_stats..dg?)rrr)rr)rrrcupyrZ dask.arrayarrayrkrrangemaxminitemZasnumpycompute)rXrZ is_finiteZis_daskZis_cupyrdaq1q2q3Ziqrupperloweroutliersr8rr9 _box_statss6   "  zBoxWhiskerPlot._box_statsc*Cs|jr>td||jj}WdqN1s20Ynt|j|fg}t|jdj}ddt dD\}}ddt dD\}} ddt dD\} } t t dg|gi} | d d } | d d d }d|vr|jd|jrddd| d}ddddd}d|d}n&ddd| d}ddddd}d|d}t|}|jdurf||j}||j}nd\}}g}|jdj}|D]\}}|jrtddt|j|D}t|dkr|d}n|}d|jv}|dur||jkr||t||n |||jj||dd }||\}}}}}} ||fD]}!|!d|qB|| fD]"}!|!d||!d|q`| | fD]8}!|!dt|| f|!dt||fq|d||d||d||d||d|| d||d|| d|| d|| d|| d|| d|t| r| d|gt| 7<| |t | 7<|rt|jt|D]*\}"}#| t|"j|#gt| 7<q|rt|jt|D]\\}"}#t|"j}$|$|vr*||$|#n |#g||$<|$|vrN||$|#n |#g||$<q||vrv|||n |g||<||vr|||n |g||<q|jrd!nd"}%|%d#||%d$|d%|d&| d'| d(| d)| i}!|%d#||%d$|d%|d&|d'|d(|d)|i}&|||| | fD](}'|'D]\}#}(t !|(|'|#<q"q|jsR|!|&|fS|dush||jkrrt"d}n(||t|j<||t|j<t t#|}|j$rt%d*krd+nd,})|j||)<|!|&|fS)-NFrcss|]}gggdVqdS))indextopbottomNr8rKir8r8r9ryrNz*BoxWhiskerPlot.get_data..css|]}ggggdVqdS)x0y0x1y1Nr8rr8r8r9ryrNcss|]}ggggdVqdSrr8rr8r8r9ryrNrrPgffffff?rRg?g@widthz>BoxWhisker width option is deprecated use 'box_width' instead.rryleftrightZheightrrrrrrrr)rx)rrrrrrrrrr)NNcss|]\}}||VqdSrsrt)rKrwvr8r8r9ryrNr hover)rrHrG_1Z_2 segment_1Z segment_2Z segment_3Z segment_4Zcircle_1r` legend_fieldra)&r]rrrorbrdrrWnamerrrAgetpopr1Zwarningr color_indexZ get_dimensionZget_dimension_indexitemsrzr{rZhandlesrjappendrurZ interfacevaluesrrkrrrrcr$)*rXrYrZregroupsvdimZr1_dataZr2_dataZs1_dataZs2_dataZw1_dataZw2_dataZout_datarrRZvbar_mapseg_mapZout_mapZ vbar2_mapZcdimZcidxrrrgrdrrrrrrrrrokdrxZkd_name bar_glyphmappingZgdatarrgr8r8r9get_datas .              $             zBoxWhiskerPlot.get_data)rT)N)N)r-r.r/r1r5rcr@rrBrrZ _x_range_type _style_groupsr r"r! style_optsZ_nonvectorized_stylesZ _stream_datar r7rVr^rhrnrrrr __classcell__r8r8r[r9rCDs0      "rCc@seZdZejdddZejddddZejdddZej gd d d d Z ej de e fd dZejdddZej deefdddZdddd dd dZgdZddeeeDddeeDdddDgdZd dd!DZed"d#Zd$d%Zd&d'Zd(d)Zd*d+ZdS), ViolinPlotNzl Allows supplying explicit bandwidth value rather than relying on scott or silverman method.r(rzB A tuple of a lower and upper bound to clip the violin at.)r)lengthr*r+)rE quartilesstickNrEz Inner visual indicator for distribution values: * box - A small box plot * stick - Lines indicating each sample value * quartiles - Indicates first, second and third quartiles )Zobjectsr)r*z- The dimension to split the Violin on.r<g?z$ Relative width of the violinTzU Deprecated in favor of color style mapping, e.g. `violin_color=dim('color')`rDviolinoutlinestatsmedian)patches multi_linerFrGscatterrH)rrrFrGrHrIrcCsg|]}dD] }||q qS))Zviolin_rOr8rKrLrr8r8r9rMiszViolinPlot.cCsg|]}dD] }||q qS))Zstats_Zoutline_r8rr8r8r9rMkscCs$g|]}dD]}d||gq qS))rErrr_)joinrr8r8r9rMms)ZcolorZalpha)r>rQZ violin_cmapcCsg|]}ttj|dqS)r)rrkrrr8r8r9rMqrN)r2Kviolin_fill_colorrScs>t|jtrt|jn|jfdd|jD}||jdfS)Ncsg|]}r|jkr|qSr8 dimensionrKr split_dimr8r9rMwrNz-ViolinPlot._get_axis_dims..r)rlsplitrrr]rW)rXrYr]r8rr9r^uszViolinPlot._get_axis_dimscst|jtrt|jn|jfdd|jDsD|jgg}}nVdd|jD}|j dkrpt |}fdd|D}dd|D}|g}}|j r||fS||fS)rpcsg|]}r|jkr|qSr8rrrr8r9rMrNz+ViolinPlot._get_factors..cSsg|]}|qSr8r8rqr8r8r9rMrNr cs$g|]}tddt|DqS)css|]\}}||VqdSrsrtrvr8r8r9ryrNz5ViolinPlot._get_factors...)rzr{rqr]r8r9rMscSs$g|]}t|dkr|dn|qSr}r~rr8r8r9rMrN) rlrrrr]rdrrorrjrrrr8)r]rr9rzs   zViolinPlot._get_factorsc& s|j}|d}jr.|jd}|j|gd}|dur|j|jd}||} t|dkrttdt|d|| t |t|j| }t |f|j t |d|} d n&ddg}dt |fd |j i|i} d| |} ggggf\} } }t|D](\}}| |dur0tgtg}}nfd d t dD\}}t||dk@}||||}}|dkr|d 9}n|ddd }|ddd }|rt|r| | dgt|| d g|dgt|dgn| g|g| ddd } | t|7} t|7qt| } t|rfdd| D} fdd|D}trttjngfddD| d}|r| |dn|t| trdd|Dt |<tttti}}}||}|t|}t|s"njdkrt| rjD]j}||}tt| |}| ||}}|d||d|d  f|d|q>njdkr.t| r|D]b}tt| |}| ||}}|d||d|d  f|d|qnjdkrd}|\} }!}"}#}$}%|d||d|$|d|#|d||d| |d|"||d<|!|d<||||fS)Nr)r)rWrrzcThe number of categories for split violin plots cannot be greater than 2. Found {0} categories: {1}z, )rrrc3s|]}|VqdSrs)dimension_values)rKZidim)kder8r9ryrNz'ViolinPlot._kde_data..cSsg|]}t|qSr8)rkZasarray)rKrr8r8r9rMrNz(ViolinPlot._kde_data..csBg|]:}t|r:fdd|tjDngqS)csg|]}|fqSr8r8rKrrrr8r9rMrNz3ViolinPlot._kde_data...)rrkabsr violin_width)rKZfy)rrscalerXysr8r9rMs,csg|]}|fqSr8r8rrr8r9rMrN)rxscSsg|] }t|qSr8str)rKrxr8r8r9rMrNrrrrrrErrrr) rWcliprmr]applyr ValueErrorformatrZ add_dimensionreprrrr enumeraterrkrrrrAZnanmaxrrrlrrrinner _stat_fnsZargminr)&rXrYelrrr split_catskwargsrWrZall_catsZkdesZx_rangerZfill_xsZfill_ysrcatZ_xsZ_ysmasklinebarssegmentsrrZstat_fnstatZsidxZsxZsyvalueZxposrrrrrrr8)rrrrrXrr9 _kde_datas         "     ,            zViolinPlot._kde_datac# s4t|jtrt|jn|jfdd|jD}|rjtd||j}Wdq|1s^0Ynt|j f|fg}rʈj j }||vrj sd||vr|| d}qΈrtt|}nd}|jr(ddddd }d d d } |jd kr ddddd} nddddd} ddd} d} nPddddd}d d d } |jd kr\ddddd} nddddd} ddd} d} |j|jd} ii}}ddtdD\}}}}}t|D],\}\}}t|}|jrtddt|j|D}|j||||fi| \}}}}}|D]\}}|||7<q|D]\}}|||7<q@|D]\}}|||qd|D]\}}|||q|D]:\}}r|D]}|||qn|||qq||d<| |d<||d<| |d<|r,dd|D|d <| |d <|rXd!d|D|| d"<||| d"<|r|d#d|D|d$<| |d$<rd%d|D} |j||t|d&d'| d(}!t|!d)|d*<|jrt t!d+krd,nd-}"t| |"<t|D],\}}|"d.r|#|||$d'd/<qd0|d1<|||fS)2Ncsg|]}rj|kr|qSr8rrrr8r9rMrNz'ViolinPlot.get_data..Frrrrg?rrr)rrrErrrrrrH)rrrrrrG)r3r4css|]}ttVqdSrs)rrArr8r8r9ry$sz&ViolinPlot.get_data..rcss|]\}}||VqdSrsrtrvr8r8r9ry*rNZ multi_line_1Z patches_1cSs.i|]&\}}|t|dtr |nt|qSrrlrzrkrrKrxrr8r8r9 Bsz'ViolinPlot.get_data..rcSs.i|]&\}}|t|dtr |nt|qSrrrr8r8r9r FsrcSs.i|]&\}}|t|dtr |nt|qSrrrr8r8r9r JsZ scatter_1cSsg|] }t|qSr8r)rKrr8r8r9rMOrNZviolin_color_mapperr)rrfr)ZfieldZ transformrr`rraZ violin_linerrZviolin_line_width)%rlrrrr]rrrorbrdrropsrrArrrrr3r4rrrr%rzr{rrZ_get_colormapperrrcr$r# startswithrreplace)#rXrYrZrer]rZ split_namerZbar_mapZkde_maprZ scatter_maprrrroZkde_dataZ line_dataZseg_dataZbar_dataZ scatter_datarrrrrrZsegsrrrxrrrZcmapperrgr8rr9rs ,             zViolinPlot.get_data) r-r.r/r1r2r3Z NumericTuplerr4ZObjectSelectorrr@rrrrrrBrrZ _draw_orderr r!r"rrr r7r^rrrr8r8r8r9rAs:     mr)9Z __future__rrr collectionsr functoolsrr1ZnumpyrkZ bokeh.modelsrrr r Z selectionr corerZcore.dimensionrrZcore.ndmappingrZ core.utilrrrrrrrZoperation.statsrZutil.transformrZchartrrYrrrpathrZstylesr r!r"utilr#r$r%r&r:rCrr8r8r8r9s,     $    ~