U SZf @sddlmZddlmZddlmZmZddl m Z m Z m Z ddl mZmZmZmZmZddlmZddlmZmZddlZddlmZddlZddlZdd lm Z m!Z!m"Z"e#ej$e#d kZ%d Z&e'eeeeed Z d dddddddddddg ddddddgdd d!d"d#d$gd%d&d'd(d)gZ(d*d+d,d-d.gZ)d/d0d1d2gZ*d3d4d5d6gZ+e(e)e*e+Z,ej-ej.ej/ej0ej1ej2gZ3e3ej4ej5ej6g7Z3Gd7d8d8e7Z8e8Z9[8da:d9d:Z;d;d<Zd?d@dAdBdCdDgZ=edEdFdGdHdIgZ>dJdKZ?dLdMZ@dNdOZAdPdQZBdRdSZCdTdUZDdVdWZEdXdYZFdZd[ZGd\d]ZHd^d_ZId`daZJdbdcZKdddeZLdfdgZMdhdiZNdjdkZOdldmZPdndoZQdpdqZRdrdsZSdtduZTdvdwZUdxdyZVdzd{ZWd|d}ZXdd~dZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ddZ`ddZadddZbddZcdS)N) namedtuple OrderedDict) IdentityMapConstantRange)olslowessrolling expandingewm)ColorscaleValidator) qualitative sequential)version) make_subplots_set_trace_grid_reference_subplot_type_for_trace_typez2.2.0Zpx_no_color_constant)r r r r rbasexyzabcrthetasizex_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_dash pattern_shapec@s>eZdZddddddddd d d d d ddgZddZddZdS) PxDefaultstemplatewidthheightcolor_discrete_sequencecolor_discrete_mapcolor_continuous_scalesymbol_sequence symbol_mapline_dash_sequence line_dash_mappattern_shape_sequencepattern_shape_mapsize_maxcategory_orderslabelscCs |dSN)resetselfrL8/tmp/pip-unpacked-wheel-5ksk5baj/plotly/express/_core.py__init__EszPxDefaults.__init__cCs^d|_d|_d|_d|_i|_d|_d|_i|_d|_i|_ d|_ i|_ d|_ i|_ i|_dS)N)r9r:r;r<r=r>r?r@rArBrCrDrErFrGrJrLrLrMrIHszPxDefaults.resetN)__name__ __module__ __qualname__ __slots__rNrIrLrLrLrMr82s$r8cCs|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)tokenrLrLrMset_mapbox_access_tokenasrVcCs|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)figrLrLrMget_trendline_resultsls rYMappingshow_in_trace_namegrouperval_mapsequenceupdatervariablefacet TraceSpec constructorattrs trace_patchmarginalcCs,z|d|WStk r&|YSXdS)NrG) Exception)argscolumnrLrLrM get_labelsrjcCs>dd|dD}z ||WStk r8|YSXdS)zInvert mapping. Find key corresponding to value column in dict args["labels"]. Returns `column` if the value does not exist. cSsi|]\}}||qSrLrL).0keyvaluerLrLrM sz invert_label..rGN)itemsrg)rhriZreversed_labelsrLrLrM invert_labels  rpcCs||jjdkS)NZifc)dtypekind)dfcol_namerLrLrM_is_continuoussrucCs`t||}}d|kr\|dks^|dkrzmake_mapping..)r[r\r]r^r`r_rar1r2rrcSsg|]}|qSrLrLrkirLrLrM sz make_mapping..rcSs|SrHrLrrLrLrMrrrowcol)r[r`r\r]r^r_ra.r4Zcolor_discretedashr6)patternshaper7_mapidentityTZ _sequencecs|dg|iiS)Nr)updatejoinrZother_variablesparentr`rLrMrs)rZrangesplitrcopy)rhr`letterZvprefixarg_namer]rLrrM make_mappingsX   rc$ sdkr(dr(t||jddg}jp4i}d}d}jD]}|t|} |dkrЇfdd|D} fdd| D|d<jt j kr|dD]} t d d | d <qd |d <d|d<qDdk r|dkr*d|krt |d<||dd<d|dd<||dd<d|| <q |dkrNjt j krd|d<q |dkrrjt j krd |d<q |dkr>dr̈drt |ddgdkr|jdd} | dj} | dj}|jjtjkr|tjd}nH|jjtjkrHz|tj}Wn&tk rFtddYnX| jjtjkrz| tj} Wntk rtdYnXttt| t|}| d||d<t}|d | d|| dd|\}}}t |t |dkstd!||d<d |td<d"|td<q | d#r|dd$}|!d%rfd&nd'}||kr|i||<||||<q |d(krt d)kr||d*<q |d+krjt j t j"t j#fkr||d,<|dkrd-}q |d.krjt j t j"t j#fkrt$t }%d(p2g}D]}|rP|sPq8|%d%d%d/%d0fkr~q8zd(&|}Wn.tt't(fk rt |}|)|YnXt|d}d1|||<q8t |d)kr|||d*<q |d2krjt j*t j+t j,fkr<||d/<d3|d4<d5|| <q̈jt j-t j.t j/t j0t j1fkr&d|krpt |d<%d6r||dd7<d3|dd4<d8|| <ng|dd7<d9dk rʈd9}ni}|D]L}|%|dkr d:t |t d:||<|dd7)||qnXd}jt j2t j3fkrBd;}||krVt ||<|||d2<d3||d4<d<||| <q |d=kr||d><q |d?kr|||<d@|| <q |dAkr|||<| dAkrdBn| }dC||<n|dDkr"|||<| dDkrdEn| }dF||<n|d>krT|||<| d>krFdGn| }dH||<nx|dIkrjt j-t j.t j/t j0t j1fkr||dJ<| dIkrdKn| }dL||<n |||<n|||<dM||| <qDjt j kr|dNksjt j"t j#fkrD|d/krDdM||| <qDjt j3t j2fkrt4|}d.rt$d.t r|D]d\}}t5|} | d.krLd.| d)}!|!rt$|!t6r|7dOdP|!||<n |8|}"qLdQd|D}#|dR9|#|dS<|dSdT7<||fS)Ua: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_closeNrrr+csvg|]n\}}r|krjtjks2td|rjtjksjdk rN|ksjtd|dkr||fqS) data_frameNZdimensions_max_cardinality)rcgo ParcoordsruParcatslenuniquerknamerirh attr_value trace_specrLrMrs  z%make_trace_kwargs..cs"g|]\}}tt||dqS))rr#)dictrjrrhrLrMr)sT)matchesaxisz%{x}z%{xaxis.title.text}z%{y}z%{yaxis.title.text}rmarkerareaZsizemodesizerefz%{marker.size} marginal_xrz marginal_y trendlinerrbyiʚ;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)errorminusZ arrayminusarrayr,rZ customdatar Z hovertextz%{hovertext}

