a 7&blk@sddlmZddlmZddlmZmZddl m Z m Z m Z ddl mZmZmZmZmZddlmZddlmZmZddlZddlZddlZddlmZmZm Z d Z!e"eeeeed Z gd gd gd gdZ#gdZ$gdZ%gdZ&e#e$e%e&Z'ej(ej)ej*ej+ej,ej-gZ.e.ej/ej0ej1g7Z.Gddde2Z3e3Z4[3da5ddZ6ddZ7edgdZ8edgdZ9ddZ:ddZ;d d!Zd&d'Z?d(d)Z@d*d+ZAd,d-ZBd.d/ZCd0d1ZDd2d3ZEd4d5ZFd6d7ZGd8d9ZHd:d;ZIdd?ZKd@dAZLdBdCZMdDdEZNdFdGZOdHdIZPdJdKZQdLdMZRdNdOZSdPdQZTdRdSZUdTdUZVdVdWZWdXdYZXdZd[ZYd\d]ZZdbd^d_Z[d`daZ\dS)cN) namedtuple OrderedDict) IdentityMapConstantRange)olslowessrolling expandingewm)ColorscaleValidator) qualitative sequential) make_subplots_set_trace_grid_reference_subplot_type_for_trace_typeZpx_no_color_constant)r r r r r) basexyzabcrthetasizex_startx_end) hover_nametextnamesvaluesparents wide_cross)idsZerror_xZ error_x_minusZerror_yZ error_y_minusZerror_z)Z error_z_minuslatlon locationsanimation_group) dimensions custom_data hover_datapath wide_variable)animation_frame facet_row facet_col line_group)colorsymbol line_dash pattern_shapec@s$eZdZgdZddZddZdS) PxDefaultstemplatewidthheightcolor_discrete_sequencecolor_discrete_mapcolor_continuous_scalesymbol_sequenceZ symbol_mapline_dash_sequenceZ line_dash_mappattern_shape_sequenceZpattern_shape_mapsize_maxcategory_orderslabelscCs |dSN)resetselfrI3lib/python3.9/site-packages/plotly/express/_core.py__init__BszPxDefaults.__init__cCs^d|_d|_d|_d|_i|_d|_d|_i|_d|_i|_ d|_ i|_ d|_ i|_ i|_dS)Nr8rGrIrIrJrFEszPxDefaults.resetN)__name__ __module__ __qualname__ __slots__rKrFrIrIrIrJr7/sr7cCs|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)tokenrIrIrJset_mapbox_access_token^srScCs|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)figrIrIrJget_trendline_resultsis rVMapping)show_in_trace_namegrouperval_mapsequenceupdatervariablefacet TraceSpec constructorattrs trace_patchmarginalcCs*z|d|WSty$|YS0dS)NrD) Exception)argscolumnrIrIrJ get_labels rhcCs<dd|dD}z ||WSty6|YS0dS)zInvert mapping. Find key corresponding to value column in dict args["labels"]. Returns `column` if the value does not exist. cSsi|]\}}||qSrIrI).0keyvaluerIrIrJ z invert_label..rDN)itemsre)rfrgZreversed_labelsrIrIrJ invert_labels   rocCs||jjdvS)NZifc)dtypekind)dfcol_namerIrIrJ_is_continuoussrtcCs`t||}}d|vr\|dks^|dkrrmzmake_mapping..)rXrYrZr[r]r\r^r0r1rrcSsg|]}|qSrIrIriirIrIrJ rmz make_mapping..rcSs|SrErIrrIrIrJrrmrowcol)rXr]rYrZr[r\r^.r3Zcolor_discretedashr5patternr6_mapidentityTZ _sequencecs|dg|iiS)Nr)updatejoinrZother_variablesparentr]rIrJrs)rWrangesplitrcopy)rfr]letterZvprefixarg_namerZrIrrJ make_mappingsX   rc$ sdvr dr ||jd}jp,i}d}d}jD]}|t|} |dkrȇfdd|D} fdd| D|d<jtj kr|dD]} t d d | d <qd |d <d|d<q<dur|dkr"d|vrt |d<||dd<d|dd<||dd<d|| <q|dkrFjtj krd|d<q|dkrjjtj krd |d<q|dkr0drdrt |ddg dkr|jdd} | dj} | dj}|jjtjkr|td}nF|jjtjkr<q|d?kr|||<d@|| <q|dAkr|||<| dAkrdBn| }dC||<n|dDkr|||<| dDkrdEn| }dF||<n|d>kr@|||<| d>kr2dGn| }dH||<nx|dIkrjtj)tj*tj+tj,tj-fvr||dJ<| dIkrdKn| }dL||<n |||<n|||<dM||| <q<jtj kr|dNvsjtj tj!fvr<|d/kr.cs"g|]\}}tt||dqS))r~r")dictrhrrfrIrJr&sT)matchesaxisz%{x}z%{xaxis.title.text}z%{y}z%{yaxis.title.text}rmarkerZareaZsizemodesizerefz%{marker.size} marginal_xry marginal_y trendlinerrrbyiʚ;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.trendline_optionsz/missing-data-handling failure in trendline codez%{y} (trend)errorZminusZ arrayminusarrayr+Z customdatarZ hovertextz%{hovertext}

