U SZf@sddlmZmZddlmZddlmZddlm Z ddl Z ddl m Z e dZdZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdddddgZdZddZddZddZ d3d!d"Z!d#d$Z"d4d&d'Z#d(d)Z$d*d+Z%d,d-Z&d.d/Z'd5d1d2Z(dS)6) exceptionsoptional_importsN)utils) make_subplots)NumberZpandasz#969696z#0f0f0f z#ffffffz#efefefz#edededz#d0d0d0?? scatter scattergl histogrambarZboxzIf you are using a dictionary for custom labels for the facet row/col, make sure each key in that column of the dataframe is in your facet labels. The keys you need are {}cCs|tkrd}nd}|S)NTF)THRES_FOR_FLIPPED_FACET_TITLES)numflippedrE/tmp/pip-unpacked-wheel-5ksk5baj/plotly/figure_factory/_facet_grid.py _is_flipped"srcCs4t|tr||}nt|tr,d||}n|}|S)Nz{}: {}) isinstancedictstrformat)Zoriginal_labelZ facet_labelsZ facet_varlabelrrr _return_label*s    rcCs.tdddtddddd|tdd d d }|S) Nrleftmiddle{Gz?Fpaperz factor({}) #000000sizecolor textanglexanchoryanchorxy showarrowxrefyreftextfont)rLEGEND_ANNOT_Xr) color_nameZ legend_titlerrr_legend_annotation4s  r3colTc Csd|d||}|std}d}|dkrJ|d||d|} d} d} q|dkr|d||d|} d} d } nb|dkrd}d }|d||d|} d } d } n0|dkrd }d}|d||d|} d } d} t| ||| | dddt|tdtdd } | S)Nrcenterrr4r rrrowZZbottom?rFr r!r#r&)rrAXIS_TITLE_COLOR) r/ZlaneZ num_of_lanesSUBPLOT_SPACINGrow_colrlr(r)r*r+r'Zannotation_dictrrr_annotation_dictDsJ  r>c CsV|dkrd}d}d}n|dkr*d}d}d}|s2d}dtd d |||d d |d d d }|S)Nr*r grr+r9r"r%r$Fr5r r) r0r,r/r'r*r(r-r+r)r.)AXIS_TITLE_SIZE)r/Z x_or_y_axisZx_posZy_posr'Zannotrrr_axis_title_annotationps, rBFcCs4g}|dD] }d|kr|d|dddgkr|d|ddd d d d }d |kr|d|dd |d<|d|dd|d<d|d<d|d<|r|dd7<||qd|krd|d<d|d<|d|dd |d<|d|dd|d<|r|dd7<||q||dd<dS)NlayoutZaxisdomaingr8Zbelowrr%widthZrectr )Z fillcolorZlayerlinetyper-r.xaxisrZx0x1gGz?Zy0r y1r yaxisZshapes)Zto_plotly_jsonkeysappend)figZannot_rect_color flipped_rows flipped_colsZ shapes_listkeyshaperrr_add_shapes_to_figs6    rTcKs*|dkr&d|d<tfd|i||d<|S)Nr rmarkersmoder%marker)r)trace trace_typer% kwargs_markerrrr_make_trace_for_scattersr\c) Cst||dd||dd}g}|s|st||}|D]z}tf| |dt||ddd|}|rt|d||d<|r|d||d <t|| ||df|}||ddq4nT|r|r|s|rt||r|n|}t|D]\}}||D]}|d|d||k}tf| |t||dd|}|rJ|||d<|r\|||d <t|| ||f|}|||r|dnd|rdn|dqt|d|r| n| |r|n|}| t ||r||n|d|r|n|||rd nd | qn|r|rt|||g}d d |D}||}||} ||}!t|D]\}"}#t| D]d\}$}%z||#|%f}Wn2t k rt j dddgg|||gd}YnX|!D]}|jdddggkr||||k}&tf| |t||dd|}|&|}'|&|}(n2tf| |t||ddd|}||}'||}(|r^|'|d<|rl|(|d <t|| ||f|}|||"d|$dq|"dkrnt| |$| |}| t ||$d||d | dqnt||"| |}| t |||"||d | dq\||fS)NTFZrowscolsZ shared_xaxesZ shared_yaxesZhorizontal_spacingZvertical_spacingZ print_gridr)r%)rHnamerXrr*r+r6r4cSsi|]}|d|dqSrrr.0itemrrr sz1_facet_grid_color_categorical..columns)rHr_rX showlegendr<r)rlistgroupbyrr\ append_trace enumerateuniquerrNr>KeyErrorpd DataFramevaluestolist))dfr*r+ facet_row facet_colr2colormap num_of_rows num_of_colsfacet_row_labelsfacet_col_labelsrZrPrQ show_boxesr; marker_color kwargs_tracer[rO annotationsZ color_groupsgrouprYgroups_by_facetjZ color_valZ data_by_colorrgroups_by_facetstuple_to_facet_group row_values col_valuesZ color_vals row_countx_val col_county_valZgroup_filteredZnew_xZnew_yrrr_facet_grid_color_categoricals0                     rc# Cst||dd||dd}g}|s|stf| t|||ddd|}|rT|||d<|rd|||d<t|| ||f|}||dd|r|r|s|rt||r|n|}t|D]\}}tf| t|||dtd d d d|}|r|d||d<|r|d||d<t|| ||f|}|||r@|dnd|rLdn|d|r`| n| }t|d ||rx|n|}|t ||r||n|d|r|n|||rd nd| dqn|r|rt|||g}dd|D}|| }|| }t|D]x\}} t|D]0\}!}"z|| |"f}Wn2t k rnt j dddgg|||gd}YnX|jdddggkrtf| t||||d ktd d d d|}ntf| dd|}|r|||d<|r|||d<t|| ||f|}|||d|!d|d kr"t||!| |}|t ||!d||d| dq"t||| |}|t ||||||d | dq||fS)NTFr])r% colorscale showscalerHrXr*r+rgffffff?)r*)r%rrZcolorbarrr6r4)rcSsi|]}|d|dqSr`rrarrrrdsz/_facet_grid_color_numerical..re)rHrgrh)rrr\rkrirjrlrrNr>rmrnrorprqrr)#rsr*r+rtrur2rvrwrxryrzrZrPrQr{r;r|r}r[rOr~rYrrrlabelsrrrrrrrrrrrr_facet_grid_color_numerical`s                  rc Cs0t||dd| | dd}g}|s|stf| t||ddd|}|rR|||d<|rb|||d<t|| |f|}||d d n|r|r|s|rt||r|n|}t|D]\}}tf| t||ddd|}|r|d ||d<|r|d ||d<t|| |f|}|||r,|d nd |r8d n|d t|d |rT|n||r`|n|}|t ||r|||n|d |r|n|| |rd nd | qn~|r(|r(t|||g}d d|D}|| }|| }t|D].\}}t|D]\}}z|||f}Wn.t k rNt j ddgg||gd}YnXtf| t||ddd|}|r|||d<|r|||d<t|| |f|}|||d |d |d krt||||}|t ||d || d | dqt||||}|t ||||| d | dq||fS)NTFr]rG)r%rGrr*r+rrr6r4cSsi|]}|d|dqSr`rrarrrrdUsz_facet_grid..rerh)rrr\rkrirjrlrrNr>rmrnrorp) rsr*r+rtrurwrxryrzrZrPrQr{r;r|r}r[rOr~rYrrrrrrrrrrrrrrr _facet_grids               rfixedc7Ks ts tdt|tjs"tdt|| dkrF|r<|sFtd|||||fD]8}|dk rTz ||WqTtk rtdYqTXqT| dkrd} | dkrtd | t krtd t | d krd }nd }d|kr|d}ni}| dd}| dd|}d|kr*|r"d|d<nd|d<d|krB|sBd|d<d|krj|sbddd|d<ni|d<|s||sd}nd}d}d}d}d}|rt || }t|}t|tr|| D]4}||kr|| }tt |q|rht || }t|}t| trh|| D]4}|| kr2|| }tt |q2d}|rt||jdts|rTd}t|trt|d|| D]}||krtdqnHtj} i}d}!|| D],}|!t | krd}!| |!||<|!d7}!qt||||||||||| | |||||||\}"}#qt||jdtrt|trd}t|d|| D]}||krtdqt||||||||||| | |||||||\}"}#nt|tr0|}$t|$t|||||||$|||| | |||||||\}"}#nt|tr|tjkrXtj|}$ntd tjt|||||||$|||| | |||||||\}"}#n:tjd!}$t|||||||$|||| | |||||||\}"}#n,t||||||||| | |||||||\}"}#| std"d#|} | s0td"d#|} |"d$j | | d%d&d'|r`|"d$j t!d(d)d*t"|d+}%t"|d,}&|##|%|##|&||"d$d-<t$|"d$d.d/<t%|"d$d.d0<d1|"d$d.d+<d|"d$d.d,<d2|"d$d.d3<|r$||"d$d-<|r$|rt&|}'|##|'d4|"d$d5d6<|#|"d$d7<|rJ|rJt'|"t(||ggd8}(|"d$D]8}d9|krz|(d+#|nd:|kr\|(d,#|q\d})d;d<||fDD]V}*t||*dtr||*D].}+zt)|+d})Wnt*k rYnXqАq|)r:|(D]&},|(|,D]}-d=|"d$|-d><qq| d?krNd+d,g}.n2| d@kr`d,g}.n | dAkrrd+g}.n| dkrg}.|.D]:},g}/g}0|"dBD]J}1|1|,dk rt |1|,dkr|/#t+|1|,|0#t|1|,qd|/kr|/,dqd|0kr|0,dqt+|/}2t|0}3t|2to:t|3t}4|4rt-.|2}2t-/|3}3|2dC|3|28}2|3dC|3|27}3|,d+kr|r|}5nt-.|3|2t0}5n(|,d,kr|r|}5nt-.|3|2t0}5nd}5|(|,D]}6|5|"d$|6dD<d|"d$|6dE<d|"d$|6dF<|rd|"d$|6dG<dH|"d$|6dE<t1|"d$|6dI<t2|"d$|6dJ<dK|"d$|6dI<t3dLdM|"d$|6dN<q|,|.kr|"d$D]0}dO |,|k r|4 r|2|3g|"d$|dP< qq|"S)Qai Returns figure for facet grid; **this function is deprecated**, since plotly.express functions should be used instead, for example >>> import plotly.express as px >>> tips = px.data.tips() >>> fig = px.scatter(tips, ... x='total_bill', ... y='tip', ... facet_row='sex', ... facet_col='smoker', ... color='size') :param (pd.DataFrame) df: the dataframe of columns for the facet grid. :param (str) x: the name of the dataframe column for the x axis data. :param (str) y: the name of the dataframe column for the y axis data. :param (str) facet_row: the name of the dataframe column that is used to facet the grid into row panels. :param (str) facet_col: the name of the dataframe column that is used to facet the grid into column panels. :param (str) color_name: the name of your dataframe column that will function as the colormap variable. :param (str|list|dict) colormap: the param that determines how the color_name column colors the data. If the dataframe contains numeric data, then a dictionary of colors will group the data categorically while a Plotly Colorscale name or a custom colorscale will treat it numerically. To learn more about colors and types of colormap, run `help(plotly.colors)`. :param (bool) color_is_cat: determines whether a numerical column for the colormap will be treated as categorical (True) or sequential (False). Default = False. :param (str|dict) facet_row_labels: set to either 'name' or a dictionary of all the unique values in the faceting row mapped to some text to show up in the label annotations. If None, labeling works like usual. :param (str|dict) facet_col_labels: set to either 'name' or a dictionary of all the values in the faceting row mapped to some text to show up in the label annotations. If None, labeling works like usual. :param (int) height: the height of the facet grid figure. :param (int) width: the width of the facet grid figure. :param (str) trace_type: decides the type of plot to appear in the facet grid. The options are 'scatter', 'scattergl', 'histogram', 'bar', and 'box'. Default = 'scatter'. :param (str) scales: determines if axes have fixed ranges or not. Valid settings are 'fixed' (all axes fixed), 'free_x' (x axis free only), 'free_y' (y axis free only) or 'free' (both axes free). :param (float) dtick_x: determines the distance between each tick on the x-axis. Default is None which means dtick_x is set automatically. :param (float) dtick_y: determines the distance between each tick on the y-axis. Default is None which means dtick_y is set automatically. :param (bool) show_boxes: draws grey boxes behind the facet titles. :param (bool) ggplot2: draws the facet grid in the style of `ggplot2`. See http://ggplot2.tidyverse.org/reference/facet_grid.html for reference. Default = False :param (int) binsize: groups all data into bins of a given length. :param (dict) kwargs: a dictionary of scatterplot arguments. Examples 1: One Way Faceting >>> import plotly.figure_factory as ff >>> import pandas as pd >>> mpg = pd.read_table('https://raw.githubusercontent.com/plotly/datasets/master/mpg_2017.txt') >>> fig = ff.create_facet_grid( ... mpg, ... x='displ', ... y='cty', ... facet_col='cyl', ... ) >>> fig.show() Example 2: Two Way Faceting >>> import plotly.figure_factory as ff >>> import pandas as pd >>> mpg = pd.read_table('https://raw.githubusercontent.com/plotly/datasets/master/mpg_2017.txt') >>> fig = ff.create_facet_grid( ... mpg, ... x='displ', ... y='cty', ... facet_row='drv', ... facet_col='cyl', ... ) >>> fig.show() Example 3: Categorical Coloring >>> import plotly.figure_factory as ff >>> import pandas as pd >>> mtcars = pd.read_csv('https://raw.githubusercontent.com/plotly/datasets/master/mtcars.csv') >>> mtcars.cyl = mtcars.cyl.astype(str) >>> fig = ff.create_facet_grid( ... mtcars, ... x='mpg', ... y='wt', ... facet_col='cyl', ... color_name='cyl', ... color_is_cat=True, ... ) >>> fig.show() z3'pandas' must be installed for this figure_factory.z"You must input a pandas DataFrame.rUz`You need to input 'x' and 'y' if you are you are using a trace_type of 'scatter' or 'scattergl'.NzIx, y, facet_row, facet_col and color_name must be keys in your dataframe.free)rfree_xfree_yrz?'scales' must be set to 'fixed', 'free_x', 'free_y' and 'free'.z'trace_type' must be in {}rgQ?gQ?rXr%r$r Zopacityg333333?rGZdarkgreyrrEzrgb(31, 119, 180)z rgb(0, 0, 0)FrTZrgbzxIf using 'colormap' as a dictionary, make sure all the values of the colormap column are in the keys of your dictionary.zlIf 'colormap' is a string, it must be the name of a Plotly Colorscale. The available colorscale names are {}ZRedsiXdrCr?zrgb(251, 251, 251))heightrFtitle paper_bgcolorzrgb(255, 255, 255)Zclosest)Z plot_bgcolorrZ hovermoder*r+rgZlegendZbgcolorZ borderwidthr topr)marginrr~)r*r+rIrLcSsg|] }|r|qSrr)rbvrrr Xsz%create_facet_grid..categoryrHrrrdatag?dtickZticklenZzerolineZ tickwidthZ gridwidthZ gridcolorr r@Ztickfontz{}axisrange)4ro ImportErrorrrprZ PlotlyErrorrZvalidate_dataframernVALID_TRACE_TYPESrpoplenrmrrrMrrCUSTOM_LABEL_ERRORZilocrclrsZvalidate_colors_dictZDEFAULT_PLOTLY_COLORSrrriZvalidate_colorscalerZ PLOTLY_SCALESrmaxupdate PLOT_BGCOLORrBrN LEGEND_COLORLEGEND_BORDER_WIDTHr3rTANNOT_RECT_COLORint ValueErrorminremovemathfloorceilMAX_TICKS_PER_AXIS GRID_WIDTH GRID_COLOR TICK_COLOR)7rsr*r+rtrur2rvZ color_is_catryrzrrFrZZscalesZdtick_xZdtick_yr{Zggplot2ZbinsizekwargsrRr;r[r|r}rwrxrPrQZ unique_keysZ show_legendvalZdefault_colorsrrOr~Zcolorscale_listZ x_title_annotZ y_title_annotZ legend_annotZ axis_labelsZstring_number_in_datavarrcZx_yZ axis_nameZ fixed_axesZ min_rangesZ max_rangesrYZ min_rangeZ max_rangeZrange_are_numbersrZ axis_titlerrrcreate_facet_grids                                                                r)r4T)FF)NNNNNNFNNNNr rNNTFr))ZplotlyrrZ plotly.colorscolorsrZplotly.figure_factoryrZplotly.subplotsrrZnumbersrZ get_modulerorr:rArrrrrr1ZLEGEND_ANNOT_Yrrrrrrrr3r>rBrTr\rrrrrrrrsp       , $*