r-rrz%%{customdata[%d]}r4 coloraxis1 coloraxisz%{z}color_is_continuouscolorsz%{color}r=r<linez %%{%s.color}r*r&r)z %{location}r#rmz%{value}r$rz %{parent}idz%{id}r"rGrz%{label}z%%{%s}rr}z%s}cSsg|]\}}|d|qS)=rL)rkkryrLrLrMrsz
Z hovertemplatez):pdconcatilocrerrdrrorcrSplomr HistogramrZdropna sort_valuesr#rqtypenpZ datetime64astypeZint64Zobject_Zfloat64 ValueErrorZ logical_not logical_orisnantrendline_functionsAssertionErrorrj startswithendswith Histogram2dHistogram2dContour isinstancegetindexAttributeErrorKeyErrorappend Choropleth ChoroplethmapChoroplethmapboxSunburstTreemapIciclePie Funnelarearrrrpstrreplacepopr)$rhrZ trace_datamapping_labelsrre fit_resultsZ hover_header attr_nameZ attr_labelZdimsdZsorted_trace_datarrZ non_missingZtrendline_functionZy_outZerror_xyZarrZ hover_is_dictZcustomdata_colsrpositionZattr_label_colmappingcatZ colorableZ_labelZmapping_labels_copyrryZk_args formatter_Z hover_linesrLrrMmake_trace_kwargss                                                                  rcCstjttjttjttjttjttj t tj t tj t tj ttjttjttjttjti }tD] }t||<qV||kr||||||dSrH)rZ Scatter3dconfigure_3d_axesZScatterternaryconfigure_ternary_axes Scatterpolarconfigure_polar_axesScatterpolarglZBarpolar Scattermap configure_mapr Densitymap Scattermapboxconfigure_mapboxr DensitymapboxZ Scattergeo configure_geor cartesiansconfigure_cartesian_axes)rhrcrXordersZ configuratorsrrLrLrMconfigure_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)rkrrLrLrMr0sz+set_cartesian_axis_opts..rr categoryorder categoryarray)rrlayoutZXAxislistreversed)rhrrrZlog_keyZ range_keyrLrLrMset_cartesian_axis_opts*s   r c Csd|d|dfkrd|jd<t|j}t|jd}|jddD]}t||d |qB|jdd D]}t||d |qb|dr|jd d d d|d|djjjdkr|j|ddk|d|djj jdkr|j d|d|drD|j d d d d|d|djj jdkr"|j |ddk|d|djjjdkrD|jd|dt ||d d }|drp|j|dddn&t d|dD]}|j||ddq~t ||d d } |dr|j | dddn&t d|dD]} |j | d| dqd|kr|dr|j ddd|kr(|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)r r _grid_ref select_yaxesr  select_xaxes update_yaxesyaxisrxaxis update_xaxesrrr) rhrXrnrowsncolsr!r"y_titlerx_titlerZ matches_yrLrLrM!configure_cartesian_marginal_axes=sj         r(cCs2d|kr|ds d|kr0|dr0t|||dSt||dd}|jddD]}|j|dt||d|qLt||dd}|jddD]&}d |kr|j|dt||d|qd |kr|d r|jd d d |kr|d r|jd d d |kr|jdd d|kr.|ddkr"|jddn |jdddS)Nrrrrrrrr is_timelinerrrrdateecdfmoderwryZtozero)Z rangemode)r(rrrr rr#r )rhrXrr&r!r'r"rLrLrMrs4       rcCsD|jtt||ddtt||ddtt||ddddS)Nrr)rr)ZaaxisZbaxisZcaxis)Zupdate_ternariesrrjrhrXrrLrLrMrs rcCstt|d|ddtd}dD]4\}}|||kr"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)r.Zrotation) angularaxis radialaxis))rr0)rr/rrr r0Zlog_rrrZrange_rcSsg|]}t|dqSrrrkrrLrLrMrsz(configure_polar_axes..rZ range_thetaZsector)rZ update_polars)rhrXrpatchvarrr0rLrLrMrs$     rcCsttt||ddtt||ddtt||ddd}dD]}||d}|d|rd |d <|d |rd d |d |D|d<n|d |r|d ||d<|||krBd|d<||||d<qB||dS)Nrr)rr)r"r!Zzaxis)rrrrrrrrcSsg|]}t|dqSrrr1rLrLrMrsz%configure_3d_axes..rrrr )rrjZ update_scenes)rhrXrr2rrrLrLrMrs"     rcCsf|d}|sHd|krHd|krHt|d|d|d|dd}|jt||d|dddS) Ncenterr'r(rr'r(zoomZ mapbox_style)Z accesstokenr4r6style)rmeanZupdate_mapboxesrTrhrXrr4rLrLrMrsrcCsd|d}|sHd|krHd|krHt|d|d|d|dd}|j||d|dddS) Nr4r'r(rr5r6Z map_style)r4r6r7)rr8Z update_mapsr9rLrLrMrsrcCs2|j|d|d|d|dt|ddddS)Nr4scope fitboundsZbasemap_visible projectionr)r4r:r;Zvisibler<)Z update_geosrr-rLrLrMrs 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)r=Zeasing)framemodeZ fromcurrentZ transition)rScatter)r=)rcrLrM frame_argss z0configure_animation_controls..frame_argsr0riz▶animaterhrmethodrz◼leftrF)rtFbuttonsg?righttop) rHr.padZ showactiverrxanchorryanchorprefixr<)rrGg?cs&g|]}|jgdg|jddqS)rrBrCrrkf)rArLrMr3s z0configure_animation_controls..) activerMrLZ currentvaluerKrrrZsteps)rframesr Z updatemenusrjZsliders)rhrcrXrL)rcrArMconfigure_animation_controlssH     rUc CsP|tjtjfkrd|kr|ddksZ|ddkrt|ddkr|ddkr|ddkr|tjkrztj}d |kr|d =ntj}t|||dg}d D]}d ||kr|d |rd}t|d krd nd|dkrdndd}|d |dkrttj |d |gtfd|d||d}n|d |dkrJttj |ddgt|d|d}n|d |dkr|ttj |ddgtdd|d}nP|d |dkrdd d }ttj |ddgtd!d"d!id#d$d%d&||id'|d}d"|ksd"|krd(|j krt|j d(<|d)d$} | |j d(d"<| |q|d*rL|d+d,d,krL| t|||S)-NZ render_modeZwebglautorr line_shapeZspliner0rwrZ marginal_rx1rry1y2)r"r!r?)opacitybingrouprcrdrerfZviolinr r-)Z scalegroupZboxT)ZnotchedZrugz line-ns-openz line-ew-openzrgba(255,255,255,0)r4allrZpointsr5)Z fillcolorrZ boxpointsjitterZhoveronrrr>rtrendline_scoper)rr@rrr ScatterglrrbrrViolinBoxrermake_trendline_spec) rhrcrdreresultrrZaxis_mapsymbolsZfirst_default_colorrLrLrMmake_trace_spec?s              rhcCsJt|tjkrtjntjdgtdddd}|drFt|dd|jd<|S)Nrlines)r?r^Ztrendline_color_overrider4r)rbrrbr@rre)rhrcrrLrLrMresrecCsdSNrrLrrLrLrM one_groupsrmcCstjD]&}||kr||dkrtt|||<q|ddkr\tjjdk rTtjj|d<nd|d<ztj|d|d<Wn(tk rtj |d|d<YnXd|kr|ddkr|djj j rdd|djj j D|d<|ddkrt j |d<d|kr>|ddkr&|djj r&|djj |d<|ddkr>tj|d<d|kr|ddkr~|djjr~dd|djjD|d<|drt|dsd d d d d g|d<d|kr|ddkr|djjrdd|djjD|d<|drt|dsddddddg|d<d|kr|ddkrT|djjrTdd|djjD|d<|drlt|dsdddd ddg|d<dS)Nr9Zplotlyr>cSsg|] }|dqSrrLr1rLrLrMrsz)apply_default_cascade..r<r?cSsg|] }|jjqSrL)rr5rkscatterrLrLrMrsZcircleZdiamondZsquarerZcrossrAcSsg|] }|jjqSrL)rrrorLrLrMrsZsoliddotrZlongdashZdashdotZ longdashdotrCcSsg|]}|jjjqSrL)rrr)rkbarrLrLrMrsr/\+r)defaultsrSgetattrpioZ templatesdefaultrgrr Template colorscalerZViridisZcolorwayrZD3datarpanyrr)rhparamrLrLrMapply_default_cascadesn                 rcCs ||kr |Std||fdS)NzaA name conflict was encountered for argument '%s'. A column or index with name '%s' is ambiguous.) NameError) field_namereserved_namesrLrLrM_check_name_not_reservedsrcCs|d}t}|D]}|tkr q|tkr0||n||g}|dkrDq|D]}|dkrXqHqHt|trn||qHt|tjr|j}|rt ||r|||k}|r||qH||j krH|jdk rH||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_attrablesrraddrSeriesrhasattrr)rhrsrfieldr"argrZin_dfrLrLrM_get_reserved_col_namess,     rc Cs|dkst|tst|tr dSt|tjr0dSz t|Wntk rRYdSX|D]V}t|tspt|tr|dks||krdSqXz t|WqXtk rYdSXqXdS)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)rrintr MultiIndexiter TypeError)columnsrrrLrLrM _is_col_lists"    rcCs>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)rrtuplerboolrlrLrLrM_isinstance_listlike,s rcCs&|dk r"||ks||kr"d|}q|S)NrrL)rrtextrarLrLrM_escape_col_name<s rcCstj||djddS)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 rPT)Zdrop)rr reset_index)rrrLrLrMto_unindexed_seriesBsrc sr|d}|dk }i}ig}t}|r.t|nt} d|krh|ddkrh|sVtdndd|D}d|ko|dot|dt} | r|dD]} t|d| rd|d| f|d| <t|d| ts|d| df|d| <|r|d| d dk r| |krtd | qtD]t kr8| gn| } | dks| dgkr\qt krlgnfd d t t | D} t t| | D]p\}\}}t |rt |tt|nd |dkrҐqd}t|tjrtd|t|tst|trNt|jdk r t|jn|| }t|tr@|j|<n ||nRt|tsft|dsdkr| r|dt|d dk rt|}|d|d }rt |krtd|t |tt|ft||||<n|std|n||jkr^|r*|||fkr*qn2d|tt|j|f}|dkrT|d7}t|n\rt ||krtd|t ||tt|fnt|}t|||||<n|r>t|dr>||j kr|j!dks|j!|krd}n|j!}t"||||g}n.|j!dk r>|j!|kr>|||j!kr>|j!}|dkrRt|| }rt |krtd|t |tt|ft|t||t|<|dk st#dt krt||<n"t|trnt|||<dkr|$t|qqt |r*t |tt|nd |%fdd|D|%fddDt&|}||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.cSsi|]\}}||qSrLrL)rkrZseriesrLrLrMrnhsz/process_args_into_dataframe..r-TrzIAmbiguous input: values for '%s' appear both in hover_data and data_framecsg|]}dt|qS)rrr)rrLrMrsz/process_args_into_dataframe..rziArgument '%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/csi|]}|tt|qSrL)rrrkrt)lengthrLrMrnscs"i|]}|t|g|qSrL)rr) constantsrrLrMrns)'rrrrorrrrrrrrr enumeratezipnextrrrrrrrrrrmrrr keysrrrrrrrr DataFrame)rh wide_modevar_name value_namedf_input df_provided df_outputranges wide_id_varsrZhover_data_is_dictrZ argument_listZ field_listrargumentrrtZ real_argumenterr_msgrL)rrrrMprocess_args_into_dataframeLs<      "                 "  rc s. |D]n}|tkr||dk rt||tr||| rd nd <||| r0d nd <d&||<|d$ pH| |d$<n<|tj7tj8fk r| pj| || rvd nd <||| rd nd <| r|tj(k r|d dk r|d dk rd/|d0<nP|d dk rd|d0<d|d<||d <d&||<n d|d0<d|d<||d <d&||<| r"d|d$<||d<|S)1a 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 N)r,r-rF __dataframe__z2.0.2rT to_pandastoPandas to_pandas_dfrrr,zPCannot accept list of column references or list of columns for both `x` and `y`.znData frame columns is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.r/)Nrmrr`rwrxryr%Z__x__Z__y__rmselect_columns_by_namecs"h|]}t|tr|kr|qSrL)rrrrrLrM s z"build_dataframe..c3s|]}|kr|VqdSrHrLrrrLrM sz"build_dataframe..rzlData frame index is a pandas MultiIndex. pandas MultiIndex is not supported by plotly express at the moment.r)rr4rzrcsg|]}|kr|qSrLrL)rkr)rrLrMrsz#build_dataframe..)rrRunumberzLPlotly 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.r3r|rv)9rrrrr rrrrparse __version__Zpandas.api.interchangerZIndexZ column_namesrrrrrrrrrrrrrrrFunnelrapiZ interchangeZfrom_dataframer#rr ImportErrorNotImplementedErrorr@BarrrrNO_COLORrrurqrrZmeltrrrrrcrd) rhrcrrZneeds_interchangingpandasZ df_not_pandasrZno_xZno_yZwide_xZwide_yrrZwide_cross_namerZ hist2d_typesZhist1d_orientationZwide_orientationZnecessary_columnsexcZmissing_bar_dimZaxno_colorrZ count_nameZ other_dimZwide_value_varsrqryZv_dtypeZorient_vrL)rrrMbuild_dataframe"s                             "               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|kr|d|krt d |j |dd qdS) Nrr)rrz*None entries cannot have not-None childrenrcSs d|Srk)rrlrLrLrMrerz-_check_dataframe_all_leaves..z4Non-leaves rows are not permitted in the dataframe zis not a leaf.)rr rZisnullrrrZnonzeror}r#rr_rapplyr)rsZ df_sortedZ null_maskZ null_indicesZnull_row_indexrrZ row_stringsrLrLrM_check_dataframe_all_leavesWs*   rcs(|d|dddd}t|dddd}g}|D]"}|d}||||<q<|}dd}i}d}|d rzt|d |d <Wn$tk rtd |d YnX|d r|d |d kr|d d } |d | <| |d <|d n:d jkrd ndddtjDd<|d <d|<|d rt|d sn|}d}nfdd} | }|||d <tt j |} | D]} | |kr||| <qtt | t dddg} tj dddg| 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,|| || <tj| |gdd} q|d r|rd}|| jkrn|d7}qV|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)$z^ Build dataframe for sunburst, treemap, or icicle when the path argument is provided. rr.NrFZ _path_copycSs$|}t|dkr|dSdSdS)Nrrz(?))rr)runiquesrLrLrMaggfunc_discretes z5process_dataframe_hierarchy..aggfunc_discreter#zDColumn `%s` of `df` could not be converted to a numerical data type.r4Z_sumrzrcSsg|] }t|qSrLr)rkelrLrLrMrsz/process_dataframe_hierarchy..rr|Tcstj|j|jfdS)N)weights)rZaveragelocrrlZ count_colnamersrLrMaggfunc_continuoussz7process_dataframe_hierarchy..aggfunc_continuousrGrrrrs)Z ignore_indexZsort_color_if_discrete_color0rr&r"r$r-)TN)rrrZ to_numericrrrr rur differencerrrqrgroupbyZaggrrrrrstriprrrrr)rhr.Zdiscrete_colornew_pathrtZ new_col_namerZagg_fZ aggfunc_colorZnew_value_col_namercolsrZ df_all_treesrlevelZdf_treeZdfgjZ sort_col_namerLrrMprocess_dataframe_hierarchyos          &       rc Csd|d<|ddks |ddkr(tdz0t|d|d}t|d|d}Wn ttfk rxtdYnX||d td d |d|d<|d|d <|d=|d|d <|d=|S)z8 Massage input for bar traces for px.timeline() Tr*rNrz#Both x_start and x_end are requiredrzCBoth x_start and x_end must refer to data convertible to datetimes.ztimedelta64[ns]rmsrr)rrZ to_datetimerrrZ timedelta64)rhrrrLrLrMprocess_dataframe_timelines(    rcs|d}|dkr||fS|d|i}|s:||fS|d}d|d<d|d<t||fdd tt|D}||j||d<||fS) Nr"rFrFsortZ clockwiser.csg|]}|kr|qSrLrLr1rrLrMrsz)process_dataframe_pie..) rrr rrfromkeysZ set_indexrr)rhrer"Zorder_inrsorderrLrrMprocess_dataframe_pies  rcsfddttD}g}d}dkrNdrNdddd}dkrd krd krt|dn`drtddr|dd d <n2|tjtjtjfkr|dd d <n |dnxdks|tj kr|dnZ|tj tj fkrD|ddrNddkr0gd<ddn |dt d|kodo|tj tj fko|tjtjtjfkp d }nd }dkr|ddkr|ddkr|tjfkr|dn |ddkr4ddk }ddk } ddkrz|tjtjfkrL| rz|szdd<n.|tjtjtjtjfkrz|rz| szdd<ddkr|r| rtdd} tdd} | r| sdd<| r| sdd<ddkrdd<|tjkr~|r,| r,ddkr,dd<|d<d} d } | dkrJ| nd|d!<| dkr`dn| |d"<| dkrvdnd|d#<d|d<|tjtjfkr4|tjkrd$nd%}||dkrddk rddkrddkrd&||<n(ddkrddkrd&||<||dkr4d'||<|tjkrld(dk rlddkrldd<|d< d)d d k r|tjtj fkrd(}n(|tjkrddkrdnd}nd*}d)d krd+|d,|d-<nd+|d.d)d,|d-<|tjtjtjfkrd }d/|d0<d1krd1dkr\d2krd2d&krtd3d4|d5<n>|tjtjtj tjtj fkrd1|d1<ntd1d4|d5<dksdkr|d?<n*d@krtd@dAkrdBndCd>|d?<dDkrdE|dE<t dDdFsdDndDj!|dD<dGkrLddkrdHndI}ddkr4dHndI}dG|<d|<tddkrnddJ<dK< dKdk r dIdk rddI< dJdk r dHdk rddH< dHdk s dIdk s dJdk rddL<dMkr4dMdk r4dMt"kr4t#dNdMt"$fdOkrVdOdkrVtdO<dPkr dPddQkrt#dRdSdPdPdT<t%D]}|kr||qfdUd|D}t&|||}||||fS)VNcsg|]}|kr|qSrLrL)rkrrrLrMrsz infer_config..rrrrErr4r>r<TrFz marker.colorr3z line.colorr-r6z line.dashr5z marker.symbolr7zfillpattern.shapezmarker.pattern.shaperwrrrxryrvr|nbinsZnbinsxZnbinsyr]ZboxmodeZ violinmodergrouprZ text_autormz%{rZ texttemplate:rrr\rr[)r\rrir!markersrur?z+textrrW)rrr,r ZvhZhvZgeojsonZ featureidkey__geo_interface__rfrrr1r2facet_col_wraprz,Value '%s' for `trendline` must be one of %srecdfnorm)Nrr}z<`ecdfnorm` must be one of None, 'percent' or 'probability'. z'%s' was provided.r{csg|]}t|qSrL)r)rkrrrLrMrs)'direct_attrablesrmaxrrurrrrrrrrrr@rrcrdrrrrrrrrrrsortedrrrrrrrrgroup_attrablesrh)rhrcre layout_patchrdZ grouped_attrsr show_colorbarZhas_xZhas_yZx_is_continuousZy_is_continuousrwrr?rmodesrZother_positionrgrouped_mappings trace_specsrLrrM infer_configsr                                       $                       rcsd|kr in |dg}t}|D]tkr>|dq&|kr^t|d|<|}t|dkr||dkr|<q&ttt|<q&|d}t|t|krt ||i}n"dd|D|j dd d }|j }fd d|D} t tt D]"\t| fd d d} q dd| D} t |D],\tkrZ| D]} | dqpqZdd| D} i}t| | D]V\} } t| dkr|| || <n.tr|| df|| <n|| d|| <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" `groups` is the dicts of groups, ordered by the order above. Its keys are tuples like [("value1", ""), ("value2", "")] where each tuple contains the name of a single dimension-group rFrrrrcSsg|]}|tkr|qSrL)rmrkgrLrLrMrsz)get_groups_and_orders..FT)rZobservedcs"g|]}tdkr|n|fqSrn)rr)required_grouperrLrMr"scs&|kr"|SdS)Nrr)r)rrrrLrMr)rz'get_groups_and_orders..rlcSsg|] }t|qSrL)r )rkrGrLrLrMr-scSsg|] }t|qSrL)rrrLrLrMr2s)rrrmrr rrrrrrindicesr rrinsertrZ get_group pandas_2_2_0)rhr\Zsingle_group_nameZ unique_cacherrsgroupsZgroupedZ group_indicesZsorted_group_namesZfull_sorted_group_namesrZsfsrL)rrrrrMget_groups_and_orderss\        rc4sf |pi}|pi}tt||tjtjtjfkrJddk rJt|tjfkrdt|\}|dkrztj }t t |||\}}}}dd|Dpt g}t |\} g} g} d} } |D]}|jkr|jd|jd<qƈ|j}|jdkr(td d fd d|D} t| } |jd kr`td d fdd|D} t| } |D]2}||jkrd|jt|jt|j|j|<qdqt|j}i}t}g}d}dd}| D]\}}t}t}d}t|||D]^\}}}|t krt|}t|jts>t|||<|jr>t|||<|jdkr|}qd| }||krtt!||<||}|D].}|j"|d} |j"tj#tj$tj%tj&tj'tj(tj)tj*tjtjtjf kr| j+||dko||kd|j"tj tj,tj-tj.fkr| j+d|d|/|d| _0d| _1t2|D]\}!}||!}z|3| |j|Wnt4k rL||dkr|j"tj,tj-fkr|jdksH|j"tj.fkr|jdkrn||dkr|j"tj.fkr|jdkr| j+t5|j|ddnP|j"tj%tj&tj'fkrF|jdkrF| j+dgt||j|gddddnYnX|jd krf|j|}"n&ddk r|j6dkrd}"nd}"|jdkr|j|}|rd|d|}"d|d|}n|j6d krd}nd}|"dkr|"| _0|dkr.|| _1q.|dj"tj7krD|j"tj-krD| j8j9rD| j+t5| j8j9ddd!kr<d"d#krddnd }#d"d$krdnd }$d!d%d&k}%|j:|#|%d'}||$;}&||$<||$<|%s|j:|#dd'}d!d%d(kr|&||$||$<d)d*kr||$|&||$<n"d)d+kr|\}'}(| +|'|(dk r|?|>|(|d-d.<||krt5g|d/||<||d0?| qqd1d| D})t|)dkrt@|)fd2d3d4})|rj|tj*tj(tj)fkr d5nd}*d6p ddg}+tAd7d8},t5|,Bd9d:|+d|+dt5tC|*|*d;d<|d=<d>D]}-|-rn|-||-<qnt5dd?|d@<|rd||d@dA<dBrʈdB|dA<n dCjDjEjFdkrdDdEi|dF<dGkrdGrdCjDjGjHdkrdH|d@dI<|r.rrrrr2rcsg|]}t|qSrLrrkrrNrLrMr^srr1csg|]}t|qSrLrrrrLrMrbsrr0z, rP) legendgroup showlegendT)ZalignmentgroupZ offsetgroup)r5rr)r5rr4rj)rrF)rr{Z showscalerrrrr,rwryrxstandardr )r ascendingZ complementaryrr}rgY@rpx_fit_results)r|rr|cSsg|]}|qSrLrLrQrLrLrMr scsd|dS)Nr0rr)rR)rhrrLrMr rzmake_figure..rr range_colorr{ make_figurer>Zcolor_continuous_midpointr))r{ZcmidZcminZcmaxZcolorbarr)r;r:)Z tracegroupgaplegendrtitler9rGrOmarginrZconstant itemsizingr)r9 overwriterrrarZoverallzOverall Trendline)rrrrrr_)rrZexclude_empty_subplots)selectorr)r)\rrrrrrrrrrrrrmrr\r^r]rarjrrrrrrorrrrr[r`rr#rrcrrrrrrrrrrcrdrrZ _subplot_rowZ _subplot_colrr_rrrfrrr4rr|Zcumsumrrrrr Zvalidate_coercerr rrGrrrceilmin init_figurerrrZ add_tracesZ update_layoutrTrereZ add_traceZ update_tracesrrrWrrU)4rhrcrerrrrrr\r col_labels row_labelsr$r%mZ sorted_valuesval subplot_typeZtrace_names_by_framerTZtrendline_rowsZtrace_name_labelsrZ group_namerrZ frame_namerrlZ trace_nameZ trace_namesrrrrrr3rZ group_sumr2r frame_listZcolorvarrZcolorscale_validatorryrXr>rRZtrendline_specZtrendline_traceZ next_colorrL)rhrrNrMr@s8         &                                              $        rcs~fddt|D}dg}dg|} |dd} dkr|ddk r|ddksld |krr|d rrd } nd } | g|d d | g} d } n"| r|dpd} n|dpd} |ddk r |ddksd |kr|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<tk r\}z||dd||ddW5d}~XYnX|jj D]}|j ddqf|S)Ncs g|]}tpddgqS)domainr)r)rkrr%rrLrMrr szinit_figure..g?rrZxyrrr4gGz?gzG?rg{Gz?Zfacet_row_spacinggQ?gQ?rg{Gzt?Zfacet_col_spacingg{Gz?cSs6d|f|jdkr2|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)rhformat)er.rrLrLrM_spacing_error_translator sz.init_figure.._spacing_error_translatorr_z bottom-left) ZrowsrspecsZ shared_xaxesZ shared_yaxesZ row_titlesZ column_titlesZsubplot_titleshorizontal_spacingvertical_spacing row_heights column_widthsZ start_cellZ HorizontalZVertical)font) rrrrrr r rr  annotationsr)rhrrr$r%r rrrrrZ main_sizerrZsubplot_labelsrrrrXrZannotrLrrMr p sn      ,     r )N)NN)dZplotly.graph_objsZ graph_objsrZ plotly.ioiorx collectionsrrZ_special_inputsrrrrrr r r r Z_plotly_utils.basevalidatorsr Z plotly.colorsrrr packagingrrrZnumpyrZplotly._subplotsrrrrrrrrrrrZrenameable_group_attrablesrr@rbrrrdrcrrrrobjectr8rvrTrVrYrZrbrjrprurrrrr r(rrrrrrrrUrhrermrrrrrrrrrrrrrrrrr rLrLrLrMs      (   &2D& :ML  W7}eE 2