{hPddlmZddlmZmZddlmZddlm Z ddl m Z ejdZ ejdZejdZd Zdd Zdd Zd Zd ZdZdZdZddZdZdZdZ ddZy))Number) exceptionsoptional_importsN) graph_objs) make_subplotspandasnumpyz scipy.statsctj|t}tj|}tj |}tj |dd}tj |dd}tj |dd}||z }d|z}tj||||z k\} tj ||||zk} |||||| | d S) z6 Calculate statistics for use in violin plot. 2linear) interpolationlowerKhigher?)minmaxq1q2q3d1d2)npasarrayfloatrr percentile) dataxvals_minvals_maxrrriqr whisker_distrrs [/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/plotly/figure_factory/_violin.py calc_statsr% s 4AvvayHvvayH q"H 5B q"G 4B q"H 5B r'C9L !\)*+ ,B !\)*+ ,B c,tt|Dcgc]4}ddj||zdzdj||zdz6}}tj||dd|d|tj j d|d d d Scc}w) zG Produces a sideways probability distribution fig violin plot. z (pdf(y), y)=({:0.2f}z, )linestonextxg?spline)widthcolorshapetext) rymodenamer1fill fillcolorline hoverinfoopacity)rangelenformatrScatterscatterLine)rr2r6 linecolorir1s r$make_half_violinrB)s s1v   )**1Q40047):J:J1Q4:PPSVV D            $ $3ix $ P    s9Bc tj|| |z gt|ztjj |dddddS)z2 Returns a rugplot fig for a violin plot. z line-ew-open)r/symbolmarkersr+Fr2)r2rmarkerr3r4 showlegendr8)rr=r;r>Marker)valspdf_maxdistancer/s r$make_violin_rugplotrL@sU     8h  #d) +!!((u^(L   r&c |tjddg||gddtjjddS)z? Returns the scatterplot fig of most of a violin plot. rr+r*r rgb(0,0,0)r.r/)rr2r4r3r7)rr=r>r?)rrs r$make_non_outlier_intervalrPOsD    a& r(      $ $3l $ C  r&c tjddg||gddj|zddj|zgdtjj ddd S) z@ Makes the upper and lower quartiles for a violin plot. rzlower-quartile: r(zupper-quartile: r*rNrOr1)rr2r1r3r7r8)rr=r<r>r?)rrs r$make_quartilesrS\sq    a& r( !1!1"!5 5 !1!1"!5 5     $ $1L $ A  r&c xtjdg|gddj|zgdtddd S) z; Formats the 'median' hovertext for a violin plot. rzmedian: r(rEsquarezrgb(255,255,255))rDr/r1)rr2r1r3rFr8)rr=r<dict)rs r$ make_medianrWmsH    # $9++B// 0 8+=>  r&c Ttjj||dddddd}|S)z- Makes the x-axis for a violin plot. Fr+)titler:showgridzerolineshowlinemirrorticksshowticklabels)rlayoutXAxis) xaxis_title xaxis_rangexaxiss r$ make_XAxisre{s>    # # $ E Lr&c Ttjj|ddddddd}|S)z- Makes the y-axis for a violin plot. TrRF)rYr_ autorangeticklenr\r[rZr])rr`YAxis) yaxis_titleyaxiss r$ make_YAxisrls>    # # $ E Lr&ctj|t}t|d}t|d}t|d}t|d}t|d}t|d}t|d} tj |} tj ||d} | | } tj| } |rd | zd z nd }| |z d z | d zg}t| | | t| | | t|| t||t|g}|r|jt|| ||||fS)z? Refer to FigureFactory.create_violin() for docstring. rrrrrrrdg@ rg?)r6)rKr/)rrrr% scipy_stats gaussian_kdelinspacerrBrPrSrWappendrL)rIr6rugplotr r!rrrrrpdfxxyymax_pdfrK plot_xrange plot_datas r$ violinplotr{sM ::dE "D$&H$&H D $ B D $ B D $ B D $ B D $ B  " "4 (C Xx -B RBffRjG'.g #AH8h&,gm?M  Jr&c g} ||D]} | | vs| j| |r| j| D]} | |vstjd|j |g}t | }t d|ddd}tj|dtj}tj|dtj}g}|D]}|j||t|}t|}t| D]\}}tj|j||t }|||z ||z z }tj"|||}t%|dj'|| \}}t)j*}|D]}|j-|d|dz|d j/d j'|dzt1| ||it)j2dgdgd t5d ||d|dgd|dggdd} |j-| d||d j/dj'dt7di|d j/| ddd|| |S)zr Refer to FigureFactory.create_violin() for docstring. Returns fig for violin plot with colorscale. zRAll values/groups in the index column must be represented as a key in group_stats.r~TrFrrzrgb{}rr`rrE)sizecmincmax colorscale showscale)rr2r3rFrGrr+rr)rsrr PlotlyErrorrr;rclrs color_parser unlabel_rgbrrrrrrrfind_intermediate_colorr{r<rrrrrer=rVrl)!rrrrrrrtrrr.rYrr4grouprrrlowcolor highcolorgroup_stats_valueskey max_value min_valuerrrIintermedintermed_colorrzryr`r trace_dummys! r$violin_colorscalers*J\"$ z !   d #$   #((+  |n %B JA  QTePU C   D,<,<=H!!&)T-=-=>I4!!+c"234&'I&'I:& 2zz",,r*;7? Oi/I 4IJ55h 8T!+ GNN>:G"  ;""$ -D   T1a!e , - H    a!e $jA &L M  $$$ # # F1IF1I7   K[!Q'M)**1-z"~>?M  Jr&c  g} ||D]} | | vs| j| |r| j| D]} | |vstjd|j |g}t | }t d|ddd}t| D]\}}tj|j||t}t||||\}}tj}|D]}|j|d|dz|dj!d j#|dzt%| ||i|dj!d j#dt'd i|dj!| dd d|| |S)r}zMIf colors is a dictionary, all the group names must appear as keys in colors.r~TrFrrr`rrr+rr)rsrrrrr;rrrrrrr{rrrrr<rerl)rrrrrrrtrrr.rYrr4rrrrrrrIrzryr`rs r$ violin_dictrns*J\"$ z !   d #$   (("  |n %B JA  QTePU C:&  2zz",,r*;7?!+DF2JPW!X ;""$ -D   T1a!e , - H    a!e $jA &L M   M)**1-z"~>?M  Jr&c ~t|trtj|d} ntj|d} |^t|t rJt |dkrtjdtd|Dstjdtrbt|tjjjr4|tjd||jj}t!|| d|\} } t#j$| dt"j&j)d |d| t+d | t-d d }|dj/tddd t#j0| |}|St|tjjjstjd|tjd|dur>t| trt3|||| |||||| | }|St5|||| |||||| | }|St| trtjdt | dkrtjdt|tstjdt7|||| |||||| | }|S)a} **deprecated**, use instead the plotly.graph_objects trace :class:`plotly.graph_objects.Violin`. :param (list|array) data: accepts either a list of numerical values, a list of dictionaries all with identical keys and at least one column of numeric values, or a pandas dataframe with at least one column of numbers. :param (str) data_header: the header of the data column to be used from an inputted pandas dataframe. Not applicable if 'data' is a list of numeric values. :param (str) group_header: applicable if grouping data by a variable. 'group_header' must be set to the name of the grouping variable. :param (str|tuple|list|dict) colors: either a plotly scale name, an rgb or hex color, a color tuple, a list of colors or a dictionary. An rgb color is of the form 'rgb(x, y, z)' where x, y and z belong to the interval [0, 255] and a color tuple is a tuple of the form (a, b, c) where a, b and c belong to [0, 1]. If colors is a list, it must contain valid color types as its members. :param (bool) use_colorscale: only applicable if grouping by another variable. Will implement a colorscale based on the first 2 colors of param colors. This means colors must be a list with at least 2 colors in it (Plotly colorscales are accepted since they map to a list of two rgb colors). Default = False :param (dict) group_stats: a dictionary where each key is a unique value from the group_header column in data. Each value must be a number and will be used to color the violin plots if a colorscale is being used. :param (bool) rugplot: determines if a rugplot is draw on violin plot. Default = True :param (bool) sort: determines if violins are sorted alphabetically (True) or by input order (False). Default = False :param (float) height: the height of the violin plot. :param (float) width: the width of the violin plot. :param (str) title: the title of the violin plot. Example 1: Single Violin Plot >>> from plotly.figure_factory import create_violin >>> import plotly.graph_objs as graph_objects >>> import numpy as np >>> from scipy import stats >>> # create list of random values >>> data_list = np.random.randn(100) >>> # create violin fig >>> fig = create_violin(data_list, colors='#604d9e') >>> # plot >>> fig.show() Example 2: Multiple Violin Plots with Qualitative Coloring >>> from plotly.figure_factory import create_violin >>> import plotly.graph_objs as graph_objects >>> import numpy as np >>> import pandas as pd >>> from scipy import stats >>> # create dataframe >>> np.random.seed(619517) >>> Nr=250 >>> y = np.random.randn(Nr) >>> gr = np.random.choice(list("ABCDE"), Nr) >>> norm_params=[(0, 1.2), (0.7, 1), (-0.5, 1.4), (0.3, 1), (0.8, 0.9)] >>> for i, letter in enumerate("ABCDE"): ... y[gr == letter] *=norm_params[i][1]+ norm_params[i][0] >>> df = pd.DataFrame(dict(Score=y, Group=gr)) >>> # create violin fig >>> fig = create_violin(df, data_header='Score', group_header='Group', ... sort=True, height=600, width=1000) >>> # plot >>> fig.show() Example 3: Violin Plots with Colorscale >>> from plotly.figure_factory import create_violin >>> import plotly.graph_objs as graph_objects >>> import numpy as np >>> import pandas as pd >>> from scipy import stats >>> # create dataframe >>> np.random.seed(619517) >>> Nr=250 >>> y = np.random.randn(Nr) >>> gr = np.random.choice(list("ABCDE"), Nr) >>> norm_params=[(0, 1.2), (0.7, 1), (-0.5, 1.4), (0.3, 1), (0.8, 0.9)] >>> for i, letter in enumerate("ABCDE"): ... y[gr == letter] *=norm_params[i][1]+ norm_params[i][0] >>> df = pd.DataFrame(dict(Score=y, Group=gr)) >>> # define header params >>> data_header = 'Score' >>> group_header = 'Group' >>> # make groupby object with pandas >>> group_stats = {} >>> groupby_data = df.groupby([group_header]) >>> for group in "ABCDE": ... data_from_group = groupby_data.get_group(group)[data_header] ... # take a stat of the grouped data ... stat = np.median(data_from_group) ... # add to dictionary ... group_stats[group] = stat >>> # create violin fig >>> fig = create_violin(df, data_header='Score', group_header='Group', ... height=600, width=1000, use_colorscale=True, ... group_stats=group_stats) >>> # plot >>> fig.show() rgbrzRIf data is a list, it must be nonempty and contain either numbers or dictionaries.c3<K|]}t|tyw)N) isinstancer).0elements r$ z create_violin..PsGwz'62Gsz0If data is a list, it must contain only numbers.zVdata_header must be the column name with the desired numeric data for the violin plot.rF )rr+r) rYrfontrrGr.rdrkrrk)r\r_r^)rr`zGError. You must use a pandas DataFrame if you are using a group header.zFThe colors param cannot be a dictionary if you are using a colorscale.rzHcolors must be a list with at least 2 colors. A Plotly scale is allowed.z,Your group_stats param must be a dictionary.)rrVrvalidate_colors_dictvalidate_colorslistr;rrallpdcoreframe DataFramevaluestolistr{rrr`FontrerlrFigurerrr)rrrrrrrtrrr.rY valid_colorsrzryr`rs r$ create_violinrsV&$00? ++FE:  dD !4yA~ ,,/ G$GG ,,I *T277==#:#:;" ,,' $++224D", LOW"  ;""""''R'0R-R.   wtU5PRSTYv> $ 7 78((    ((,  U ",-!  "  *  "  ,- ,,* < 1$ ,,/ k40 ,,E$ CJr&)#1f77b4z rgb(0, 0, 0))r)rT) NNNFNTFiiXzViolin and Rug Plot)numbersrplotlyrr plotly.colorsrrplotly.graph_objsrplotly.subplotsr get_modulerrrpr%rBrLrPrSrWrerlr{rrrrr&r$rs/()   *   ))))-8 8.  " ""$"N@FdNET     Pr&