B &[_6@sddlmZddlmZddlmZmZddlmZe dZ e dZ e dZ e dZ d d d d d d d d fd dZGdddeZd S))absolute_import) OrderedDict) exceptionsoptional_imports) graph_objsZnumpyZscipyzscipy.cluster.hierarchyz scipy.spatialbottomNcCs t|dS)Ncomplete)schlinkage)xr E/tmp/pip-install-l29rncou/plotly/plotly/figure_factory/_dendrogram.pyrc Csjtr tr tstd|j}t|dkr0td|dkr@tjj }t ||||||||d} t j | j | jdS)ac 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 ImportErrorshapelenrZ PlotlyErrordistancepdist _DendrogramrZFigurerr) X orientationlabels colorscalerrrrs dendrogramr r r create_dendrograms$D   r%c @sXeZdZdZdddejejdddddddf dd Zd d Zd d ZddZ ddZ dS)rz9Refer to FigureFactory.create_dendrogram() for docstring.rNxaxisyaxiscCs t|dS)Nr)r r )r r r r r~rz_Dendrogram.c Cs||_||_||_||_g|_g|_|jd|jdi|_|ji|jii|_|jdkr`d|j|j<n d|j|j<|jdkrd|j|j<n d|j|j<| dkrtj j } | ||| | | | \} }}}}||_||_| }| }g|_ x>tt|D].}||dkr|||j kr|j ||qWt|j t|dkrtt|j }tt|j }t||dt||t|}dd|D|_ |j ||||_| |_dS)N)leftr)rightrgcSsg|]}|qSr r ).0vr r r sz(_Dendrogram.__init__..)r r!r&r'rleavessignrrrrget_dendrogram_tracesflatten zero_valsrangerappendintminmaxsortset_figure_layout)selfrr r!r"widthheightr&r'rrrrZ dd_tracesZxvalsZyvalsordered_labelsr/Z yvals_flatZ xvals_flatiZl_borderZr_borderZcorrect_leaves_posr r r __init__ssF     z_Dendrogram.__init__c Csddddddddd }tt|d d d }|d krLddddddddg}n|}x@tt|D],}t||}|t|krb||||<qbWddddddddddg }x>|D]6\}} y|| ||<Wqtk rd|t<YqXqW|S) z Returns colorscale used for dendrogram tree clusters. :param (list) colorscale: Colors to use for the plot in rgb format. :rtype (dict): A dict of default colors mapped to the user colorscale. ZredZgreenZblueZcyanZmagentaZyellowZblackZwhite)rgbcmykwcSs|dS)Nrr )tr r r rrz,_Dendrogram.get_color_dict..)keyNzrgb(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))ZC0rC)ZC1rB)ZC2rA)ZC3rD)ZC4rE)ZC5rF)ZC6rG)ZC7rB)ZC8rA)ZC9rD) rsorteditemsr4rkeyslistKeyErrorn) r;r"dZdefault_colorsZrgb_colorscaler?rGZnew_old_color_mapZncocr r r get_color_dictsN  z_Dendrogram.get_color_dictc sddddddddd}tjdkrj}jd kr:j}|jkrNij|<fd d jDj|d <jj|d <dj|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. ZlinearZoutsideZallticksZtozeroTF)typeZticksZmirrorZ rangemodeZshowticklabelsZzerolineZshowgridZshowliner)r)r+csg|]}|jqSr )r0)r,Zzv)axis_keyr;r r r.sz/_Dendrogram.set_axis_layout..ZtickvalsZticktextarrayZtickmode)rr!r&r r'rr3update)r;rUZ axis_defaultsZaxis_key_labelsr )rUr;r set_axis_layouts&     z_Dendrogram.set_axis_layoutcCs6|jddd||d||j||j|jS)zP Sets and returns default layout object for dendrogram figure. FZclosest)Z showlegendZautosizeZ hovermoder<r=)rrWrXr&r')r;r<r=r r r r:(s   z_Dendrogram.set_figure_layoutc Cs||}||}tj||j|jd|d} t| d} t| d} t| d} t| d} ||}g}x$tt| D]}|jdkr| |}n| |}|jdkr| |}n| |}| |}d}|r||}t d t |j |j |t |j |j|d t ||d |d d }yt|j d}Wntk r@d}YnXyt|jd}Wntk rnd}YnXd||d<d||d<||q~W|| | | | 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!Zno_plotricoorddcoordZivl color_list)toprNZscatterlines)colortext)rTr rFmodemarkerr_Z hoverinfor*r r&rFr'r/)r r$r r!rrVrSr4rdictnpmultiplyr0r&r'r6 ValueErrorr5)r;rr"rrrrrQZPrYrZr>r[colorsZ trace_listr?xsZysZ color_keyZhovertext_labeltraceZx_indexZy_indexr r r r1<sZ          z!_Dendrogram.get_dendrogram_traces) __name__ __module__ __qualname____doc__rdinfr@rSrXr:r1r r r r rps  8M#r) __future__r collectionsrZplotlyrrZplotly.graph_objsrZ get_modulerdrr rr%objectrr r r r s        W