B &[_YM @sddlmZddlmZddlmZmZddl m Z m Z m Z ddl mZddlmZmZddlZddlZddlZddlmZmZmZdZd d d d d dddddddg ddddddgdddddd gd!d"d#d$d%gZd&d'd(d)d*gZd+d,d-d.gZd/d0d1gZeeeeZ ej!ej"ej#ej$ej%ej&gZ'e'ej(ej)ej*g7Z'Gd2d3d3e+Z,e,Z-[,da.d4d5Z/d6d7Z0ed8d9d:d;dd?gZ1ed@dAdBdCdDgZ2dEdFZ3dGdHZ4dIdJZ5dKdLZ6dMdNZ7dOdPZ8dQdRZ9dSdTZ:dUdVZ;dWdXZd]d^Z?d_d`Z@dadbZAdcddZBdedfZCdgdhZDdidjZEdkdlZFdmdnZGdodpZHdqdrZIdsdtZJdudvZKdwdxZLdydzZMd{d|ZNd}d~ZOddZPddZQddZRdddZSddZTdS)N) namedtuple OrderedDict) IdentityMapConstantRange)ColorscaleValidator) qualitative sequential) make_subplots_set_trace_grid_reference_subplot_type_for_trace_typeZpx_no_color_constantbasexyzabcrthetasizex_startx_end hover_nametextnamesvaluesparents wide_crossidsZerror_xZ error_x_minusZerror_yZ error_y_minusZerror_zZ error_z_minuslatlon locationsanimation_group dimensions custom_data hover_datapath wide_variableanimation_frame facet_row facet_col line_groupcolorsymbol line_dashc@s(eZdZddddddddgZd d Zd S) PxDefaultstemplatewidthheightcolor_discrete_sequencecolor_continuous_scalesymbol_sequenceline_dash_sequencesize_maxcCs4d|_d|_d|_d|_d|_d|_d|_d|_dS)N)r2r3r4r5r6r7r8r9)selfr<8/tmp/pip-install-l29rncou/plotly/plotly/express/_core.py__init__6szPxDefaults.__init__N)__name__ __module__ __qualname__ __slots__r>r<r<r<r=r1*sr1cCs|adS)z Arguments: token: A Mapbox token to be used in `plotly.express.scatter_mapbox` and `plotly.express.line_mapbox` figures. See https://docs.mapbox.com/help/how-mapbox-works/access-tokens/ for more details N) MAPBOX_TOKEN)tokenr<r<r=set_mapbox_access_tokenHsrEcCs|jS)a Extracts fit statistics for trendlines (when applied to figures generated with the `trendline` argument set to `"ols"`). Arguments: fig: the output of a `plotly.express` charting call Returns: A `pandas.DataFrame` with a column "px_fit_results" containing the `statsmodels` results objects, along with columns identifying the subset of the data the trendline was fit on. )_px_trendlines)figr<r<r=get_trendline_resultsSs rHMappingshow_in_trace_namegrouperval_mapsequenceupdatervariablefacet TraceSpec constructorattrs trace_patchmarginalcCs&y |d|Stk r |SXdS)Nlabels) Exception)argscolumnr<r<r= get_labelqs rZcCs8dd|dD}y||Stk r2|SXdS)zInvert mapping. Find key corresponding to value column in dict args["labels"]. Returns `column` if the value does not exist. cSsi|]\}}||qSr<r<).0keyvaluer<r<r= }sz invert_label..rVN)itemsrW)rXrYZreversed_labelsr<r<r= invert_labelxs r`cCs||jjdkS)NZifc)dtypekind)dfcol_namer<r<r=_is_continuoussrecCsxt||}d|krp|dkr.d|kr.|ddksR|dkrJd|krJ|ddksR|dkrp|rjd|dpbd |fSd Sn|SdS) Nhistfuncr orientationhrvrz%s of %scount)rZ)rXrYrolelabelr<r<r=get_decorated_labels rmc sdksdkr0td|idgddddSdks@d krd krLd nd }td||id d tddDdddkrdnddSd\}}dkrd}dkrd}d}||ddkrt}n||d}td|||||dfddddS)Nr-r*FcSs|S)Nr<)tracerir<r<r=zmake_mapping..)rJrKrLrMrOrNrPr+r,rrcSsg|]}|qSr<r<)r[ir<r<r= sz make_mapping..ricSs|S)Nr<)rorir<r<r=rprqrowcol)rJrOrKrLrMrNrP.r.Zcolor_discretedashr0_mapidentityTZ _sequencecs||iiS)N)update)rori)parentrOr<r=rprq)rIrangesplitrcopy)rXrOletterZvprefixarg_namerLr<)r{rOr= make_mappingsL  rc# s dkr dr ||jd}jp,i}d}d}xjD]}|t|} |dkrЇfdd|D} fdd| D|d<jtj krx|dD]} t d d | d <qWd |d <d|d<q@dk s jtj kr|dks jtj tj gkr@|dkr@|dkrZd|kr(t |d<||dd<d|dd<||dd<d|| <q@|dkr|jtj kr d|d<q@|dkrjtj kr d |d<q@|dkrdkr dr dr t|ddgdkr ddlm} |jdd } | dj}| dj}|jjtjkr@|td!}nH|jjtjkry|tj}Wn&tk rtd"dYnX|jjtjkry|tj}Wntk rtd#YnX| dttt|t||d<d$kr,| j j!||d%d&}|dddf|d<d'}nd(kr| j"|| #|d%d&$}|%|d<d)}t|j&d*kr|d+d|j&dd|j&df7}n|d,d|j&df7}|d-|j'7}d |t(d<d.|t(d<q@|)d/r8|dd0}|*d1rd2nd3}||kr&i||<||||<q@|d4krP||d5<q@|d6krjtj tj tj gkr ||d7<|dkr d8}q@|d9krVjtj tj tj gkr t+t },d4pg}x|D]t}|r|sqyd4-|}Wn.tt.t/fk r(t|}||YnXt|d}d:|||<qW|||d5<q@|d;kr؈jtj0tj1gkr||d<d<|d=<d>|| <q jtj2tj3tj4tj5gkr~d|krt |d<,d?r||dd@<d<|dd=<dA|| <ng|dd@<dBdk rdB}ni}x|D]L}|,|dkr`dCt|tdC||<|dd@||q,WnXd}jtj6tj7gkrdD}||krt ||<|||d;<d<||d=<dE||| <q@|dFkr||dG<q@|dHkr|||<dI|| <q@|dJkrB|||<| dJkr4dKn| }dL||<q@|dMkrt|||<| dMkrfdNn| }dO||<q@|dGkr|||<| dGkrdPn| }dQ||<q@|dRkrjtj2tj3tj4tj5gkr||dS<| dRkrdTn| }dU||<n |||<q@r|||<dV||| <q@Wjtj7tj6gk rt8|}d9rt+d9t rxp|9D]d\}}t:|}|d9krdd9|d} | rt+| t;r||"|d[<|d[d\7<||fS)]a:Populates a dict with arguments to update trace Parameters ---------- args : dict args to be used for the trace trace_spec : NamedTuple which kind of trace to be used (has constructor, marginal etc. attributes) trace_data : pandas DataFrame data mapping_labels : dict to be used for hovertemplate sizeref : float marker sizeref Returns ------- trace_patch : dict dict to be used to update trace fit_results : dict fit information to be used for trendlines Z line_closerNrnr%csvg|]n\}}r|krjtjks2td|rjtjksjdk rN|ksjtd|dkr||fqS) data_frameNZdimensions_max_cardinality)rRgo ParcoordsreParcatslenunique)r[namerY)rX attr_value trace_specr<r=rss   z%make_trace_kwargs..cs"g|]\}}tt||dqS))rlr)dictrZ)r[rrY)rXr<r=rssT)matchesaxisz%{x}z%{xaxis.title.text}z%{y}z%{yaxis.title.text})rrrrmarkerareaZsizemodesizerefz%{marker.size} marginal_xrj marginal_y trendline)olslowessrrr)byiʚ;zCould not convert value of 'x' ('%s') into a numeric type. If 'x' contains stringified dates, please convert to a datetime column.z3Could not convert value of 'y' into a numeric type.rdrop)missingzLOWESS trendline

rzOLS trendline
z%s = %g * %s + %g
z %s = %g
zR2=%f

z%{y} (trend)errorminusZ arrayminusarrayr&Z customdatarZ hovertextz%{hovertext}

r'z%%{customdata[%d]}r. coloraxis1 coloraxisz%{z}color_is_continuouscolorsz%{color}Zcolor_discrete_mapr5linez %%{%s.color}r$r r#z %{location}rr]z%{value}rr{z %{parent}idz%{id}rrVrlz%{label}z%%{%s}}z%s}cSsg|]\}}|d|qS)=r<)r[krir<r<r=rssz
Z hovertemplatez)?appendilocrTr~rSrm iteritemsrRrSplomr Histogram Histogram2dHistogram2dContourrZdropnaZstatsmodels.apiapi sort_valuesrratypenpZ datetime64astypeintZobject_Zfloat64 ValueErrorZ logical_not logical_orisnanZ nonparametricrZOLSZ add_constantZfitZpredictparamsZrsquaredrZ startswithendswith isinstancegetindexAttributeErrorKeyError ChoroplethChoroplethmapboxSunburstTreemapPie Funnelarearrrr_r`strreplacepopjoin)#rXrZ trace_datamapping_labelsrrT fit_resultsZ hover_header attr_nameZ attr_labelZdimsdsmZsorted_trace_datarrrZerror_xyZarrZ hover_is_dictZcustomdata_colsrupositionZattr_label_colmappingcatZ colorableZ_labelZmapping_labels_copyrriZk_args formatter_Z hover_linesr<)rXrrr=make_trace_kwargss                     "   $                                                   rcCsrtjttjttjttjttjttj t tj t tj t tj ttjti }xtD] }t||<qFW||krn|||||dS)N)rZ Scatter3dconfigure_3d_axesZScatterternaryconfigure_ternary_axes Scatterpolarconfigure_polar_axesScatterpolarglZBarpolar Scattermapboxconfigure_mapboxr DensitymapboxZ Scattergeo configure_geor cartesiansconfigure_cartesian_axes)rXrRrGordersZ configuratorsrr<r<r=configure_axess   rcCsd|}d|}||krP||rPd|d<||krl||rldd||D|d<n||krl||rl|||d<|||krd|d <t|tjjr|||ntt||||d <dS) Nlog_range_logrcSsg|]}t|dqS) )mathr)r[rr<r<r=rssz+set_cartesian_axis_opts..r|r categoryorder categoryarray)rrlayoutZXAxislistreversed)rXrrrZlog_keyZ range_keyr<r<r=set_cartesian_axis_optss  rc Csd|d|dgkrd|jd<t|j}t|jd}x"|jddD]}t||d |qDWx"|jdd D]}t||d |qhW|dr|jd d d d|d|djjjdkr|j|ddk|d|djj jdkr|j d|d|drL|j d d d d|d|djj jdkr*|j |ddk|d|djjjdkrL|jd|dt ||d d }|drx|j|dddn*x(t d|dD]}|j||ddqWt ||d d } |dr|j | dddn*x(t d|dD]} |j | d| dqWd|kr|dr|j ddd|kr8|dr8|jddd t |d} |dr~x*t d|ddD]}|j| d|dqdW|drx*t d|ddD]} |j dd| dqWdS)N histogramrroverlaybarmoderr)rur)rtrFrn)showticklabelsshowlineticksr|rtr2)showgridrtT)rrrr|ru)rru) title_textrtrulog_xr)rlog_yr)rrrtx2)rrru)rr _grid_ref select_yaxesr select_xaxes update_yaxesyaxisrxaxis update_xaxesrmr|r) rXrGrnrowsncolsrry_titlertx_titleruZ matches_yr<r<r=!configure_cartesian_marginal_axessV         rcCs d|kr|ds d|kr0|dr0t|||dSt||dd}x.|jddD]}|j|dt||d|qNWt||dd}x6|jddD]&}d |kr|j|dt||d|qWd |kr|d r|jd d d |kr|d r|jd d d |kr|jdd |jS)Nrrrr)ru)rr)rt is_timelinerr)rrdate) rrmrrzrrrrr)rXrGrrrrrr<r<r=rLs(       rc CsJ|jttt||ddtt||ddtt||dddddS)Nr)rrr)ZaaxisZbaxisZcaxis)Zternary) update_layoutrrZ)rXrGrr<r<r=rns rcCsttt|d|ddtdd}xDdD]<\}}|||kr*d|d|d <||||d|d <q*W|dd }|d rd |d<|drdd|dD|d<n|dr|d|d<|dr|d|dd<|j|ddS)N directionZ start_angle)rZrotation) angularaxis radialaxis)polar))rr)rrrr rrrZlog_rrrZrange_rcSsg|]}t|dqS)r)rr)r[rr<r<r=rssz(configure_polar_axes..r|Z range_thetaZsector)r)rrz)rXrGrrvarrrr<r<r=rxs$   rc Cstttt||ddtt||ddtt||dddd}xdD]}|d|d }|d |rd |d <|d |rdd|d |D|d<n|d |r|d ||d<|||krJd|d<||||d<qJW|j|ddS)Nr)rrr)rrZzaxis)scene)rrrr rrrrrcSsg|]}t|dqS)r)rr)r[rr<r<r=rssz%configure_3d_axes..r|rrr)r)rrZrz)rXrGrrrrr<r<r=rs"     rcCsl|d}|sHd|krHd|krHt|d|d|d|dd}|jtt||d|ddd dS) Ncenterr!r"r)r!r"zoomZ mapbox_style)Z accesstokenr r style)Zmapbox)rZmeanrrC)rXrGrr r<r<r=rsrcCs,|jt|d|dt|dddddS)Nr scope projection)r)r rr)Zgeo)rr)rXrGrr<r<r=rs rc sfddd|kr|drt|jdkrddgddddgd gd ddgd d d dddddd dd g|j_d dd dt||ddid ddddd fdd|jDd g|j_dS)Ncs |tjkddd|dddS)N)durationZredrawZ immediateTZlinear)rZeasing)framemodeZ fromcurrentZ transition)rScatter)r)rRr<r= frame_argssz0configure_animation_controls..frame_argsr*riz▶animate)rXrlmethodrz◼leftrF)rtFbuttonsg?righttop) rrpadZ showactiverrxanchorryanchorprefixr<)rrg?cs&g|]}|jgdg|jddqS)rr)rXrlr)r)r[f)rr<r=rssz0configure_animation_controls..) activer rZ currentvaluerrrrZsteps)rframesrZ updatemenusrZZsliders)rXrRrGr<)rRrr=configure_animation_controlss4    r&c Cs|tjtjgkrrd|krr|ddksL|ddkrrt|ddkrr|ddkrr|tjkrltj}d|krr|d=ntj}t|||dg}xdD]}d ||kr|d |rd}t|d krd nd |d krdndd}|d |dkrttj|d |gtfd|d||d}n|d |dkr@ttj |ddgt|d|d}n|d |dkrrttj |ddgtdd|d}nP|d |dkrddd}ttj |ddgtdd did!d"d#d$||id%|d}d |ksd |krd&|j krt|j d&<|d'd"} | |j d&d <| |qWd(|kr~|d(r~t|tjkr>tjntjd(gtd)d*dd}|d+rtt|d+d,|j d-<| ||S).NZ render_modeZwebglautorir*rg)rrZ marginal_rx1rry1y2)rrrg?)opacitybingroup)rRrSrTrUZviolinrr')Z scalegroupZboxT)ZnotchedZrugz line-ns-openz line-ew-openzrgba(255,255,255,0)r.allrZpointsr/)Z fillcolorrZ boxpointsjitterZhoveronrrr6rlines)rZtrendline_color_override)r.r) rrrr ScatterglrrQrrViolinBoxrTr) rXrRrSrTresultrrZaxis_mapsymbolsZfirst_default_colorr<r<r=make_trace_specs              r5cCsdS)Nrnr<)rr<r<r= one_groupVsr6cCsxFddgdddgdddgD]&}||kr||dkrtt|||<qW|ddkrvtjjdk rntjj|d<nd |d<ytj|d|d<Wn(tk rtj|d|d<YnXd|kr|ddkr|djj j rd d |djj j D|d<|ddkrt j |d<d|kr\|ddkrD|djj rD|djj |d<|ddkr\t j|d<d|kr|ddkr|djjrd d |djjD|d<|drt|dsd ddddg|d<d|kr2|ddkr|djjrdd |djjD|d<|drt|ds2ddddddg|d<|dddk rZ|ddrZd|d<|dddk r|ddrd|d<dS)Nr5r6r7r8r2r3r4r9ZplotlycSsg|] }|dqS)rr<)r[rr<r<r=rs{sz)apply_default_cascade..cSsg|] }|jjqSr<)rr/)r[scatterr<r<r=rssZcircleZdiamondZsquarerZcrosscSsg|] }|jjqSr<)rrw)r[r7r<r<r=rssZsoliddotrwZlongdashZdashdotZ longdashdotr,rr+r)getattrdefaultspioZ templatesdefaultrWrrTemplate colorscaler ZViridisZcolorwayr ZD3datar7anyr)rXparamr<r<r=apply_default_cascadeZsT             rBcCs ||kr |Std||fdS)NzaA name conflict was encountered for argument '%s'. A column or index with name '%s' is ambiguous.) NameError) field_namereserved_namesr<r<r=_check_name_not_reserveds rFcCs|d}t}x|D]}|tkr"q|tkr2||n||g}|dkrFqx|D]}|dkr\qLqLt|trr||qLt|tjr|j}|rt ||r|||k}|r||qL||j krL|jdk rL||jqLWqW|S)z This function builds a list of columns of the data_frame argument used as arguments, either as str/int arguments or given as columns (pandas series type). rN) set all_attrablesarray_attrablesrraddpdSeriesrhasattrr)rXrcrEfieldrargrZin_dfr<r<r=_get_reserved_col_namess,       rPc Cs|dkst|tst|tr dSt|tjr0dSy t|Wntk rPdSXxZ|D]R}t|tspt|tr|dks||jkrdSqXy t|WqXtk rdSXqXWdS)zReturns True if arg looks like it's a list of columns or references to columns in df_input, and False otherwise (in which case it's assumed to be a single column or reference to a column). NFT)rrrrK MultiIndexiter TypeErrorcolumns)df_inputrOrr<r<r= _is_col_lists"     rVcCs>t|ts2t|tr t|dks2t|ts2|dkr6dSdSdS)zReturns True if x is an iterable which can be transformed into a pandas Series, False for the other types of possible values of a `hover_data` dict. A tuple of length 2 is a special case corresponding to a (format, data) tuple. rNFT)rrtuplerbool)rr<r<r=_isinstance_listlikes   rYcCs,x&|dk r&||jks||kr&d|}qW|S)Nr)rT)rUrdextrar<r<r=_escape_col_names r[cCst|jddS)aU assuming x is list-like or even an existing pd.Series, return a new pd.Series with no index, without extracting the data from an existing Series via numpy, which seems to mangle datetime columns. Stripping the index from existing pd.Series is required to get things to match up right in the new DataFrame we're building T)r)rKrL reset_index)rr<r<r=to_unindexed_seriessr]c s<|d}|dk }t}t}t}t} |r6t|nt} d|krn|ddkrn|s^tdn||j||j<d|ko|dot|dt} | r"x|dD]} t |d| rd|d| f|d| <t|d| t s|d| df|d| <|r|d| ddk r| |krtd| qWxt D]ȉt krF| gn| } | dks*| dgkrjq*t krzgnfd d tt| D}xZtt| |D]F\}\}}t|}|dkrʐqd}t|tjrtd |t|tst|trFt|jdk rt|jn|| }t|tr8|j||<n ||nBt|ts^t|d sdkr| r|dt|ddk rt|}|d|d}|rt||krtd |t|tt|j|ft|||<n|std|n||jkrR|r|||fkrqn2d|tt|j|f}|dkrH|d7}t|nX|rt|||krtd|t||tt|j|fnt|}t||||<n|r.t|dr.||jkr|jdks|j|krd}n|j}t||||g}n.|jdk r.|j|kr.|||jkr.|j}|dkrBt|| }|rxt||krxtd|t|tt|j|ft||t|<|dk st dt krt||<n"t|trnt|||<dkr| !t|qWq*Wx|D]}tt|||<qWx|D]}||||<qW|| fS)ac After this function runs, the `all_attrables` keys of `args` all contain only references to columns of `df_output`. This function handles the extraction of data from `args["attrable"]` and column-name-generation as appropriate, and adds the data to `df_output` and then replaces `args["attrable"]` with the appropriate reference. rNr%zjNo data were provided. Please provide data either with the `data_frame` or with the `dimensions` argument.r'TrzIAmbiguous input: values for '%s' appear both in hover_data and data_framecsg|]}dt|qS)r)r)r[rr)rDr<r=rsFsz/process_args_into_dataframe..ziArgument '%s' is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.__len__zAll arguments should have the same length. The length of hover_data key `%s` is %d, whereas the length of previously-processed arguments %s is %dzString or int arguments are only possible when a DataFrame or an array is provided in the `data_frame` argument. No DataFrame was provided, but argument '%s' is of type str or int.z^Value of '%s' is not the name of a column in 'data_frame'. Expected one of %s but received: %srz6 To use the index, pass it in directly as `df.index`.zAll arguments should have the same length. The length of column argument `df[%s]` is %d, whereas the length of previously-processed arguments %s is %drzAll arguments should have the same length. The length of argument `%s` is %d, whereas the length of previously-processed arguments %s is %dzData-frame processing failure, likely due to a internal bug. Please report this to https://github.com/plotly/plotly.py/issues/new and we will try to replicate and fix it.r))"rK DataFramerrrGrPrrTrrYrWrHrIrr|r enumerateziprQrSrrrFrlrr]rrMr]rrr[AssertionErrorrJ)rX wide_modevar_name value_namerU df_provided df_output constantsranges wide_id_varsrEZhover_data_is_dictrZ argument_listZ field_listrrargumentrNlengthrdZ real_argumenterr_msgr<)rDr=process_args_into_dataframe s                            rncs2xJ|D]B}|tkr||dk rt||tr8t||n t||||<qW|ddk }|r~t|dtjs~t|d|d<|d}|dddk}|dddk}|rdn t||d}|rdn t||d}d} d} d} d} tj tj g} |t kr|r |r t d|r|r|rd} t|j tjr:tdt|j |d <|j j} | d ksd| |krhd } |tjkr|d dpd }n|d dpd}||d <d|d<n||krd} |r|dn|d|d <|r|d |j kr|j j} t|d tjrt|d |d <| d ks.|r2| |kr2d } |tjkrN|rHdnd }n|rXdnd }d||rjdnd<d|d<|s|s|rdnd} | rt|dg} t|| g} d}|tjtjtjg| kr| sJ||krJxldD]d}||ddkr|r|jnt||<|tjkr |}n$|d dkr|dkr| ||rjdnd<|||r|dnd<d||<|dp| |d<|tj%tj&gkr| p| ||rdnd<| ||rdnd<|tjkr| ||rdnd<| ||rdnd<|dp| |d<|r&d|d<||d<|S)"a Constructs a dataframe and modifies `args` in-place. The argument values in `args` can be either strings corresponding to existing columns of a dataframe, or data arrays (lists, numpy arrays, pandas columns, series). Parameters ---------- args : OrderedDict arguments passed to the px function and subsequently modified constructor : graph_object trace class the trace type selected for this figure NrrrFzPCannot accept list of column references or list of columns for both `x` and `y`.TznData frame columns is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.r))Nr]rrOrgrhrirZ__x__Z__y__r])rrzlData frame index is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.r)rlr.rjrcsg|]}|kr|qSr<r<)r[r)rjr<r=rsmsz#build_dataframe..)rrr#unumberzLPlotly Express cannot process wide-form data with columns of different type.)Zid_varsZ value_varsrdrezWide-mode name-inference failure, likely due to a internal bug. Please report this to https://github.com/plotly/plotly.py/issues/new and we will try to replicate and fix it.r-)'rIrrrrKr_rrVrrrrrrTrQrSrFunnelZIndexrr[rBarrrrrNO_COLORrnrerarbZmeltrrGrbrr1r2)rXrRrNrfrUZno_xZno_yZwide_xZwide_yrcrdZwide_cross_namereZ hist2d_typesZwide_orientationZmissing_bar_dimZaxno_colorrgZ count_nameZ other_dimZwide_value_varsrariZv_dtypeZorient_vr<)rjr=build_dataframes"                $                rucCs|jt|jd}|}|t}t|jddj d}xJ|D]B}|j |}t|j dd}||d sBt d|j |qBWd||<t|j dddd}xNt|dd D]:\}}||d|kr|d|krt d |j |dd qWdS) N)rr)rrz*None entries cannot have not-None childrenrncSs d|S)Nrn)r)rr<r<r=rprqz-_check_dataframe_all_leaves..z4Non-leaves rows are not permitted in the dataframe zis not a leaf.)rrrTZisnullrrrZnonzeror@rrr-rapplyr`)rcZ df_sortedZ null_maskZ null_indicesZnull_row_indexrtrrZ row_stringsr<r<r=_check_dataframe_all_leavess&    rxcsB|d|dddd}t|dddd}g}x*|D]"}|d}||||<q>W|}dd}i}d}|d ryt|d |d <Wn$tk rtd |d YnX|d r|d |d kr|d d } |d | <| |d <|d n:d jkr d ndddtjDd<|d <d|<|d rt|d sv|}d}nfdd} | }|||d <tt j |} x | D]} | |kr||| <qWtt | t dddg} tj dddg| d} x&| D]} | |  | j | | <qWx2t|D]$\}}tj | jd}||d|}|}|| t|d<d|d<|| t|d<|t|dkr|d}xh|t|kr||| td|d|d<||| td|d|d<|d7}qW|djd|d<| rD|| || <| j|dd} q.W|d r|rd}x|| jkr|d7}qnW|d  t| |<| j|d} | |d<d|d<d|d<d|d <d|d!<|d r>|d"s|d g|d"<nHt|d"tr,|d"|d s>d#|d"|d <n|d"|d |S)$zU Build dataframe for sunburst or treemap when the path argument is provided. rr(NrvFZ _path_copycSs$|}t|dkr|dSdSdS)Nrrz(?))rr)runiquesr<r<r=aggfunc_discretes z5process_dataframe_hierarchy..aggfunc_discreterzDColumn `%s` of `df` could not be converted to a numerical data type.r.Z_sumrjrncSsg|] }t|qSr<)r)r[elr<r<r=rssz/process_dataframe_hierarchy..rsumTcstj|j|jfdS)N)weights)rZaveragelocr)r) count_colnamercr<r=aggfunc_continuoussz7process_dataframe_hierarchy..aggfunc_continuousrVr{r)rT/)Z ignore_indexZsort_color_if_discrete_color0)rr rrr')TN)rxrrKZ to_numericrrTrrrerG differencer_rrar`groupbyZaggr\r~rrrstriprrrr)rXr(Zdiscrete_colornew_pathrdZ new_col_namerzZagg_fZ aggfunc_colorZnew_value_col_namercolsruZ df_all_treesrrlevelZdf_treeZdfgjZ sort_col_namer<)rrcr=process_dataframe_hierarchys           &&    rc Csd|d<|ddks |ddkr(tdy0t|d|d}t|d|d}Wn ttfk rxtdYnX||d |d|d<|d|d <|d=|d|d <|d=|S) z8 Massage input for bar traces for px.timeline() TrrNrz#Both x_start and x_end are requiredrzCBoth x_start and x_end must refer to data convertible to datetimes.ztimedelta64[ms]rr)rrKZ to_datetimerSr)rXrrr<r<r=process_dataframe_timelineBs   rcs$fddttD}g}d}dkrNdrNdddd}dkrd krd krt|dn\drtddr|dd d <n.|tjtjgkr|dd d <n |dnxdks|tjkr|dnZ|tj tj gkr@|ddrJddkr,gd<ddn |dt d|kodo|tj tj gko|tjtjgkp d }nd }dkr|ddkr|ddkrddk }ddk } ddkrD|tj tjgkr| rD|sDdd<n.|tjtjtjtjgkrD|rD| sDdd<ddkr|r| rtdd} tdd} | r| sdd<| r| sdd<ddkrdd<|tj krH|r| rddkrdd<|d<d} d} | dkr| nd|d<| dkr*dn| |d<| dkr@dnd|d <d|d<|tjtjgkr|tjkrvd!nd"}||dkrddk rddkrddkrd#||<n(ddkrddkrd#||<||dkrd$||<|tjkr6d%dk r6ddkr6dd<|d<|tjtjgkrTd }d&|d'<d(krΈd(dkrd)krΈd)d#krtd*d+|d,<n:|tjtj tjtj gkrd(|d(<ntd(d+|d,<dkrd-d.rd/nd0|d1<n<|tjkr0dks|tjkr0d2d.r&d3nd0|d1<d4krLtd4d5|d6<d7krddkrhd8nd9}ddkr~d8nd9}d7|<d|< d8ddk s̈ d9ddk s̈ d:ddk rdd;<x"tD]}|kr||qWfd.rrrr9rr.r6r5TrFz marker.colorr-z line.colorr'r0z line.dashr/z marker.symbolrgrrrhrirfr|nbinsZnbinsxZnbinsyr,ZboxmodeZ violinmodergrouprrrr+rg?)r+rr/rz +markers+textrnrmarkersz+textZ line_shape)shaperrUrrr+facet_col_wrapcsg|]}t|qSr<)r)r[r)rXr<r=rss)direct_attrablesrImaxrrerrrrrrrXrrrr1r2rrrqrrrrrgroup_attrablesr5)rXrRrT layout_patchrSZ grouped_attrsr show_colorbarZhas_xZhas_yZx_is_continuousZy_is_continuousrgrrrZother_positionrgrouped_mappings trace_specsr<)rXr= infer_config[s                                        rcsd|kr in |dg}i}x|jD]}t|dkr>|f}||xx|D]ptkrN|d}krt|<n(x&|D]}|kr|qWt|jd|<qNWq(Wx.) r~groupsrrr6rrsortedrrr`)rXrKgroupedZ group_namesZ group_values group_nameryvalr<)rurrrr= get_orderingss,       rc- sR|pi}|pi}tt||tjtjgkrFddk rFt|dkr\tj}tt|||\}}}}dd|Dpt g}dj |dd} t || \} } g} g} x|D]}|j r|j dkrtd d fd d| |j D} |j d kr,td d fdd| |j D} x@| |j D]2}||jkr8|jt|jt|j|j|<q8WqWt|j}i}t}g}d}}d}xJ| D]@}| t|dkr|n|d}t}t}d}xnt|||D]^\}}}|t krt|}t|jts0t|||<|jr0t|||<|jdkr|}qWd|}||krht||<||}xl|D]b}|j|d} |jtj tj!tj"tj#tj$tj%tjtjgkr| j&||dko||kd|jtjtj'tj(tj)gkr| j&d|d|*|d| _+d| _,xt-|D]\}!}||!}||jkr\|jt|jt|j|j|<y|.| |j|Wnt/k rL||dkr|jtj'tj(tj)gkr|jdkrn||dkr|jtj)gkr|jdkr| j&t0|j|ddnL|jtj"tj#gkrF|jdkrF| j&dgt||j|gddddnYnX|j d krf|j|}"n(t12ddr|j3dkrd}"nd}"2d d}#|j dkr|j|}|#rd|d|#}"d|d|#}n|j3d!krd}nd}t4||"}|"dkr|"| _+t4||}|dkr|| _,qW|djtj5krh|jtj(krh| j6j7rh| j&t0| j6j7ddt8|||9|\}$}%| &|$|%dk r|:|9|%|d"d#<||krt0g|d$||<||d%:| qxWqWd&d|D}&t|&dkrt;|&fd'd(d)}&|r|tj%tj$gkr8d*nd}'d+pLddg}(t|'|'d0d1|d2<x$d3D]}*|*r|*||*<qWt0dd4|d5<|rd||d5d6<d7rd7|d6<n d8j?j@jAdkrd9d:i|d;<d<krNd<rNd8j?jBjCdkrNd=|d5d><tD||&||| | }+xT|&D]L},xD|,d%D]8} t| tjErqvtF| |+j?|+jG|| j+d| j,qvWqhW|+Ht|&dkr|&dd%ng|+I|d8krd8dk r|+jId8dd?t|dkr"|&ng|+_JtKL||+_MtN||+tO||+|+S)@Nr(ZtimelinecSsg|]}|jptqSr<)rKr6)r[rr<r<r=rs,szmake_figure..rF)sortrur,rcsg|]}t|qSr<)r)r[s)r!r<r=rs:srtr+csg|]}t|qSr<)r)r[r)r!r<r=rs=srrrnr*z, )r)Z legendgroup showlegendT)ZalignmentgroupZ offsetgroupr/r.)r.)rr)rr>Z showscalerrrrrrvZpx_fit_results)r?rr?cSsg|]}|qSr<r<)r[r#r<r<r=rsscsd|dS)Nr*r)r)r#)rXrr<r=rprqzmake_figure..)r\r range_colorr> make_figurer6Zcolor_continuous_midpoint)r)r>ZcmidZcminZcmaxZcolorbarr)r4r3)Z tracegroupgaplegendrtitler2rr"marginrZconstant itemsizing)r2 overwrite)PrBrurrrrrrrrr6rrrKrPrZrLrMrr rrZ get_grouprarrrrJrOrrrGrRrrrrrrrzr1r2rrJ _subplot_row _subplot_colr`rNrrrXrrUrrrr.rr~rrrZvalidate_coercermrrrrr init_figurerr rZ add_tracesrr%rKr_rFrr&)-rXrRrTrrrrrrKrZsorted_group_namesZsorted_group_values col_labels row_labelsmr subplot_typeZtrace_names_by_framer%Ztrendline_rowsrrZtrace_name_labelsrrrZ frame_namerur\Z trace_nameZ trace_namesrrorrrtrpatchr frame_listZcolorvarrZcolorscale_validatorrirGrr<)rXrr!r=rsf     *                                               $   rcsfddt|D}xb|D]Z}xT|dD]H} | jd} | jd} t| tjr`ddi|| | <q*d| ji|| | <q*WqWdg} dg|} |dkrt|d d r|d d ksd |kr|d rd }nd}|g|dd|g} d}n4|ddr|ddp d}n|ddpd}t|dd r||dd ksTd |krZ|d rZd }nd}|gdd|g} d}n|ddpd}nd}d}|dd}|r(dg|}x"t ||kr| dqWxJt|D]>}x6tD]*}||d|||||<qWqWt ||dd|r>gn t t ||rRgn||r^|ng||| | dd }x|jjD]}|jddqzW|S)Ncsg|]}igqSr<r<)r[r)rr<r=rsszinit_figure..r?rrdomaing?ZxyrFrr.gGz?gzG?g{Gz?rrZfacet_row_spacinggQ?gQ?rg{Gzt?Zfacet_col_spacingg{Gz?g?r-z bottom-left) ZrowsrspecsZ shared_xaxesZ shared_yaxesZ row_titlesZ column_titlesZsubplot_titleshorizontal_spacingvertical_spacing row_heights column_widthsZ start_cell)font)r|rrrrrrrXrrrr rrr annotationsrz)rXrrrrrrrrroZrow0Zcol0rrZ main_sizerrrZsubplot_labelsrrrrGZannotr<)rr=rsl       " 0  r)NN)UZplotly.graph_objsZ graph_objsrZ plotly.ioior; collectionsrrZ_special_inputsrrrZ_plotly_utils.basevalidatorsrZ plotly.colorsr r rZpandasrKZnumpyrZplotly.subplotsr r r rsrrIrZrenameable_group_attrablesrHrr0rrrqr2r1rrrrobjectr1r:rCrErHrIrQrZr`rermrrrrrrrrrrrr&r5r6rBrFrPrVrYr[r]rnrurxrrrrrrr<r<r<r=s   J   -E"  :VM  K[}) s