cQPddlmZddlmZddlmZmZddlmZ ddl m Z ddl m Z ejdZejdZejd Zd Zdd ZddZdZdZdZdZdZd dZdZdZdZ d!dZdS)")absolute_import)Number) exceptionsoptional_importsN) graph_objs) make_subplotspandasnumpyz scipy.statsc&t|tj}t|}t|}t|dd}t|dd}t|dd}||z }d|z}t||||z k} t||||zk} |||||| | d S) z6 Calculate statistics for use in violin plot. 2linear) interpolationlowerKhigher?)minmaxq1q2q3d1d2)npasarrayfloatrr percentile) dataxvals_minvals_maxrrriqr whisker_distrrs =lib/python3.11/site-packages/plotly/figure_factory/_violin.py calc_statsr&s 4""AvvayyHvvayyH q"H 5 5B q"G 4 4B q"H 5 5B r'C9L !\)*+ , ,B !\)*+ , ,B  #1f77b4 rgb(0, 0, 0)cfdttD}tjdd|d|tjd|ddd S) zG Produces a sideways probability distribution fig violin plot. cg|]B}dd|zdzd|zdzCS)z (pdf(y), y)=({:0.2f}z, ))format).0ir ys r% z$make_half_violin../sb     )**1Q400047):J:J1Q4:P:PPSVV   r'linestonextxg?spline)widthcolorshapetext) r r1modenamer:fill fillcolorline hoverinfoopacity)rangelenrScatterscatterLine)r r1r> linecolorr:s`` r%make_half_violinrH+s     s1vv   D         $ $3ix $ P P    r'c tj|| |z gt|ztj|dddddS)z2 Returns a rugplot fig for a violin plot. z line-ew-open)r8symbolmarkersr4Fr1)r1r markerr;r< showlegendr@)rrDrCrEMarker)valspdf_maxdistancer8s r%make_violin_rugplotrRBs^    8h  #d)) +!((u^(LL     r'c ztjddg||gddtjddS)z? Returns the scatterplot fig of most of a violin plot. rr4r3r rgb(0,0,0)r7r8)r r1r<r;r?)rrDrErF)rrs r%make_non_outlier_intervalrVQsK   a& r(     $ $3l $ C C    r'c tjddg||gdd|zdd|zgdtjddd S) z@ Makes the upper and lower quartiles for a violin plot. rzlower-quartile: r,zupper-quartile: r3rTrUr:)r r1r:r;r?r@)rrDr.rErF)rrs r%make_quartilesrY^s~   a& r( !1!1"!5!5 5 !1!1"!5!5 5    $ $1L $ A A    r'c tjdg|gdd|zgdtddd S) z; Formats the 'median' hovertext for a violin plot. rzmedian: r,rKsquarezrgb(255,255,255))rJr8r:)r r1r:r;rLr@)rrDr.dict)rs r% make_medianr]osX   # $9++B/// 0 8+=>>>    r'c Ttj||dddddd}|S)z- Makes the x-axis for a violin plot. Fr4)titlerBshowgridzerolineshowlinemirrorticksshowticklabels)rlayoutXAxis) xaxis_title xaxis_rangexaxiss r% make_XAxisrk}s@   # # $  E Lr'c Ttj|ddddddd}|S)z- Makes the y-axis for a violin plot. TrXF)r_re autorangeticklenrbrar`rc)rrfYAxis) yaxis_titleyaxiss r% make_YAxisrrs@   # # $  E Lr'TcVt|tj}t|d}t|d}t|d}t|d}t|d}t|d}t|d} t|} t||d} | | } t| } |rd | zd z nd }| |z d z | d zg}t| | | t| | | t|| t||t|g}|r&| t|| ||||fS)z? Refer to FigureFactory.create_violin() for docstring. rrrrrrrdg@ rg?)r>)rQr8)rrrr& scipy_stats gaussian_kdelinspacerrHrVrYr]appendrR)rOr>rugplotr!r"rrrrrpdfxxyymax_pdfrQ plot_xrange plot_datas r% violinplotrs ::dBH % %D$&H$&H D  $ B D  $ B D  $ B D  $ B D  $ B  " "4 ( (C Xx - -B RBffRjjG'.5g ##AH8h&,gmrzrfxaxis{}yaxis{}r4closestr_rM hovermodeautosizeheightr7)rysortgroupbyrCr enumeraterr get_grouprrrLayout append_traceupdater.rkrr)r data_header group_headercolorsuse_colorscale group_statsrzrrr7r_ group_namer<gbLfig color_indexkgrrOrrrfitems r%violin_no_colorscalers*J\"$$ z ! $   d # # #  |n % %B JA  QTePU   CK:&&  2zz",,r**;7BB #f++ % K!+ F;/" " "  ;"$$ - -D   T1a!e , , , ,q  H    a!e $ $jA &L&L M    M)**1--z"~~>???M  Jr'c g} ||D]} | | vr| | |r| | D]} | |vrtjd||g}t | }t d|ddd}tj|dtj }tj|dtj }g}|D]}|||t|}t|}t| D] \}}t|||tj}|||z ||z z }tj|||}t%|d|| \}}t)j}|D]}||d|dz|d d |dzt1| ||it)jdgdgd t5d ||d|dgd|dggdd} || d||d ddt7di|d | 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.rTrFrrzrgb{}rrfrrK)sizecmincmax colorscale showscale)r r1r;rLrMrr4rr)ryrr PlotlyErrorrrCrclrs color_parser unlabel_rgbrrrrrrrfind_intermediate_colorrr.rrrrrkrDr\rr)!rrrrrrrzrrr7r_rr<grouprrrlowcolor highcolorgroup_stats_valueskey max_value min_valuerrrOintermedintermed_colorrrrfr trace_dummys! r%violin_colorscaler sa*J\"$$ z ! $   d # # #   # (+   |n % %B JA  QTePU   C  D,<==H!&)T-=>>I44!!+c"23333&''I&''I:&&  2zz",,r**;7BB Oi/I 4IJ5h 8TT!+ GNN>::G" " "  ;"$$ - -D   T1a!e , , , , H    a!e $ $jA &L&L M    $ # # F1IF1I7       K[!Q'''M)**1--z"~~>???M  Jr'c g} ||D]} | | vr| | |r| | D]} | |vrtjd||g}t | }t d|ddd}t| D]\}}t | ||tj }t||||\}}tj}|D]}||d|dz|dd |dzt%| ||i|dd dt'd i|d| dd d|| |S)rzMIf colors is a dictionary, all the group names must appear as keys in colors.rTrFrrrfrrr4rr)ryrrrrrCrrrrrrrrrrrr.rkrr)rrrrrrrzrrr7r_rr<rrrrrrrOrrrfrs r% violin_dictrps)*J\"$$ z ! $   d # # #    ("   |n % %B JA  QTePU   C:&&   2zz",,r**;7BB!+DF2JPW!X!X!X ;"$$ - -D   T1a!e , , , , H    a!e $ $jA &L&L M    M)**1--z"~~>???M  Jr'FXViolin and Rug Plotc t|trtj|d} ntj|d} |}t|t rTt |dkrtjdtd|DstjdtrYt|tj j j r5|tjd||j}t!|| d|\} } t#j| d t"jd |d | t+d | t-d d  }|dtd d d t#j| |}|St|tj j j stjd|tjd|d urKt| 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() rgbNrzRIf data is a list, it must be nonempty and contain either numbers or dictionaries.c3@K|]}t|tVdS)N) isinstancer)r/elements r% z create_violin..Rs,GGwz'622GGGGGGr'z0If 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 )rr4r) r_rfontrrMr7rjrqrrq)rbrerd)rrfzGError. 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.)rr\rvalidate_colors_dictvalidate_colorslistrCrrallpdcoreframe DataFramevaluestolistrrrrfFontrkrrrFigurerrr)rrrrrrrzrrr7r_ valid_colorsrrrfrs r% create_violinrsV&$;0?? +FE::  dD ! ! 4yyA~  ,/ GG$GGGGG  ,I  5*T27=#:;; 5  ,' $+2244D", LOW" " "  ;""''R'00R--R..     wtU5PRSSSTTTYv>>> $ 788 (    (,  U "A ,-- !  "   *  "   ,--  ,* <  1$  ,/ k400  ,E$  CJr')r(r))r()r(T) NNNFNTFrrr) __future__rnumbersrplotlyrr plotly.colorsrrplotly.graph_objsrplotly.subplotsr get_modulerrrvr&rHrRrVrYr]rkrrrrrrrr'r%rs&&&&&&////////(((((())))))  **  ))))-88 8.       "   ""$"$"$"$"N@@@FdddNEEET     PPPPPPr'