r,rrz%%{customdata[%d]}r3 coloraxis1 coloraxisz%{z}color_is_continuousZcolorsz%{color}r=r<linez %%{%s.color}r)r%r(z %{location}r"rkz%{value}r#rz %{parent}idz%{id}r!rDr~z%{label}z%%{%s}rr}z%s}cSsg|]\}}|d|qS)=rI)rikrxrIrIrJrrmz
Z hovertemplatez)7appendilocrcrrbrZ iteritemsrarSplomr HistogramrZdropna sort_valuesr"rptypenpZ datetime64astypeintZobject_Zfloat64 ValueErrorZ logical_notZ logical_orZisnantrendline_functionsrh startswithendswith Histogram2dHistogram2dContour isinstancegetindexAttributeErrorKeyError ChoroplethChoroplethmapboxSunburstTreemapIciclePie Funnelarearrrrnrostrreplacepopr)$rfrZ trace_datamapping_labelsrrc fit_resultsZ hover_headerZ attr_nameZ attr_labelZdimsdZsorted_trace_datarrZ non_missingZtrendline_functionZy_outZerror_xyZarrZ hover_is_dictZcustomdata_colsrpositionZattr_label_colmappingcatZ colorableZ_labelZmapping_labels_copyrrxZk_args formatter_Z hover_linesrIrrJmake_trace_kwargss                                                                  rcCsntjttjttjttjttjttj t tj t tj t tj ttjti }tD] }t||<qD||vrj|||||dSrE)rZ Scatter3dconfigure_3d_axesZScatterternaryconfigure_ternary_axes Scatterpolarconfigure_polar_axesScatterpolarglZBarpolar Scattermapboxconfigure_mapboxr DensitymapboxZ Scattergeo configure_geor cartesiansconfigure_cartesian_axes)rfrarUordersZ configuratorsrrIrIrJconfigure_axes s  rcCsd|}d|}||vrP||rPd|d<||vrl||rldd||D|d<n||vrl||rl|||d<|||vrd|d <t|tjjr|||ntt||||d <dS) Nlog_range_logrcSsg|]}t|dqS mathr)rirrIrIrJr&rmz+set_cartesian_axis_opts..rr categoryorder categoryarray)rrlayoutZXAxislistreversed)rfrrrZlog_keyZ range_keyrIrIrJset_cartesian_axis_opts s   rc Csd|d|dfvrd|jd<t|j}t|jd}|jddD]}t||d |qB|jdd D]}t||d |qb|dr|jd d d d|d|djjjdur|j|ddk|d|djj jdur|j d|d|drD|j d d d d|d|djj jdur"|j |ddk|d|djjjdurD|jd|dt ||d d }|drp|j|dddn&t d|dD]}|j||ddq~t ||d d } |dr|j | dddn&t d|dD]} |j | d| dqd|vr|dr|j ddd|vr(|dr(|jddd t |d} |drjt d|ddD]}|j| d|dqR|drt d|ddD]} |j dd| dqdS)N histogramrroverlaybarmoderrrrrrFr)showticklabelsshowlineticksrrr9)showgridrT)rrrrr)rr) title_textrrlog_xrrlog_y)rrrx2)rrr)rr _grid_ref select_yaxesr select_xaxes update_yaxesyaxisrxaxis update_xaxesrrr) rfrUrnrowsncolsrry_titlerx_titlerZ matches_yrIrIrJ!configure_cartesian_marginal_axes3sZ           rcCs2d|vr|ds d|vr0|dr0t|||dSt||dd}|jddD]}|j|dt||d|qLt||dd}|jddD]&}d |vr|j|dt||d|qd |vr|d r|jd d d |vr|d r|jd d d |vr|jdd d|vr.|ddkr"|jddn |jdddS)Nrrrrrrrr is_timelinerrrrdateecdfmodervrxZtozero)Z rangemode)rrr rrr rr)rfrUrrrrrrIrIrJrxs4       rcCsD|jtt||ddtt||ddtt||ddddS)Nrrrr)ZaaxisZbaxisZcaxis)Zupdate_ternariesrrhrfrUrrIrIrJrs rcCstt|d|ddtd}dD]4\}}|||vr"d||d<|||||d<q"|d }|d rd |d <|d rdd|d D|d<n|d r|d |d<|dr|d|d<||dS)N directionZ start_angle)rZrotation) angularaxis radialaxis))rr)rrrrrrZlog_rrrZrange_rcSsg|]}t|dqSrrrirrIrIrJrrmz(configure_polar_axes..rZ range_thetaZsector)rZ update_polars)rfrUrpatchvarrrrIrIrJrs$     rcCsttt||ddtt||ddtt||ddd}dD]}||d}|d|rd |d <|d |rd d |d |D|d<n|d |r|d ||d<|||vrBd|d<||||d<qB||dS)Nrrrr)rrZzaxis)rrrrrrrrcSsg|]}t|dqSrrrrIrIrJrrmz%configure_3d_axes..rrrr)rrhZ update_scenes)rfrUrr rrrIrIrJrs"     rcCsf|d}|sHd|vrHd|vrHt|d|d|d|dd}|jt||d|dddS) Ncenterr&r'r)r&r'zoomZ mapbox_style)Z accesstokenr"r#Zstyle)rZmeanZupdate_mapboxesrQ)rfrUrr"rIrIrJrsrcCs2|j|d|d|d|dt|ddddS)Nr"scope fitboundsZbasemap_visible projectionr)r"r$r%Zvisibler&)Z update_geosrrrIrIrJrs rc sfddd|vr|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)r'Zeasing)framemodeZ fromcurrentZ transition)rScatter)r')rarIrJ frame_argss z0configure_animation_controls..frame_argsr/riz▶animaterfr~methodrz◼leftrF)rtFbuttonsg?righttop) r2rpadZ showactiverrxanchorryanchorprefixr<)rr1g?cs&g|]}|jgdg|jddqS)rr,r-rrif)r+rIrJrs z0configure_animation_controls..) Zactiver7r6Z currentvaluer5rrrZsteps)rframesrZ updatemenusrhZsliders)rfrarUrI)rar+rJconfigure_animation_controlssF     r>c CsB|tjtjfvrrd|vrr|ddksL|ddkrrt|ddkrr|ddurr|tjkrltj}d|vrr|d=ntj}t|||dg}dD]}d ||vr|d |rd}t|d krd nd |d krdndd}|d |dkr ttj|d |gtfd|d||d}n|d |dkr| d)d*d*kr>| t|||S)+NZ render_modeZwebglautorrr/rvrZ marginal_rZx1r rZy1Zy2)rrr?)opacitybingroupr`Zviolinrr,)Z scalegroupZboxT)ZnotchedZrugz line-ns-openz line-ew-openzrgba(255,255,255,0)r3allrZpointsr4)Z fillcolorrZ boxpointsZjitterZhoveronrrr>rtrendline_scoper)rr*rr Scatterglrr_rrViolinBoxrcrrmake_trendline_spec) rfrarbrcresultrrZaxis_mapZsymbolsZfirst_default_colorrIrIrJmake_trace_spec(s             rJcCsJt|tjkrtjntjdgtdddd}|drFt|dd|jd<|S)Nrlines)r)r`Ztrendline_color_overrider3r)r_rrEr*rrc)rfrarrIrIrJrHtsrHcCsdSNrrIrrIrIrJ one_groupsrOcCsltjD]&}||vr||durtt|||<q|ddur\tjjdurTtjj|d<nd|d<ztj|d|d<Wn&tytj |d|d<Yn0d|vr|ddur|djj j rdd|djj j D|d<|ddurt j |d<d|vr<|ddur$|djj r$|djj |d<|ddurcSsg|] }|dqS)rrIrrIrIrJrsz)apply_default_cascade..r<r?cSsg|] }|jjqSrI)rr4riscatterrIrIrJrs)ZcircleZdiamondZsquarerZcrossr@cSsg|] }|jjqSrI)rrrPrIrIrJrs)ZsoliddotrZlongdashZdashdotZ longdashdotrAcSsg|]}|jjjqSrI)rrshape)ribarrIrIrJrs)r/\r+r)defaultsrPgetattrpioZ templatesdefaultrerrZTemplate colorscalerZViridisZcolorwayrZD3datarQanyrT)rfZparamrIrIrJapply_default_cascadesb                   r_cCs ||vr |Std||fdS)NzaA name conflict was encountered for argument '%s'. A column or index with name '%s' is ambiguous.) NameError) field_namereserved_namesrIrIrJ_check_name_not_reservedsrccCs|d}t}|D]}|tvr q|tvr0||n||g}|durDq|D]}|durXqHqHt|trn||qHt|tjr|j}|rt ||r|||u}|r||qH||j urH|jdurH||jqHq|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)rfrrrbfieldr!argrZin_dfrIrIrJ_get_reserved_col_namess,     rmc Cs|dust|tst|tr dSt|tjr0dSz t|WntyPYdS0|D]V}t|tsnt|tr|dus||jvrdSqVz t|WqVtyYdS0qVdS)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)rrrrh MultiIndexiter TypeErrorcolumns)df_inputrlrrIrIrJ _is_col_lists"      rscCs>t|ts2t|tr t|dks2t|ts2|dur6dSdSdS)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. r NFT)rrtuplerboolrNrIrIrJ_isinstance_listlikes rvcCs(|dur$||jvs||vr$d|}q|S)Nrrq)rrrsZextrarIrIrJ_escape_col_name$s rxcCst|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)Zdrop)rhri reset_indexrNrIrIrJto_unindexed_series*srzc s$|d}|du}t}t}t}t} |r6t|nt} d|vrn|ddurn|s^tdn||j||j<d|vo|dot|dt} | r|dD]} t |d| rd|d| f|d| <t|d| t s|d| df|d| <|r|d| ddur| |vrtd| qt D]‰t vr>| gn| } | dus"| dgurbq"t vrrgnfd d tt| D}tt| |D]F\}\}}t|}|durqd}t|tjrtd |t|tst|tr:t|jdur t|jn|| }t|tr,|j||<n ||nBt|tsRt|d sdkr| r|dt|ddurt|}|d|d}|rt||krtd |t|tt|j|ft|||<n|std|n||jvrF|r|||fvrqn2d|tt|j|f}|dkr<|d7}t|nX|rt|||krtd|t||tt|j|fnt|}t||||<n|r"t|dr"||jur|jdus|j|vrd}n|j}t||||g}n.|jdur"|j|vr"|||jur"|j}|dur6t|| }|rlt||krltd|t|tt|j|ft||t|<|dusJdt vrt||<n"t|trnt|||<dkr| t|qq"|D]}tt|||<q|D]}||||<q|| 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)rrrrarIrJrqrmz/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.)!rh DataFramerrrdrmrrqrrvrtrerfrrr enumerateziprnrprrrcr~rrkrrjrzrrrxrg)rf wide_modevar_name value_namerr df_provided df_outputZ constantsZranges wide_id_varsrbZhover_data_is_dictrZ argument_listZ field_listrargumentrklengthrsZ real_argumenterr_msgrIr|rJprocess_args_into_dataframe4s&                         rcs|D]B}|tvr||durt||tr6t||n t||||<q|ddu}|rzt|dtjszt|d|d<|d}|ddu}|ddu}|rdn t||d}|rdn t||d}d} d} d} d} tj tj g} |tj kpd|v}|t vr|r|rt d|r|r|rd} t|jtjrDtd t|j|d <|jj} | d vsn| |vrrd } |tjkr|d pd}n|d pd}||d <d|d<n||krd} |r|dn|d|d <|r|d |jur|jj} t|d tjrt|d |d <| d vs4|r8| |vr8d } |rN|rHdnd}n|rXdnd}d||rjdnd<d|d<|s|s|rdnd} | rt|dg} t|| g} d}|tjtjtjg| vr|s| sJ||krJdD]b}||dur|r|jnt||<|tjkr"|}n$|d dur|dkr>dnd|d <q| r| dur||kr|d dur|r|dnd|d <|rt|jtjrtd|j|d<ntt|d| | gd|d<d}t|dtkr|dtkrd}d|d<t|| | | \}t|d| | g}| s|r|tjkr|dkrBdnd}t|||sh|||<d||<n$|d dur|dkrdnd|d <|| vr|d =| rfdd|d D}|d =| dkr|d} n| dkr|d} n|d} |d=d}|D]F}||jj}|dvrdn|}|dur.|}n||krt dq|j || | d }t!|jt!t"|jksxJd!|| #t|| <|dk}|r| ||rdnd<| ||rdnd<|dp| |d<nB|tjtjg| vrH| ||rdnd<| ||rdnd<|tj kr*|dp$| |d<d"|vr|d"p@| |d"<n|tjkrt|| r| ||rndnd<| ||rdnd<|dp| |d<n>| ||rdnd<|||rdnd<d||<|dp| |d<n<|tj$tj%fvr| p| ||rdnd<| ||rdnd<|r|tjkr|ddurL|ddurLd#|d$<nP|ddur|d|d$<d|d <||d<d||<n d|d$<d|d <||d<d||<|rd|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 NrrrFrzPCannot 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.)Nrkrr]rvrwrxr$Z__x__Z__y__rkrzlData frame index is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.r)r~r3ryrcsg|]}|vr|qSrIrI)rirrrIrJrrmz#build_dataframe..)rr<uZnumberzLPlotly Express cannot process wide-form data with columns of different type.)Zid_varsZ value_varsrrzWide-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.r2r{ru)&rfrrrrhr~rrsrrrrrrrqrnrprFunnelZIndexrxr*BarrrrrNO_COLORrrtrprqZmeltrrdrrFrG)rfrarkrrrZno_xZno_yZwide_xZwide_yrrZwide_cross_namerZ hist2d_typesZhist1d_orientationZwide_orientationZmissing_bar_dimaxZno_colorrZ count_nameZ other_dimZwide_value_varsrprxZv_dtypeZorient_vrIrrJbuild_dataframes\                "               rcCs|jt|jd}|}|t}t|jddj d}|D]B}|j |}t|j dd}||d s@t d|j |q@d||<t|j dddd}t|dd D]:\}}||d|vr|d|vrt d |j |dd qdS) Nrr)rrz*None entries cannot have not-None childrenrcSs d|SrM)rrNrIrIrJrrmz-_check_dataframe_all_leaves..z4Non-leaves rows are not permitted in the dataframe zis not a leaf.)rrrqZisnullrrrZnonzeror^r"rrCrZapplyr)rrZ df_sortedZ null_maskZ null_indicesZnull_row_indexrrZ row_stringsrIrIrJ_check_dataframe_all_leavess*   rcs|d|dddd}t|dddd}g}|D]"}|d}||||<q<|}dd}i}d}|d rzt|d |d <Wn"tytd |d Yn0|d r|d |d kr|d d } |d | <| |d <|d n:d jvrd ndddtjDd<|d <d|<|d rt|d sl|}d}nfdd} | }|||d <tt j |} | D]} | |vr||| <qtt | t gd} tj gd| d} | D]} | |  | j | | <qt|D] \}}tj | jd}||d|}|}|| t|d<d|d<|| t|d<|t|dkr|d}|t|kr||| td|d|d<||| td|d|d<|d7}q|djd|d<| r&|| || <| j|dd} q|d r|rd}|| jvrd|d7}qL|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 sd$|d#|d <n|d#|d |S)%z^ Build dataframe for sunburst, treemap, or icicle when the path argument is provided. rr-NrFZ _path_copycSs$|}t|dkr|dSdSdS)Nrrz(?))rr)runiquesrIrIrJaggfunc_discretes z5process_dataframe_hierarchy..aggfunc_discreter"zDColumn `%s` of `df` could not be converted to a numerical data type.r3Z_sumryrcSsg|] }t|qSrIr{)rielrIrIrJr2rmz/process_dataframe_hierarchy..rr{Tcstj|j|jfdS)N)Zweights)rZaveragelocrrNZ count_colnamerrrIrJaggfunc_continuous?sz7process_dataframe_hierarchy..aggfunc_continuous)rDrrrwrDrrrU)Z ignore_indexZsort_color_if_discrete_color0rr%r!r#r,)TN)rrrhZ to_numericrrqrrrtrd differencer~rrprgroupbyZaggryrrrrstriprrrr)rfr-Zdiscrete_colornew_pathrsZ new_col_namerZagg_fZ aggfunc_colorZnew_value_col_namercolsrZ df_all_treesrlevelZdf_treeZdfgjZ sort_col_namerIrrJprocess_dataframe_hierarchys           &       rc Csd|d<|ddus |ddur(tdz0t|d|d}t|d|d}WnttfyvtdYn0||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)rrhZ to_datetimerpr)rfrrrIrIrJprocess_dataframe_timelines    rcsfddttD}g}d}dvrNdrNdddd}dvrd vrd vrt|dn`drtddr|dd d <n2|tjtjtjfvr|dd d <n |dnxdvs|tj kr|dnZ|tj tj fvrD|ddrNddur0gd<ddn |dt d|vodo|tj tj fvo|tjtjtjfvp d }nd }dvr|ddvr|ddvr|ddvrddu}ddu} ddur`|tjtjfvr2| r`|s`dd<n.|tjtjtjtjfvr`|r`| s`dd<ddur|r| rtdd} tdd} | r| sdd<| r| sdd<ddurdd<|tjkrd|r| rddurdd<|d<d} d} | dkr0| nd|d <| dkrFdn| |d!<| dkr\dnd|d"<d|d<|tjtjfvr|tjkrd#nd$}||durddurddkr܈ddkrd%||<n(ddkrddkrd%||<||durd&||<|tjkrRd'durRddurRdd<|d< d(d d ur|tjtj fvr|d'}n(|tjkrddkrdnd}nd)}d(d urd*|d+|d,<nd*|d-d(d+|d,<|tjtjfvrd }d.|d/<d0vrxd0dur>d1vrxd1d%krxtd2d3|d4<n:|tjtj tjtj fvrfd0|d0<ntd0d3|d4<dvsdvrt} d5d r|d5 d6sΈ dsΈ d7r|d7 d6r|d6t|dkr|d5d8||d9<n<|tjkrRdvs8|tjkrRd7d6rHd:nd;|d9<d<vrptd<d=|d><n*d?vrtd?d@krdAndBd=|d><dCvrֈdD|dD<tdCdEsȈdCndCj|dC<dFvr ddkrdGndH}ddkrdGndH}dF|<d|< dIdurH dHdurHddH< dJdurp dGdurpddG< dGdus dHdus dJdurddK<dLvrdLdurdLtvrt dMdLt!fdNvrdNdurtdN<dOvrD dOddPvr8t dQdRdOdOdS<t"D]}|vrH||qHfdTd|D}t#|||}||||fS)UNcsg|]}|vr|qSrIrI)rirrrIrJrrmz infer_config..rrrrBr r3r>r<TrFz marker.colorr2z line.colorr,r5z line.dashr4z marker.symbolr6zmarker.pattern.shapervrrrwrxrur{nbinsZnbinsxZnbinsyrBZboxmodeZ violinmodergrouprZ text_autorkz%{rZ texttemplate:rrrArr@)rArrKr ZmarkersrWr)z+textrZ line_shape)rSrrrZvhZhvZgeojsonZ featureidkey__geo_interface__rdrrr1r0facet_col_wraprz,Value '%s' for `trendline` must be one of %srecdfnorm)Nr}r|z<`ecdfnorm` must be one of None, 'percent' or 'probability'. z'%s' was provided.rzcsg|]}t|qSrI)r)rirrrIrJrirm)$direct_attrablesrfmaxrrtrrrrrrrrurrr*rFrGrrrrrrdrgrrrrrjrrrkeysgroup_attrablesrJ)rfrarc layout_patchrbZ grouped_attrsr show_colorbarZhas_xZhas_yZx_is_continuousZy_is_continuousrvrr)rZmodesrZother_positionrgrouped_mappings trace_specsrIrrJ infer_configs^                                      $                       rcsd|vr in |d|D]Ptkrt|d}vrN|<qttt|<qg}|jD] }t|dkr|f}||qxt tt |D](\tkrt |fddd}q|fS)a `orders` is the user-supplied ordering with the remaining data-frame-supplied ordering appended if the column is used for grouping. It includes anything the user gave, for any variable, including values not present in the dataset. It's a dict where the keys are e.g. "x" or "color" `sorted_group_names` is the set of groups, ordered by the order above. It's a list of tuples like [("value1", ""), ("value2", "")] where each tuple contains the name of a single dimension-group rCrrcs&|vr"|SdS)Nrr)grrrrIrJrrmzget_orderings..rj) rrOrrrfromkeysgroupsrrrrsorted)rfrYgroupedrsorted_group_names group_namerIrrJ get_orderingsps&      rc4 sJ |pi}|pi}tt||tjtjtjfvrJddurJt|dkr`tj}tt |||\}}}}dd|Dpt g}dj |dd} t || \} g} g} d} }|D]}|j vr|jd |jd <q|j }|jd kr"td d fdd|D} t| }|jdkrZtdd fdd|D} t| } |D]2}||jvr^|jt|jt|j|j|<q^qt|j}i}t}g}d}dd }| D]}| t|dkr|n|d }t}t}d }t|||D]^\}}}|t krt|}t|jtsPt|||<|jrPt|||<|jdkr|}qd|}||vrt ||<||}|D] } | j!|d}!| j!tj"tj#tj$tj%tj&tj'tjtjtjf vr|!j(||d ko||vd| j!tjtj)tj*tj+fvr|!j(d|d|,|d|!_-d|!_.t/|D]\}"}||"}z|0|!|j|Wnt1yP| |d kr| j!tj)tj*fvr|jdvsL| j!tj+fvr|jdvrn| |d kr| j!tj+fvr|jdkr|!j(t2|j|ddnL| j!tj$tj%fvrJ|jdkrJ|!j(dgt||j|gddddnYn0|jdkrj|j|}#n&d dur| j3d!krd}#nd}#|jd kr|j|}|rd|d|}#d|d|}n| j3d"krd}nd}|#dkr|#|!_-|dkr8||!_.q8|d j!tj4krH| j!tj*krH|!j5j6rH|!j(t2|!j5j6ddd#vr@d$d%krhd!nd"}$d$d&krd!nd"}%d#d'd(k}&|j7|$|&d)}||%8}'||%9||%<|&s|j7|$dd)}d#d'd*kr|'||%||%<d+d,kr||%|'||%<n"d+d-kr@d.||%|'||%<t:| ||;|\}(})|!(|(|)dur|<|;|)|d/d0<||vrt2g|d1||<||d2<|!qqd3d|D}*t|*dkrt=|*fd4d5d6}*|rj|tj'tj&fvr d7nd}+d8p ddg},t>d9d:}-t2|-?d;d<|,d |,dt2t@|+|+d=d>|d?<d@D]}.|.rn|.||.<qnt2d dA|dB<|rd||dBdC<dDrʈdD|dC<n dEjAjBjCdurdFdGi|dH<dIvrdIrdEjAjDjEdurdJ|dBdK<|r.rF)sortrrrrr1rcsg|]}t|qSrIr{risr8rIrJrrmrr0csg|]}t|qSrIr{rrrIrJrrmrr/z, r:) legendgroup showlegendT)ZalignmentgroupZ offsetgroup)r4rr)r4rr3rL)rr )rr\Z showscalerrrrrrvrxrwZstandardr)r ascendingZ complementaryrr|r}gY@rpx_fit_results)r]rr]cSsg|]}|qSrIrIr;rIrIrJrUrmcsd|dS)Nr/rr)r<)rfrrIrJrXrmzmake_figure..rr range_colorr\ make_figurer>Zcolor_continuous_midpointr)r\ZcmidZcminZcmaxZcolorbarr)r;r:)Z tracegroupgaplegendrtitler9r1r9marginrZconstant itemsizingr)r9Z overwriterrDrZoverallzOverall Trendline)rrrrrC)rrZexclude_empty_subplots)selectorr)r)Yr_rrrrrrrrrrOrrrYr[rZr^rhrrrrrZ get_grouprrrrrXr]rr"rdrarrrrrrrrFrGrrgZ _subplot_rowZ _subplot_colrr\rrrdrrr3rr{Zcumsumrrrrr Zvalidate_coercerrrr1rrrZceilmin init_figurerrr Z add_tracesZ update_layoutr=rHrcZ add_traceZ update_tracesrhr~rTrr>)4rfrarcrrrrrrYrr col_labels row_labelsrrmZ sorted_valuesval subplot_typeZtrace_names_by_framer=Ztrendline_rowsZtrace_name_labelsrrrrZ frame_namerrjZ trace_nameZ trace_namesrrrrrr!rZ group_sumr r frame_listZcolorvarrZcolorscale_validatorrxrUr(Ztrendline_specZtrendline_traceZ next_colorrI)rfrr8rJrs        &                                               $        rcsfddt|D}dg}dg|} |dd} dkr|ddur|ddksld |vrr|d rrd } nd } | g|d d | g} d } n"| r|dpd} n|dpd} |ddur |ddksd |vr|d rd } nd } | gd d | g}d} n|dpd} n8| r4|dp0d} n|dpBd} |dpRd} | rdg|}t||kr|dqht|D]:}tD]*}||d |||||<qqdd}zLt||dd| rgn tt|| rgn|| r |ng| | | |dd }Wn>ty^}z$||dd||ddWYd}~n d}~00|jj D]}|j ddqh|S)Ncs g|]}tpddgqS)domainr)r)rirrrrIrJrrmzinit_figure..g?rrZxyrrr3gGz?gzG?rg{Gz?Zfacet_row_spacinggQ?gQ?rg{Gzt?Zfacet_col_spacingg{Gz?cSs6d|f|jdvr2|jddj|df|_|dS)z Translates the spacing errors thrown by the underlying make_subplots routine into one that describes an argument adjustable through px. z %s spacingrz5 Use the {facet_arg} argument to adjust this spacing.) facet_argN)rfformat)errrIrIrJ_spacing_error_translatorsz.init_figure.._spacing_error_translatorrCz bottom-left) ZrowsrspecsZ shared_xaxesZ shared_yaxesZ row_titlesZ column_titlesZsubplot_titleshorizontal_spacingvertical_spacing row_heights column_widthsZ start_cellZ HorizontalZVertical)Zfont) rrrrrrrrr annotationsr)rfrrrrrrrrrrZ main_sizerrZsubplot_labelsrrrrUrZannotrIrrJrsn      ,    " r)NN)]Zplotly.graph_objsZ graph_objsrZ plotly.ioiorZ collectionsrrZ_special_inputsrrrrrr r r r Z_plotly_utils.basevalidatorsr Z plotly.colorsrrrZpandasrhZnumpyrZplotly.subplotsrrrrrrrfrZrenameable_group_attrablesrer*rErrrGrFrrrrobjectr7rXrQrSrVrWr_rhrortrrrrrrrrrrrrr>rJrHrOr_rcrmrsrvrxrzrrrrrrrrrrIrIrIrJs    (   &2E& :LL  Kn}X$ *