c|6ddlmZddlmZddlmZmZddlmZej dZ ej dZ ej dZ ej dZ d d d d d d d fd ZGd deZd S))absolute_import) OrderedDict) exceptionsoptional_imports) graph_objsnumpyscipyzscipy.cluster.hierarchyz scipy.spatialbottomNc8t|dSNcompleteschlinkagexs Alib/python3.11/site-packages/plotly/figure_factory/_dendrogram.pyrsQ 33c :trtrtstd|j}t |dkrt 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.N)distfun linkagefun hovertextcolor_threshold)datalayout)scpscsr ImportErrorshapelenr PlotlyErrordistancepdist _DendrogramrFigurerr) X orientationlabels colorscalerrrrs dendrograms rcreate_dendrogramr.sH  c   ?   A 1vv{CABBB%,$ '   J  */*:K L L LLrc ^eZdZdZdddejejddddddf dZdZd Zd Z d Z dS) r&z9Refer to FigureFactory.create_dendrogram() for docstring.r Nxaxisyaxisc8t|dSr rrs rrz_Dendrogram.~sS[[J77rc ||_||_||_||_g|_g|_|jd|jdi|_|ji|jii|_|jdvrd|j|j<nd|j|j<|jdvrd|j|j<nd|j|j<| tj j } | ||| | | | \} }}}}||_||_| }| }g|_ tt|D]=}||dkr/|||j vr |j ||>t|j t|dzkrt#t%|j }t#t'|j }t||dzt#||z t|z }d|D|_ |j ||||_| |_dS)N)leftr )rightr gcg|]}|Sr9).0vs r z(_Dendrogram.__init__..s<<.s QqTr)keyN)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)) )C0rb)C1ra)C2r`)C3rc)C4rd)C5re)C6rf)C7ra)C8r`)C9rcrk) rsorteditemsrBr"keyslistKeyErrorn) rIr+ddefault_colorsrgb_colorscalerRrfnew_old_color_mapncocs rget_color_dictz_Dendrogram.get_color_dictsS %VAGGII>>%J%J%JKK  (   NN(Ns>..001122 6 6A^((**++A.A3~&&& 6$21$5q!    ( 5 5FB 5%3B%7r"" 5 5 5%5q!!!  5 s CC+*C+c ddddddddd}tjdkrqj}jd vrj}|jvr ij|<fd jDj|d <jj|d <d j|d<j|jS)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)r5r7c0g|]}|jzSr9)r>)r:zvaxis_keyrIs rr<z/_Dendrogram.set_axis_layout..s2888-/TYx((888rtickvalsticktextarraytickmode)r"r*r0r)r1rrAupdate)rIr axis_defaultsaxis_key_labelss`` rset_axis_layoutz_Dendrogram.set_axis_layouts !"    t{  q ?"jO#44 -"&*dk1 2/1 O,8888837>888DK ( 48<{DK ( 47>DK ( 4 H$$]333{8$$rc|jddd||d||j||j|jS)zP Sets and returns default layout object for dendrogram figure. Fclosest) showlegendautosize hovermoderJrK)rrrr0r1)rIrJrKs rrHz_Dendrogram.set_figure_layout(sm #!&       TZ((( TZ((({rc ||}||}t||j|jd|} t| d} t| d} t| d} t| d} ||}g}tt| D]H}|jdvr | |}n| |}|jdvr | |}n| |}| |}d}|r||}td t |j |j |t |j |j|d t|| |d } t|j d}n#t $rd}YnwxYw t|jd}n#t $rd}YnwxYwd|z|d<d|z|d<||J|| | | | dfS)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)rrremodemarkerr hoverinfor6rr0rer1r=)rr-r)r*rrrrBr"dictnpmultiplyr>r0r1rD ValueErrorrC)rIr(r+rrrrr|ZPrrrOrcolors trace_listrRxsys color_keyhovertext_labeltracex_indexy_indexs rr?z!_Dendrogram.get_dendrogram_traces<sk0 GAJJ JqMM NN (;+   1X;''1X;''1U8,,YYq// $$Z00 s6{{##' %' %A#44 AYAY#44 AYAY"1 I"O /"+A,++di 3R88++di 3R88&"3444$ E djn--     djn--    !7]E'N 7]E'N   e $ $ $ $66>1X;FFs$;G G%$G%)H HH) __name__ __module__ __qualname____doc__rinfrVrrrHr?r9rrr&r&psCC fv77CCCCJKKKZ!%!%!%F(SGSGSGSGSGrr&) __future__r collectionsrplotlyrrplotly.graph_objsr get_modulerrrrr.objectr&r9rrrs#'&&&&&######////////((((((! ))!!'**!!";<<!!/22   33\M\M\M\M~_G_G_G_G_G&_G_G_G_G_Gr