g76ddlmZddlmZmZddlmZejdZejdZ ejdZ ejdZ dd d d d d d fd Z Gd d e Zy )) OrderedDict) exceptionsoptional_imports) graph_objsnumpyscipyzscipy.cluster.hierarchyz scipy.spatialbottomNc.tj|dSNcompleteschlinkagexs ^/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/plotly/figure_factory/_dendrogram.pyrsQ 3c Ntr trts td|j}t |dk7rt jd|tjj}t||||||||} tj| j| jS)a[ Function that returns a dendrogram Plotly figure object. This is a thin wrapper around scipy.cluster.hierarchy.dendrogram. See also https://dash.plot.ly/dash-bio/clustergram. :param (ndarray) X: Matrix of observations as array of arrays :param (str) orientation: 'top', 'right', 'bottom', or 'left' :param (list) labels: List of axis category labels(observation labels) :param (list) colorscale: Optional colorscale for the dendrogram tree. Requires 8 colors to be specified, the 7th of which is ignored. With scipy>=1.5.0, the 2nd, 3rd and 6th are used twice as often as the others. Given a shorter list, the missing values are replaced with defaults and with a longer list the extra values are ignored. :param (function) distfun: Function to compute the pairwise distance from the observations :param (function) linkagefun: Function to compute the linkage matrix from the pairwise distances :param (list[list]) hovertext: List of hovertext for constituent traces of dendrogram clusters :param (double) color_threshold: Value at which the separation of clusters will be made Example 1: Simple bottom oriented dendrogram >>> from plotly.figure_factory import create_dendrogram >>> import numpy as np >>> X = np.random.rand(10,10) >>> fig = create_dendrogram(X) >>> fig.show() Example 2: Dendrogram to put on the left of the heatmap >>> from plotly.figure_factory import create_dendrogram >>> import numpy as np >>> X = np.random.rand(5,5) >>> names = ['Jack', 'Oxana', 'John', 'Chelsea', 'Mark'] >>> dendro = create_dendrogram(X, orientation='right', labels=names) >>> dendro.update_layout({'width':700, 'height':500}) # doctest: +SKIP >>> dendro.show() Example 3: Dendrogram with Pandas >>> from plotly.figure_factory import create_dendrogram >>> import numpy as np >>> import pandas as pd >>> Index= ['A','B','C','D','E','F','G','H','I','J'] >>> df = pd.DataFrame(abs(np.random.randn(10, 10)), index=Index) >>> fig = create_dendrogram(df, labels=Index) >>> fig.show() zmFigureFactory.create_dendrogram requires scipy, scipy.spatial and scipy.hierarchyz X should be 2-dimensional array.)distfun linkagefun hovertextcolor_threshold)datalayout)scpscsr ImportErrorshapelenr PlotlyErrordistancepdist _DendrogramrFigurerr) X orientationlabels colorscalerrrrs dendrograms rcreate_dendrogramr- sH c ?  A 1v{AB,,$$ ' J   *//*:K:K LLrc peZdZdZdddej ej ddddddf dZdZd Zd Z d Z y) r%z9Refer to FigureFactory.create_dendrogram() for docstring.r Nxaxisyaxisc.tj|dSr r rs rrz_Dendrogram.zsS[[J7rc  ||_||_||_||_g|_g|_|jd|jdi|_|ji|jii|_|jdvrd|j |j<nd|j |j<|jdvrd|j |j<nd|j |j<| tjj} |j||| | | | \} }}}}||_||_|j}|j}g|_ tt|D];}||dk(s |||jvs|jj!||=t|jt|dzkDryt#t%|j}t#t'|j}t||dzt#||z t|z }|Dcgc]}|c}|_ |jj)|j+|||_| |_ycc}w)N)leftr )rightr g)r(r)r/r0rleavessignrrr#r$get_dendrogram_tracesflatten zero_valsranger!appendintminmaxsortset_figure_layout)selfr'r(r)r*widthheightr/r0rrrr dd_tracesxvalsyvalsordered_labelsr7 yvals_flat xvals_flatil_borderr_bordercorrect_leaves_posvs r__init__z_Dendrogram.__init__os''     ZZDJJ2 zz2tzz26   1 1$%DIIdjj !$&DIIdjj !   2 2$%DIIdjj !$&DIIdjj ! ?ll((G<@.s QqTr)key)rgb(0,116,217)zrgb(35,205,205)zrgb(61,153,112)z rgb(40,35,35)zrgb(133,20,75)zrgb(255,65,54)zrgb(255,255,255)zrgb(255,220,0)) )C0r])C1r\)C2r[)C3r^)C4r_)C5r`)C6ra)C7r\)C8r[)C9r^rg) rsorteditemsr<r!keyslistKeyErrorn) rCr*ddefault_colorsrgb_colorscalerLranew_old_color_mapncocs rget_color_dictz_Dendrogram.get_color_dicts %VAGGI>%JK   N(Ns>..012 6A^((*+A.A3~&&$21$5q! 6  ( 5FB 5%3B%7r" 5  5%5q!  5s!B--CCc ddddddddd}t|jdk7r|j}|jd vr |j}||j vri|j |<|j Dcgc]}||j|zc}|j |d <|j|j |d <d |j |d <|j |j||j |Scc}w)z Sets and returns default axis object for dendrogram figure. :param (str) axis_key: E.g., 'xaxis', 'xaxis1', 'yaxis', yaxis1', etc. :rtype (dict): An axis_key dictionary with set parameters. linearoutsidealltickstozeroTF)typeticksmirror rangemodeshowticklabelszerolineshowgridshowliner)r4r6tickvalsticktextarraytickmode) r!r)r/r(r0rr;r8update)rCaxis_key axis_defaultsaxis_key_labelszvs rset_axis_layoutz_Dendrogram.set_axis_layouts !"   t{{ q "jjO#44"&**dkk1/1 O,37>>8-/TYYx((8DKK ( 48<{{DKK ( 47>DKK ( 4 H$$]3{{8$$8s5C<c|jjddd||d|j|j|j|j|jS)zP Sets and returns default layout object for dendrogram figure. Fclosest) showlegendautosize hovermoderDrE)rrrr/r0)rCrDrEs rrBz_Dendrogram.set_figure_layout$sZ #!&    TZZ( TZZ({{rc ||}||}tj||j|jd|} tj | d} tj | d} tj | d} tj | d} |j |}g}tt| D] }|jdvr| |}n| |}|jdvr| |}n| |}| |}d}|r||}td tj|j|j|tj|j|j|d t|| |d } t|jd} t|jd}d||d<d||d<|j!||| | | | dfS#t$rd}YTwxYw#t$rd}YLwxYw)ao Calculates all the elements needed for plotting a dendrogram. :param (ndarray) X: Matrix of observations as array of arrays :param (list) colorscale: Color scale for dendrogram tree clusters :param (function) distfun: Function to compute the pairwise distance from the observations :param (function) linkagefun: Function to compute the linkage matrix from the pairwise distances :param (list) hovertext: List of hovertext for constituent traces of dendrogram :rtype (tuple): Contains all the traces in the following order: (a) trace_list: List of Plotly trace objects for dendrogram tree (b) icoord: All X points of the dendrogram tree as array of arrays with length 4 (c) dcoord: All Y points of the dendrogram tree as array of arrays with length 4 (d) ordered_labels: leaf labels in the order they are going to appear on the plot (e) P['leaves']: left-to-right traversal of the leaves T)r(r)no_plotricoorddcoordivl color_list)topr Nscatterlines)colortext)rrr`modemarkerr hoverinfor5rr/r`r0r7)rr,r(r)nprr~r<r!dictmultiplyr8r/r0r> ValueErrorr=)rCr'r*rrrrrxZPrrrIrcolors trace_listrLxsys color_keyhovertext_labeltracex_indexy_indexs rr9z!_Dendrogram.get_dendrogram_traces8s0 AJ qM NN ((;;+  !H+&!H+&!E(+XXa o. $$Z0 s6{#' %A#44AYAY#44AYAY"1 I"O"+A,++dii 3R8++dii 3R8&"34$ E djjn- djjn- ! ]E'N  ]E'N   e $O' %R66>1X;FF     s$GG0 G-,G-0 G>=G>) __name__ __module__ __qualname____doc__rinfrQr~rrBr9rdrrr%r%lsPC ffvv7CJKZ!%F(SGrr%) collectionsrplotlyrrplotly.graph_objsr get_modulerrrrr-objectr%rdrrrs#/(!  )!!!'*!!!";<!!!/2   3\M~_G&_Gr