B &[_@sddlmZddlmZmZddlmZddlm Z ddl m Z ddl Z ddl mZedZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdZdddddgZdZddZ ddZ!ddZ"d4d"d#Z#d$d%Z$d5d'd(Z%d)d*Z&d+d,Z'd-d.Z(d/d0Z)d6d2d3Z*dS)7)absolute_import) exceptionsoptional_importsN)utils) make_subplots)NumberZpandasz#969696z#0f0f0f z#ffffffz#efefefz#edededz#d0d0d0g?g? 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-install-l29rncou/plotly/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) Nrleftmiddleg{Gz?Fpaperz factor({}) z#000000)sizecolor) textanglexanchoryanchorxy showarrowxrefyreftextfont)rLEGEND_ANNOT_Xr) color_nameZ legend_titlerrr_legend_annotation6sr.colTc 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)Nr centerrr/g?g{Gz?rrowZZbottomg?irFrr)r r!) r"r#r$r%r&r'r(r)r*r+)rrAXIS_TITLE_COLOR) r*ZlaneZ num_of_lanesSUBPLOT_SPACINGrow_colrlr#r$r%r&r"Zannotation_dictrrr_annotation_dictFsHr7c CsV|dkrd}d}d}n|dkr*d}d}d}|s2d}dtd d |||d d |d d d }|S)Nr%g?grr&iz#000000)r!r Fr0rr) r+r'r*r"r%r#r(r&r$r))AXIS_TITLE_SIZE)r*Z x_or_y_axisZx_posZy_posr"Zannotrrr_axis_title_annotationrs*r:FcCs:g}x"|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<||qW||dd<dS)NlayoutZaxisdomaingg?Zbelowr )r!widthZrectr)Z fillcolorZlayerlinetyper(r)xaxisrZx0x1gGz?Zy0g?y1g?yaxisZshapes)Zto_plotly_jsonkeysappend)figZannot_rect_color flipped_rows flipped_colsZ shapes_listkeyshaperrr_add_shapes_to_figs4  rKcKs*|dkr&d|d<tfd|i||d<|S)N)r r markersmoder!marker)r)trace trace_typer! kwargs_markerrrr_make_trace_for_scattersrRc) Cs0t||dd||dd}g}|s|st||}x|D]z}tf| |dt||ddd|}|rv|d||d<|r|d||d <t|| ||df|}||ddq6Wnp|r|r|s|rt||r|n|}xBt|D]\}}x||D]}|d|d||k}tf| |t||dd|}|rT|||d<|rf|||d <t|| ||f|}|||r|dnd|rdn|dqWt|d|r| n| |r|n|}| t ||r||n|d|r|n|||rd nd | qWn|r(|r(t|||g}d d |D}||}||} ||}!xt|D]\}"}#xvt| D]h\}$}%y||#|%f}Wn2t k rt j dddgg|||gd}YnXx|!D]}|jdddggkr8||||k}&tf| |t||dd|}|&|}'|&|}(n2tf| |t||ddd|}||}'||}(|rx|'|d<|r|(|d <t|| ||f|}|||"d|$dqW|"dkrt| |$| |}| t ||$d||d | dqWt||"| |}| t |||"||d | dqpW||fS)NTF)rowscols shared_xaxes shared_yaxeshorizontal_spacingvertical_spacing print_gridr)r!)r?namerNr r%r&r1r/cSsi|]}|d|dqS)r rr).0itemrrr sz1_facet_grid_color_categorical..)columns)r?rZrN showlegend)r5r)rlistgroupbyrrR append_trace enumerateuniquerrEr7KeyErrorpd DataFramevaluestolist))dfr%r& facet_row facet_colr-colormap num_of_rows num_of_colsfacet_row_labelsfacet_col_labelsrPrGrH show_boxesr4 marker_color kwargs_tracerQrF annotationsZ color_groupsgrouprOgroups_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_categoricals   ,           rc# Cst||dd||dd}g}|s|stf| t|||ddd|}|rT|||d<|rd|||d<t|| ||f|}||dd|r|r|s|rt||r|n|}xt|D]\}}tf| t|||dtd d d d|}|r|d||d<|r|d||d<t|| ||f|}|||rD|dnd|rPdn|d|rd| n| }t|d ||r||n|}|t ||r||n|d|r|n|||rd nd| dqWn|r|rt|||g}dd|D}|| }|| }xt|D]~\}} x>t|D]0\}!}"y|| |"f}Wn2t k r|t j dddgg|||gd}YnX|jdddggkrtf| t||||d ktd d d d|}ntf| dd|}|r|||d<|r|||d<t|| ||f|}|||d|!d|d kr0t||!| |}|t ||!d||d| dq0Wt||| |}|t ||||||d | dqW||fS)NTF)rSrTrUrVrWrXrY)r! colorscale showscale)r?rNr%r&r gffffff?)r%)r!rrZcolorbarrr1r/)rcSsi|]}|d|dqS)r rr)r[r\rrrr]sz/_facet_grid_color_numerical..)r^)r?r_)r5r)rrrRrbr`rarcrrEr7rdrerfrgrhri)#rjr%r&rkrlr-rmrnrorprqrPrGrHrrr4rsrtrQrFrurOrwrxrvlabelsrryrzr{r|r}r~rrrrr_facet_grid_color_numericalbs  &         rc Cs>t||dd| | dd}g}|s|stf| t||ddd|}|rR|||d<|rb|||d<t|| |f|}||d d n|r|r|s|rt||r|n|}xt|D]\}}tf| t||ddd|}|r|d ||d<|r |d ||d<t|| |f|}|||r.|d nd |r:d n|d t|d |rV|n||rb|n|}|t ||r~||n|d |r|n|| |rd nd | qWn|r6|r6t|||g}d d|D}|| }|| }x@t|D]2\}}xt|D]\}}y|||f}Wn.t k rXt j ddgg||gd}YnXtf| t||ddd|}|r|||d<|r|||d<t|| |f|}|||d |d |d krt||||}|t ||d || d | dqWt||||}|t ||||| d | dqW||fS)NTF)rSrTrUrVrWrXrYr>)r!r>)r?rNr%r&r rr1r/cSsi|]}|d|dqS)r rr)r[r\rrrr]Wsz_facet_grid..)r^)r5r)rrrRrbr`rarcrrEr7rdrerfrg) rjr%r&rkrlrnrorprqrPrGrHrrr4rsrtrQrFrurOrwrxrvrryrzr{r|r}r~rrrrr _facet_grids  &         rfixedc7Ks ts tdt|tjs"tdt|| dkrF|r<|sFtdxJ|||||gD]8}|dk rVy ||WqVtk rtdYqVXqVW| dkrd} | dkrtd | t krtd t | d krd }nd }d|kr|d}ni}| dd}| dd|}d|kr.|r&d|d<nd|d<d|krF|sFd|d<d|krn|sfddd|d<ni|d<|s|sd}nd}d}d}d}d}|rt || }t|}t|trxD|| D]4}||kr|| }tt |qW|rtt || }t|}t| trtxD|| D]4}|| kr<|| }tt |q<qFWq8W| d?krzd+d,g}.n2| d@krd,g}.n | dAkrd+g}.n| dkrg}.xX|.D]N},g}/g}0xV|"dBD]J}1|1|,dk rt |1|,dkr|/#t+|1|,|0#t|1|,qWxd|/kr4|/,dqWxd|0krP|0,dq8Wt+|/}2t|0}3t|2tovt|3t}4|4 rt-.|2}2t-/|3}3|2dC|3|28}2|3dC|3|27}3|,d+kr|r|}5nt-.|3|2t0}5n(|,d,k r |r|}5nt-.|3|2t0}5nd}5x|(|,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< qW|,|.krx<|"d$D]0}dO |,|k r|4 r|2|3g|"d$|dP< qWqW|"S)Qaj 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.)r r z`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?rNr!r r Zopacityg333333?r>Zdarkgreyr )r!r=zrgb(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 {}ZRedsiXdr;r8zrgb(251, 251, 251))heightr=title paper_bgcolorzrgb(255, 255, 255)Zclosest)Z plot_bgcolorrZ hovermoder%r&r_ZlegendZbgcolorZ borderwidthg?topr$marginrru)r%r&r@rCcSsg|] }|r|qSrr)r[vrrr Zsz%create_facet_grid..categoryr?rrrdatag?dtickZticklenZzerolineZ tickwidthZ gridwidthZ gridcolorr )r!r Ztickfontz{}axisrange)4rf ImportErrorrrgrZ PlotlyErrorrZvalidate_dataframereVALID_TRACE_TYPESrpoplenrdrrrDriCUSTOM_LABEL_ERRORZilocrclrsZvalidate_colors_dictZDEFAULT_PLOTLY_COLORSrrr`Zvalidate_colorscalerZ PLOTLY_SCALESrmaxupdate PLOT_BGCOLORr:rE LEGEND_COLORLEGEND_BORDER_WIDTHr.rKANNOT_RECT_COLORint ValueErrorminremovemathfloorceilMAX_TICKS_PER_AXIS GRID_WIDTH GRID_COLOR TICK_COLOR)7rjr%r&rkrlr-rmZ color_is_catrprqrr=rPZscalesZdtick_xZdtick_yrrZggplot2ZbinsizekwargsrIr4rQrsrtrnrorGrHZ unique_keysZ show_legendvalZdefault_colorsrxrFruZcolorscale_listZ x_title_annotZ y_title_annotZ legend_annotZ axis_labelsZstring_number_in_datavarr\Zx_yZ axis_nameZ fixed_axesZ min_rangesZ max_rangesrOZ min_rangeZ max_rangeZrange_are_numbersrZ axis_titlerrrcreate_facet_grids                                                        r)r/T)FF)NNNNNNFNNNNr rNNTFr )+ __future__rZplotlyrrZ plotly.colorscolorsrZplotly.figure_factoryrZplotly.subplotsrrZnumbersrZ get_modulerfrr3r9rrrrrr,ZLEGEND_ANNOT_Yrrrrrrrr.r7r:rKrRrrrrrrrrsj        + $*