hddlmZddlmZmZddlmZmZmZm Z ddl m Z m Z m Z mZmZddlZddlZddlZddlmZddlmZddlmZdd lmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+dd l,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2dd l3m4Z4ddl5m6cm7Z8ddl9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBddlCmDZDddlEmFZFddlGmHZHddlImJZJmKZKmLZLmMZMmNZNddlOmPZPmQZQmRZRmSZSmTZTmUZUe r*ddlVmWZWddlXmYZYddlZm[Z[ddl\m]Z]ddl^m_Z_m`Z`maZambZbddlcmdZdd1dZeGdd eZfGd!d"efeZgGd#d$egZhGd%d&egZiGd'd(efZjGd)d*ejZkGd+d,efZlGd-d.elZmGd/d0efZny)2) annotations)ABCabstractmethod)HashableIterableIteratorSequence) TYPE_CHECKINGAnyLiteralcastfinalN)libAbstractMethodError)cache_readonly)find_stack_level) is_any_real_numeric_dtypeis_boolis_floatis_float_dtype is_hashable is_integeris_integer_dtype is_iterator is_list_like is_numberis_numeric_dtype)CategoricalDtypeExtensionDtype) ABCDataFrameABCDatetimeIndexABCIndex ABCMultiIndexABCPeriodIndex ABCSeries)isna) DataFrame)Version) pprint_thing)tools)%register_pandas_matplotlib_converters)reconstruct_data_with_by)unpack_single_str_list)get_standard_colors) decorate_axesformat_dateaxismaybe_convert_indexmaybe_resample use_dynamic_x)create_subplots flatten_axesformat_date_labels get_all_linesget_xlimhandle_shared_axes)ArtistAxes)Axis)Figure) IndexLabelNDFrameTPlottingOrientationnpt)SeriescDddlm}t|j| S)z? Check if there is a color letter in the style string. r) BASE_COLORS)matplotlib.colorsrFset isdisjoint)stylerFs ED:\jyotish\venv\Lib\site-packages\pandas/plotting/_matplotlib/core.py_color_in_stylerLcs .;**51 11c eZdZUdZeed9dZdZdZed:dZ de d< d;d d d d d ddd  ddZe e d?dZdZe e d@dZdAdZe edBdZe dCdZe dCdZe e dDdZe dEdZe edFdZe edGdZe edHdZedZe e dZe dIdZ e dCdZ!dJdZ"e dCd Z#e dKd!Z$edKd"Z%e dJd#Z&e e dL dMd$Z'e ed:d%Z(e dNd&Z)e dOd'Z*dCd(Z+e e dPd)Z,e ed*Z-d Z.e d+Z/ee0 dQ dRd,Z1d-Z2e d:d.Z3e edSdTd/Z4e dUd0Z5e dVd1Z6e dVd2Z7e dWd3Z8 dX dYd4Z9e e dZd5Z:e d[ d\d6Z;e d]d7Zn||_>||_?||_@|#|_A|#jcdtj})|j|)|j||_Ed|jvsJ|j|jj|_5ycc}%wcc}%w)NrrXzNo group keys passed!hist)kindTFz axes.gridr\r]r^xerryerr)rerfcmapz*Only specify one of `cmap` and `colormap`.color)Gmatplotlib.pyplotpyplot ValueErrorcommaybe_make_listby isinstancer(columnsrrTgroupbyr._groupedrdtype_validate_subplots_kwargsubplots_validate_sharexsharexshareyfigsizelayoutxticksyticksxlimylimtitle use_indexxlabelylabelfontsizerot_rot_set _default_rotrcParamsgridlegendlegend_handles legend_labels_validate_log_kwdr\r]r^rarJr_r`axpop _get_nseries_parse_errorbarserrorsr[booltuplelistnpndarrayr# secondary_y TypeErrorcolormaptable include_boolkwdsr no_default_validate_color_argsrh _ensure_frame)*rSr[rdrnrurwrxrryrrrrfigrr}r~r{r|rrrrrrrzrcolumnr\r]r^r_r`rarJrpltcolrerfnseriesrhs* rK__init__zMPLPlot.__init__sN ( "b>45 5%%b) dI &"226: #'<< #/C3CDI3NC<  $|| +$'').>tCy.I+   77 4::#7 LL)?)HIDM T ;; d<  4j11&"bA         "    ?DH!DM!DM((DH <'5S\\+-FD  ,.-/J00> J00> 4j228VD   $  xx%xx%##D)$Z00tWM d$Z00tWM d#T2  +eT2::x'PQ&-K& T>hHI I T> HHV,DM$DM ( #..1..udmmD dii'''&&tyy1 G  s9P P(%Pc||| d}t|Sd}t|St|s tdt|S)NTFzsharex must be a bool or None)rrr)rwrrns rKrvzMPLPlot._validate_sharex sU >zbjF|F|;< <F|rMcx|%t|tst|tr|dk(r|Std|d|d)Nsymz keyword 'z'' should be bool, None, or 'sym', not '')rorstrrk)clskwdvalues rKrzMPLPlot._validate_log_kwd/sH M%&5#&5E>LuCE7! L  rMct|tr|St|ts tdd}||vr!tddj |d|dt|t r t d|j}t|tr t d|jt|k7r t d g}t}|D]}t|s td |j|}|d k(jr1tj |d k(|} td t#| d t|} |j%| } | rtd| d|j'| }|j)t+||j-|} | D]%} |j/| }|j)|f'|S)a_ Validate the subplots parameter - check type and content - check for duplicate columns - check for invalid column names - convert column names into indices - add missing columns in a group of their own See comments in code below for more details. Parameters ---------- subplots : subplots parameters as passed to PlotAccessor Returns ------- validated subplots : a bool or a list of tuples of column indices. Columns in the same tuple will be grouped together in the resulting plot. z(subplots should be a bool or an iterable)linebarbarhrckdedensityareapiez2When subplots is an iterable, kind must be one of z, z. Got .z3An iterable subplots for a Series is not supported.zOAn iterable subplots for a DataFrame with a MultiIndex column is not supported.zTAn iterable subplots for a DataFrame with non-unique column labels is not supported.zPWhen subplots is an iterable, each entry should be a list/tuple of column names.zColumn label(s) z not found in the DataFrame.z3Each column should be in only one subplot. Columns z! were found in multiple subplots.)rorrrkjoinr&rQrpr$nuniquelenrHrget_indexer_foranyrextractr intersectionunionappendr differenceget_loc)rur[rdsupported_kindsrpout seen_columnsgroupidx_locs bad_labelsunique_columns duplicatesunseen_columnsridx_locs rKrtz MPLPlot._validate_subplots_kwarg?s2 h %OHh/GH H    &))O45VD6D  dI &%E ,, g} -%$  ?? G ,%+ $&)e E& >..u5HB##%ZZB>  &tJ'7&88TU!ZN%22>BJ )l*KM(--n=L JJuX '),!++L9$Foof-G JJz "% rMc|tjur0d|jvr!|tjdt y|j dk(r|t|s|g}t|tr|j dk(rt|dvr|g}|tjdt |jLt|jr |j}n |jg}|D]}t|std|S)NcolorszC'color' and 'colormap' cannot be used simultaneously. Using 'color' stacklevel)zCannot pass 'style' string with a color symbol and 'color' keyword argument. Please use one or the other or pass 'style' without a color symbol)rrrwarningswarnrrrrorrrJrLrk)rSrhrstylesss rKrzMPLPlot._validate_color_argss CNN "499$)= $/1  <<1 !2<;NGE eU # (9c%jF>RGE   MMU+-  :: !DJJ'**"1%$G rMc#K|jD](\}}|tj|jf*ywrW)itemsrasarrayvalues)r[rrs rK _iter_datazMPLPlot._iter_datas6  :: WW TZZ6%9t}}% % WW TZZ5%8t||$ $::a= rMc8|j|jSrW)rr[rRs rKrzMPLPlot.nseriess  ++rMc8|jjyrW)rdraw_if_interactiverRs rKdrawz MPLPlot.draws $$&rMc<|j|j}|j||j|j |j ||j D]/}|j||j||j1yrW) _compute_plot_datar _make_plot _add_table _make_legend_adorn_subplotsaxes_post_plot_logic_common_post_plot_logicr[)rSrrs rKgeneratezMPLPlot.generatesw !hh    S!))B  ( ( ,  ! !"dii 0rMct|jdk7xs2t|jdk7xst|jdk7S)zcheck whether ax has datar)rlinesartists containers)rs rK_has_plotted_objectzMPLPlot._has_plotted_objects=288}!TS_%9TS=OST=TTrMc8|j|s|j|St|dr |jSt|dr|S||j }}|j |_|j |_||c|_|_|j|s|jjd|jdus|jdur|jd|S|jdk(s|jdk(r|jd|S)Nright_axleft_axFTlogrsymlog)on_right _get_ax_layerhasattrrtwinx _get_lines_get_patches_for_fillrr get_yaxis set_visibler]r^ set_yscale)rSraxes_numorig_axnew_axs rK_maybe_right_yaxiszMPLPlot._maybe_right_yaxiss }}X&%%b) ) 2z ";;  R #I!"((*VG ' 2 2F --  (  -G fn ++G4!!#//6yyD DKK4$7!!%(Me#t{{e';!!(+MrMc |jdSNr _axes_and_figrRs rKrz MPLPlot.fig(s!!!$$rMc |jdSNrr rRs rKrz MPLPlot.axes-s!!!$$rMc |jrt|jtr |jnt |j}t ||j |j|j|j|j|j\}}n|j8|jj|j}|jd}nM|jj}|j|j!|j|j}t#|}|j$dus|j&dur |Dcgc]}|j)dc}n=|j$dk(s|j&dk(r|Dcgc]}|j)dc}|j*dus|j&dur |Dcgc]}|j-dc}n=|j*dk(s|j&dk(r|Dcgc]}|j-dc}t/t0d|}||fScc}wcc}wcc}wcc}w) N)naxesrwrxryrrz layout_type)ryoTrrrr=)rurorrrr5rwrxryrrz _layout_typerfigure add_subplot get_figureset_size_inchesr6r\r^ set_xscaler]rr r )rSrrraaxes_seqs rKr zMPLPlot._axes_and_fig3s == *4==$ ? SEW ({{{{ 77{{ --ICWW_((//$,,/7C??3'D''$$&C||'##DLL177DD! 99  t 3*. /$QQ\\% $ / YY% 4;;%#7-1 2TQ\\( #T 2 99  t 3*. /$QQ\\% $ / YY% 4;;%#7-1 2TQ\\( #T 2($/} 0 2 0 2sH?II  Ic|jrP|j8t|js#|jj |jS|jSt |jtxr |j}t|jxr"t|j|jk(}|s|r |j|jddS|jdS)z$ Return result axes rF)primary) rurzrrrreshaperorrrrr)rSsec_trueall_secs rKresultzMPLPlot.result[s =={{&|DGG/D(tyy(($++66yy !$"2"2D9Nd>N>NHT--.:(()T\\9 7))$))A,)FFyy|#rMcNt|jtr|St|jst |jr;t|jt r!|j dtjSt|dkDrtj|S|S)Nfloat)dtypena_valuer) ror!rrrr to_numpyrnanrrr[s rK_convert_to_ndarrayzMPLPlot._convert_to_ndarrayts| djj"2 3K TZZ (N4::,FJ JJM ==w=@ @ t9q=::d# # rMcRt|trD|j}||jd}||j }|S|j |}|S|j dvrD|j |jn|j|j z}|jdd|f}|S)N)name)rcr) ror&rar)to_framerTrnrploc)rSr[racolss rKrzMPLPlot._ensure_frames dI &JJE}!2}}}  }}%}0 ZZ? *#'77?4<< tww8ND88AtG$D rMc|j}|j3d|_t|j|j|j}|j d}t jdddg}|jdur|jt jd}|jdk(rt jg}dg}|jd k(r|jgd |j|| }|jd d k(}|r td|j!t#|j$|_y)NT)rnr,F)copydatetime datetimetz timedeltarscatter)objectcategorystring)includeexcluderrzno numeric data to plot)r[rnrur-rp infer_objectsrnumberrrbool_rTextend select_dtypesrrapplyrsr&)rSr[ include_type exclude_type numeric_datais_emptys rKrzMPLPlot._compute_plot_datasyy 77  DM+DII$'' UD!!u!- :|[I     $    ) :: II;L'=L :: "    @ A)), )U %%b)Q. 56 6 &&tDz'E'EF rMct|rWr)rSrs rKrzMPLPlot._make_plots !$''rMc|jdury|jdur|jj}n |j}|jd}t j||y)NFTr)rr[ transpose_get_axr+)rSr[rs rKrzMPLPlot._add_tablesS ::   ZZ4 99&&(D::D \\!_ BrMc.|jdk(s |jt|j|j|j|j t|j|j |j t|dr;t|j|jj |j yy|jdk(rt|j|j |j|j t|j|j|j t|dr;t|j|jj |j yyt)z!Common post process for each axesrUN)rr)rr horizontal) rYrs_apply_axis_propertiesxaxisrryaxisrrrk)rSrs rKrzMPLPlot._post_plot_logic_commons=   z )T-=-=-E J - -dhh .  J - -bhh - Or:&T 11KK%% 2'    - J - -dhh .  J - -bhh - Or:&T 11KK%% 2'  rMcy)z7Post process for each axes. Overridden in child classesNrXrSrr[s rKrzMPLPlot._post_plot_logicsrMc *t|jdkDrW|j|}|j|\}}t |t|||z|||j |j |jD]}t|d|}|j|j|j|j|j|j|j|j|j|j|j|j|j $|j#t%|j |j'|j&|j(r|j*rt-|j(rt|j(|j.k7r.t1dt|j(d|j.t3|j|j(D]\}}|j5|y|j7|j(yt-|j(r d}t1||jdj5|j(yy)z%Common post process unrelated to datar)axarrnplotsrnrowsncolsrwrxrNz~The length of `title` must equal the number of columns if using `title` of type `list` and `subplots=True`. length of title = z number of columns = zNUsing `title` of type `list` is not supported unless `subplots=True` is passed)rr _get_subplots_get_axes_layoutr:rwrxgetattrr| set_yticksr{ set_xticksr~set_ylimr}set_xlimr set_ylabelr*rrrurrrkzip set_titlesuptitle)rSrall_axesrPrQrrmsgs rKrzMPLPlot._adorn_subplotss tyy>A ))#.H005LE5 8}em{{{{ ))BZ,B{{& dkk*{{& dkk*yy$ DII&yy$ DII&{{& l4;;78 GGDII '* ::}} +4::$,,6(125TZZ0AB337<<. B&)DJJ%? E U+&@LL, +;%S/) ! &&tzz2/ rMc||P|j|jz}|D])}||j|||j|+yy)a Tick creation within matplotlib is reasonably expensive and is internally deferred until accessed as Ticks are created/destroyed multiple times per draw. It's therefore beneficial for us to avoid accessing unless we will act on the Tick. N)get_majorticklabelsget_minorticklabels set_rotation set_fontsize)axisrrlabelsras rKrHzMPLPlot._apply_axis_properties)s` ?h2--/$2J2J2LLF?&&s+'&&x0  3rMc&t|jjts/|jjj}| t |}|St t |jjj}dj|SN,) ror[rpr$r)r*mapnamesr)rSr) stringifieds rK legend_titlezMPLPlot.legend_title=sm$))++];99$$))D#D)KlDII,=,=,C,CDK88K( (rMcb|js"|jr|j|r|dz }|S)z Append ``(right)`` to the label of a line if it's plotted on the right axis. Note that ``(right)`` is only appended when ``subplots=False``. z (right))rur_r)rSraindexs rK_mark_right_labelzMPLPlot._mark_right_labelIs+}}T]]55I Z E rMcp|jj||jj|y)z Append current handle and label to ``legend_handles`` and ``legend_labels``. These will be used to make the legend. N)rrr)rShandleras rK_append_legend_handles_labelsz%MPLPlot._append_legend_handles_labelsTs, ""6* !!%(rMc|j|jd\}}g}g}d}|js0||jj }t t jt dkr |j}n |j}|jDcgc]}|j }}|jrv|jdk(r1|t|jz }|t|jz }n||jz }||jz }|j |j}t|dkDr|j||d|yy|jrz|jrm|jD]]}|j!st#j$5t#j&ddt(|jd ddd_yyycc}w#1swYqxYw) Nrr(z3.7reversebest)r+rignorez.No artists with labels found to put in legend.)r+)_get_ax_legendrru get_titleget_textr)mpl __version__ legendHandlesr get_textsrreversedrrlr get_visiblercatch_warningsfilterwarnings UserWarning)rSrleghandlesrerxs rKrzMPLPlot._make_legend^s%%diil3C}} 0023??+gen<!//G!00G03 @1!**,@{{;;)+x(;(;<>#!002 //$L'  f - 32  +]!A&32s%G,1.G11G: c|j}t|ddxs t|dd}d}||j}|||}|}||fS)zQ Take in axes and return ax and legend under different scenarios rNr) get_legendrT)rrother_ax other_legs rKrwzMPLPlot._get_ax_legendsa mmo2y$/P72z43P   ++-I ;90CB3wrMcddlm}|Sr )rirj)rSrs rKrz MPLPlot.plts  ( rMc|jj}|jdv}|jrt |t r |j j}|St|jr|j}|St |ts|r|j}|Sd|_ ttt|}|Sttt|}|S)N)r/date datetime64timeT)r[rn inferred_typerror% to_timestamp _mpl_reprrr!r"_need_to_set_indexrranger)rSrn is_datetypers rK _get_xtickszMPLPlot._get_xtickss ))-WW  >>%0&&(224"+5;;7 OO%E#34 OO% +/'s5z*+U3u:&'ArMc t|}|jr?tjj |}tjj ||}t |tr|j}|rjd|vr'tj|jd|d<d|vr'tj|jd|d<|j||fi|S||||fn||f}|j|i|SNrerf) r'rrmaarray masked_whereror#rgeterrorbarplot) rrryrJ is_errorbarrmaskargss rK_plotz MPLPlot._plots Aw 88: AA""4+A a " A ~!xx(89V ~!xx(89V 2;;q!,t, ,%*$5Aq%=Aq6D277D)D) )rMc|jS)zCSpecify whether xlabel/ylabel should be used to override index name)rrRs rK_get_custom_index_namezMPLPlot._get_custom_index_name {{rMct|jjtr^|jjj}t j |r)dj|Dcgc] }t|c}}n0d}n-|jjj}| t|}|j}| t|}|Scc}wrg) ror[rnr$rjrl any_not_nonerr*r)r)rSr)r index_names rK_get_index_namezMPLPlot._get_index_names diioo} 599??((D&xx$ ?$Qa$ ?@99??''D#D)002  ! +D !@s&Cc:|r t|d|St|d|S)z,get left (primary) or right (secondary) axesrr)rT)rrrs rKrzMPLPlot._get_ax_layers% 2y"- -2z2. .rMct|jtr'tfdt |jDSS)zJReturn the index of the axis where the column at col_idx should be plottedc30K|] \}}|vr|ywrWrX).0 group_idxrcol_idxs rK z/MPLPlot._col_idx_to_axis_idx..s&*B&Ye#*Bs)rorurnext enumerate)rSrs `rK_col_idx_to_axis_idxzMPLPlot._col_idx_to_axis_idxs= dmmT **3DMM*B NrMc"|jrB|j|}|j|}|j||}||j|<n!|jd}|j||}|j j d|S)NrT)rurrrrr)rSirs rKrEzMPLPlot._get_axs ==))!,A1B((Q/BDIIaL1B((Q/B ""4( rMct|jtr |jSt|jttt j tfr%|jj||jvSyrW) rorrrrrrr#r[rp)rSrs rKrzMPLPlot.on_rightsa d&& -## # d&&bjj((K L99$$Q'4+;+;; ; MrMcd}|jnt|jtr |j|}nCt|jtr|jj ||}n |j}d|vxs|j du}|duxs t| }|s |jr2|r0t|tr |||d<||fS||t|z|d<||fS#t$rYuwxYw)z Manage style and color based on column number and its label. Returns tuple of appropriate style and kwds which "color" may be added. Nrh) rJror IndexErrordictrrrLrur)rSrrcol_numrarJ has_color nocolor_styles rK_apply_style_colorszMPLPlot._apply_style_colorss :: !$**d+ JJw/EDJJ- ue4 tO@t}}D'@  C_U-C)C M&$' &u W d{!'wV'< =W d{"sC'' C32C3c| |j}|dk(r |j}n|jj|}t ||j |S)Nrh) num_colorsrrh)rrhrrr/r)rSr color_kwdsrhs rK _get_colorszMPLPlot._get_colors;sP  J  JJEIIMM*-E"!]]  rMc|d|fSd}t|tr |||}||fSt|tr ||fSt|tr:|||}t j |}t j ||df}||fSt|tr^||j}||jj|}t j |}t j ||df}||fSt|r3t|rt j t|}nt j |}|j}t|trS|ddk(rKt j|d}|j}|dt!|k7rst#dt!|dt|trK|j$dk(r<|d|k7s|ddk7s|dt!|k7rt#d |d t!|dt!|dk(rt j ||df}||fSt'|r&t j |g|t!|f}||fSd |d }t#|) a Look for error keyword arguments and return the actual errorbar data or return the error DataFrame/dict Error bars can be specified in several ways: Series: the user provides a pandas.Series object of the same length as the data ndarray: provides a np.ndarray of the same length as the data DataFrame/dict: error values are paired with keys matching the key in the plotted DataFrame str: the name of the column within the plotted DataFrame Asymmetrical error bars are also supported, however raw error values must be provided in this case. For a ``N`` length :class:`Series`, a ``2xN`` array should be provided indicating lower and upper (or left and right) errors. For a ``MxN`` :class:`DataFrame`, asymmetrical errors should be in a ``Mx2xN`` array. Nc<|j|j}|SrW)reindexrn)r[es rK match_labelsz.MPLPlot._parse_errorbars..match_labelsgs $**%AHrMrrz>Asymmetrical error bars should be provided with the shape (2, )rz;Asymmetrical error bars should be provided with the shape (z, 2, z No valid z detected)ror!rr&r atleast_2dtilerrrpdroprrrr expand_dimsrrkrr)raerrr[rrevalues err_shaper^s rKrzMPLPlot._parse_errorbarsMs. ;:   c< (tS)C|DyyT " vDyqY 'tS)C--$C''#|,ChDycS !3i&&G ))#./D--(C''#|,CZDyW# 3mmDI.mmC( I$ *y|q/@nnS!,II Q<3t9,$..1$i[;D,/CHHMq\W,!! )!! D 1$++2)5T 1F 3x1}ggcGQ<0Dys^''#d)$CDyeWI.CS/ !rMci}tddg||gD][\}}|s |j|}t|ttfr|||j vr||}n d}n ||||}|W|||<]|Sr)rZrror!rkeys) rSrarnrerfrkwflagrs rK_get_errorbarszMPLPlot._get_errorbarssVV,tTl;HBkk"ocL$#78(Uchhj-@!%j"&3?e*C?!$F2J< rMcttjtdkrddlm}nddlm}|j Dcgc] }t||r|j|"c}Scc}w)Nz3.8r)Subplotr<) r)rzr{matplotlib.axesrr=get_axesroget_subplotspec)rSrKlassrs rKrRzMPLPlot._get_subplotssb 3?? #gen 4 8 5lln $2u%"*<*<*>*J $   s%A-c|j|}t}t}|D]N}|jj}|j |dd|j |ddPt |t |fS)Nrr)rRrH get_position get_pointsaddr)rSrrx_sety_setrpointss rKrSzMPLPlot._get_axes_layouts|!!#&B__&113F IIfQil # IIfQil #  E CJ''rM)returnr)r str | None)NNFNFTNNTNNNNNNNNNNNFNFNFN)*rnIndexLabel | Nonerubool | Sequence[Sequence[str]]rw bool | Nonerxrrrryztuple[float, float] | Nonerz bool | strrHashable | Nonerrr int | Nonerz bool | tuple | list | np.ndarrayrrrrrrr\bool | None | Literal['sym']r]rr^rr_rr`rrarrNone)rwrrrrrrrrr)rurr[Series | DataFramerdrrzbool | list[tuple[int, ...]])r[z.DataFrame | dict[Hashable, Series | DataFrame]rz%Iterator[tuple[Hashable, np.ndarray]]r[rrint)rrrr)rr=rr)rr=rrrr=)rr?)rzSequence[Axes])rztuple[Sequence[Axes], Figure])rr(rr?rrrr=rrNN)rdr>rrrr)rarrnrrr)rqr;rarrr)rr=)NFrr=r np.ndarrayrr)T)rr)rrrr)rr)rdict[str, Any]rrrar)Nrh)rrrr)rarr[rArrrztuple[Any, NDFrameT])NNTT)rerrfrrr)rr?)rr?rztuple[int, int])>__name__ __module__ __qualname____doc__propertyrrTrrrY__annotations__rr staticmethodrv classmethodrrtrrrrrrrrrrrr rr&rrrrrrrrHrlrorrrrwrrrr,rrrrrrErrrrrrRrSrXrMrKrOrOlsS""LL  O  $38".2 !     "&"&#8="$(9Y2<.3-2/4!%IY2  Y2 1 Y2  Y2Y2Y2,Y2Y2( )Y2* +Y2,-Y2.6/Y223Y267Y28"9Y2<+=Y2>+?Y2@-AY2BCY2DEY2FGY2L MY2v        ,   &     g0g8JgRUg %g gR%N 1<1 .1 1 !  , , ' '  1  1 U U ! !F % % % % $ $L$$0  $     #G #GJ(    4FF ;3 ;3z 5911(21 1 1$  ) )   ) )(.T         8*IN**'*BF*+*,  & / /         < < *58AD <"&!   $ \\'\25\ \ \| FJ,0?C  *        (  (rMrOcReZdZdZdZddZed dZed dZed dZ y) PlanePlotzM Abstract class for plotting on plane, currently scatter and hexbin. singlec tj||fi|||t|jdzt |r=|j j js|j j |}t |r=|j j js|j j |}||_||_ y)Nz requires an x and y column) rOrrkrTrr[rp_holds_integerrr)rSr[rrkwargss rKrzPlanePlot.__init__st.v. 9 TZZ*GGH H a=!2!2!A!A!C !!!$A a=!2!2!A!A!C !!!$ArMcyrrXrs rKrzPlanePlot._get_nseriessrMc|j|j}}|j |jn t|}|j |jn t|}|j ||j |yrW)rrrr*r set_xlabelrY)rSrr[rrrrs rKrzPlanePlot._post_plot_logics]vvtvv1 $ 7\!_ $ 7\!_ f frMc J|jd}|j|fd|i|S)Nrr) collectionscolorbar)rSrrrimgs rK_plot_colorbarzPlanePlot._plot_colorbars-"nnR s||C/B/$//rMNrrr)rr=rr?) rrrrrrrrrr rXrMrKrrsOL      0 0rMrczeZdZed dZ d ej dd d fdZd dZd dZ ddZ ddZ xZ S) ScatterPlotcy)Nr2rXrRs rKrTzScatterPlot._kindsrMN)r normc 8|d}nt|r||jvr||}||_||_||_t ||||fi|t|r=|jjjs|jj|}||_ y)N) rrprr rsuperrrr[rc) rSr[rrrrr rr __class__s rKrzScatterPlot.__init__s 9A ^T\\ 1QA    q!.v. a=!2!2!A!A!C !!!$ArMc .|j|j|j|jf\}}}}|jd}t |xr||jj v}|xr't|j|jt}|j} |j| ||} |j| |\} } |j| |} |jr |j}nd}|j ||j"||j"f| || | |j$d|j&}| r|r|nd}|j)|||}|rt+|j|j,j.}|j1t3j4d|dz ||j6j9|j|j,j.||j;|||j=|dd}|j=|dd}t+|dkDst+|dkDrSt?|fi|}|jAd|d <|jB||j"||j"fd d i|yy) Nr)rrargrrr()rra?F)rarnrf)rarnreecolor linestylenone)"rrrr[rrrpror!rrh _get_c_values_get_norm_and_cmap _get_colorbarrrar2rrrr rcat categories set_ticksrlinspacerset_yticklabelsrrrr get_facecolorr)rSrrrrr[r c_is_columncolor_by_categoricalrhc_valuesrrgcbrar2 cbar_labelcbarn_catserrors_xerrors_yerr_kwdss rKrzScatterPlot._make_plot4sf 9 1a YYq\!!n?dii.?.?)? * z IIaL   00  %%e-A;O,,X7KL d   + 6 ;;JJEE"** GNN GNN  ff  ii   )rJ&&rs*&ED#TYYq\--889r{{3 fEF'' ! (8(8(C(CD    . .  &&Qae&D&&Qae&D x=1 H  1H11H!(!6!6!8!;HX  BKKQQ U& UH U!2rMc|j}| | td|||jjd}|S||}|S|r%|j|j j }|S|r|j|j}|S|}|S)Nz&Specify exactly one of `c` and `color`zpatch.facecolor)rrrrr[rcodesr)rSrhr$r#rr%s rKrzScatterPlot._get_c_valueshs FF =U.DE E 9xx(():;H H "yy|''--H  yy|**HHrMcb|j}|j*tjj |j}n1t |t st|rtjd}nd}|r|ddlm }t|j|jj}|jt|j Dcgc] }|| c}}t#j$d||dz}|j'||j } | |fS|j(} | |fScc}w)NGreysr)rr)rrrz colormapsget_cmaprorr matplotlibrrr[rrListedColormaprNrr  BoundaryNormr) rSr%r$rrgrr)rboundsrs rKrzScatterPlot._get_norm_and_cmapxs FF == $==))$--8DHc*/?/I==)DD D$4 )1))445F((5=)I=a$q'=)IJD[[FFQJ7F&&vtvv6D Tz99DTz*Js D,c|jxs|}|j}|tjurt |xr|S|SrW)rr rrr)rSr%r# plot_colorbarr&s rKrzScatterPlot._get_colorbars< 4 ]]  #H-?- ? rM)rzLiteral['scatter']r)r zbool | lib.NoDefaultrrr)r$rr#r)r$r)r#rrr) rrrrrTrrrrrrr __classcell__rs@rKr r s]    *- ' 82Vh 4rMr cJeZdZeddZdddd fdZd dZd dZxZS) HexBinPlotcy)NhexbinrXrRs rKrTzHexBinPlot._kindsrMT)r c t||||fi|t|r=|jjj s|jj|}||_||_t|j|jjdk(rt|jdzt|j|jjdk(rt|jdzy)Nrz requires x column to be numericz requires y column to be numeric)rrrr[rprCr rr_get_numeric_datarkrTr)rSr[rrrAr rrs rKrzHexBinPlot.__init__s q!.v. a=!2!2!A!A!C !!!$A   tyy 224 5 :TZZ*LLM M tyy 224 5 :TZZ*LLM M ;rMc|j|j|j|jf\}}}}|jd}|j xsd}t jj|}|j}|d} n||j} |j||j||jf| |d|j|r|j||yy)NrBuGn)rArg)r)rrr[rArrrzr1r2r rr?rr ) rSrrrr[rArrgr&r%s rKrzHexBinPlot._make_plots 4669 1dA YYq\}}&}}%%d+ ]] 9HAw~~H $q'..$q'..UH4U499U     , rMcyrWrXrRs rKrzHexBinPlot._make_legend rM)rzLiteral['hexbin']rW)r rrrrr) rrrrrTrrrr:r;s@rKr=r=s-  Nt N-" rMr=ceZdZdZeddZeddZddZeddZ eddZ ddZ e d dd Z eddd Zedd Zee dd Zee dd Zee ddZddZy)LinePlotrcyNrUrXrRs rKrYzLinePlot.orientationrMcy)NrrXrRs rKrTzLinePlot._kindrMc ddlm}tj||fi||jr!|j j d|_|d|_d|jvr*t|jjd|_yy)Nr) plot_paramsrx_compat) pandas.plottingrOrOrr`r[fillnarQrrr)rSr[rrOs rKrzLinePlot.__init__sp/t.v. << ((q(1DI#J/  " z!:;DM #rMc\|j xr|jxr|jSrW)rQr_use_dynamic_xrRs rK _is_ts_plotzLinePlot._is_ts_plots(== MT^^M8K8K8MMrMcLt|jd|jSr )r4rEr[rRs rKrUzLinePlot._use_dynamic_xsT\\!_dii88rMc |jrNt|jd|j}|j}|j }|j }n8|j}|j}|j|j}|j}tj|jj}|j}t!|D] \} \} } |j| } |j"j%} |j&|j&| d<|j)|| | | \}} |j+| | }t-| fi|} t/| } |j1| | } | | d<|| || f|| ||d| }|j3|d| |jst5| }t7|\}}| j9|| y)Nrr%rhrarnrnra)rJ column_num stacking_idr)rVr2rEr[rn_ts_plotrrrr_get_stacking_idrlrrrrrrr.rhrrrr*rorrr8r9rX)rSrr[rplotfitr\rrrrarrrrJrnewlinesrleftrights rKrzLinePlot._make_plots    &t||A BD AMMEB  "AJJEdii0B++- && (:(:(<= !!#&r]MAzqaB99>>#Dzz% $ W 22 KE4((uA(>F''D 'E**5*:E!DM ''  H  . .x{E B!&b)&uo e D%(M+rMNc |dk(r|j||t||j||||d}tj|||fd|i|} |j |||| S)NrrarJ)_initialize_stackerr_get_stacked_valuesrOr_update_stacker) rrrrrJr[r\ry_valuesrs rKrzLinePlot._plot"sn ?  # #B SV <**2{AtG}M b!XCUCdC B Q/ rMc t|||\}}t||t|drt|j|t|drt|j||j j ||j|f|j||jtj|jfd|i|}t||j|j|S)NrrrJ)r3r0rrr _plot_datarrTrrnrrrr1freq)rSrrr[rJrrkrs rKr]zLinePlot._ts_plot6s $D"d3 d b$ 2y ! "**d + 2z " "++t , dDJJ56 2tzz2::dkk+BX%XSWX BGGTZZ0 rMcF|jrt|jSyrW)r`idr[rRs rKr^zLinePlot._get_stacking_idMs <<dii= rMc|yt|dsi|_t|dsi|_tj||j|<tj||j|<y)N_stacker_pos_prior_stacker_neg_prior)rrorprzeros)rrr\ns rKrezLinePlot._initialize_stackerTsc   r/0$&B !r/0$&B !-/XXa[k*-/XXa[k*rMc ||St|ds|j||t||dk\jr|j||zS|dkjr|j ||zSt d|d)NrorzWWhen stacked is True, each column must be either all positive or all negative. Column 'z,' contains both positive and negative values)rrerallrorprk)rrr\rras rKrfzLinePlot._get_stacked_valuesds  Mr/0  # #B S[ A aK   %%k2 k   %%k2   gI K  rMc|y|dk\jr|j|xx|z cc<y|dkjr|j|xx|z cc<yyr )rtrorp)rrr\rs rKrgzLinePlot._update_stackers^    aK     ! !+ .& 8 .k    ! !+ .& 8 .!rMcjddlm}fd}|jrW|j}|Dcgc] }|| }}|jj |||j ||j xrOjjxr |jxr)|j xs|jxr |j}|j} |r*|jsd|_t!||j| |jr|j#| yyycc}w)Nr) FixedLocatorct|r|jr t|} tj|S#t $rYywxYw)Nr()rrrr*rn Exception)rr[s rK get_labelz,LinePlot._post_plot_logic..get_labelsE{q||~F #DJJqM22  sA A  A )r)matplotlib.tickerrwr get_xticksrIset_major_locatorset_xticklabelsrUrn _is_all_datesrrurwrrrr7r) rSrr[rwrzr{r xticklabels conditionrs ` rKrzLinePlot._post_plot_logics 2   " "]]_F178A9Q<K8 HH & &|F'; <   { +##% % G))> B Q/f rMcltj|||tt|j dkD}|j l|si|dk\j j r|jddy|dkj j r|jddyyyyr )rHrrrget_shared_y_axesr~rtrW)rSrr[ is_shared_ys rKrzAreaPlot._post_plot_logics!!$D1$r33567!; 99 [  $$& At$!)"&&( D!$)&1 rM)rzLiteral['area']r)NNNFrr) rrrrrTrrrrrXrMrKrrs^ L& !+ +  +++Z %rMrceZdZeddZdZeddZddddddd dd Zedd Z ed Z ed Z e d dd Z edZddZddZ ddZy)BarPlotcy)NrrXrRs rKrTz BarPlot._kind rMZcyrJrXrRs rKrYzBarPlot.orientationrKrMcenterrrF)alignbottomrbwidthpositionrc vt|t|_||_||_||_t jt||_ t|rt j|}t|rt j|}||_ ||_ ||_tj ||fi|yrW)ror& _is_series bar_width_align _positionrarangertick_posrrrrbrrOr) rSr[rrrbrrrrs rKrzBarPlot.__init__s%T95 ! #d),  XXf%F  88D>D  t.v.rMc4|j|jz SrW)r tickoffsetrRs rKax_poszBarPlot.ax_pos5s}}t..rMc*|js |jr|j|jzS|jdk(r;|j|j z }|j|jdz z|dzzS|j|jzS)Nedger)r`rurrrrrSws rKrzBarPlot.tickoffset9st <<4==>>DNN2 2 [[F "-A>>T^^c%9:QWD D>>DNN2 2rMc|js |jr|jdk(r|jdz Sy|jdk(r|j|jz }|dzSy)Nrrrr)r`rurrrrs rK lim_offsetzBarPlot.lim_offsetCsW <<4=={{f$~~)) [[F "-As7NrMc 2|j|||f||d|S)N)rr)rrrrrrrrrs rKrz BarPlot._plotQs%rvvaA=e===rMc|jSrW)rrRs rK _start_basezBarPlot._start_base^rrMc0|j}t|}tjt|jx}}|j }|jj d}t|j|D]\}\} } |j|} |jj} |jr|| d<n$t|tr || | d<n |||z| d<|j| |} t| fi| } t!| } |j#| |} d| vsd| vr'| j%dt&j(d | d<d}|j*r| d k\j-rd }||j.z}|j0| d <|j2r\|j4d z }|j6| |j8|z| |j4f|| |j*d | }| j;| n|j<r| dkD}tj>||||j.z}|j4d z }|j6| |j8|z| |j4f|| |j*d | }|tj>|| dz}|tj>|d| z}nF|j4|z }|j6| |j8|dz|zz| |f|| |j*d | }|jA|| y)Nrr%rhrYrZrfrerz xtick.colorrrr)rrarr)!rrrrqr[rrSrrrErr.rrorrr*rorrzrrrtrrrurrrr[r`whererr)rSrrncolors pos_prior neg_priorKr[rrarrrrrrrrs rKrzBarPlot._make_plotbs!!#f+ "TYY 88 I LLyy"&tD'ABMAzqaB99>>#D &W FD) &u W &q7{ 3W ((uA(>F''D 'E**5*:E4Vt^488H;M;U!$m!<XExxQ!VLLND,,,E KKDM}}NNQ&!tzzKK!ONN      U#1uy)>6:jjAjsc*jIA6;DJJqM6JK6Jsc*6JIKQ$&8R4'$..84??J R!5!5!7FFSBKs C  Cc:|j||f|j/|jtj|jn,|j|j |j |||jr|j|yyyrW) rXr{rVrrrrrrrSrr) ticklabels start_edgeend_edges rKrzBarPlot._decorate_tickssv Z*+ ;; " MM"((4;;/ 0 MM$-- (   z *   MM$ !/ rMN)rzLiteral['bar', 'barh']rr)rrrFrr=rrrzint | npt.NDArray[np.intp]rrrr rr=r)rrz list[str]rr rr rr)rrrrrTrrYrrrrrrrrrrrrXrMrKrr s. L   / />//33  -. >  >  > * > > >J>    ! |+NOP Pt9$9&9rMct||||dur(tjd|dtt y)N)rrFzPiePlot ignores the 'z ' keywordr)rrrrrr)rrrrs rKrzPiePlot._validate_log_kwd sE !c!7   MM'uI6+-  rMcyrWrX)rSrhrs rKrzPiePlot._validate_color_argsrZrMc |jt|jd}|jj d|t |j |jD]O\}\}}|j|}|t|}|j||jj}d}|jjD cgc] } t| } } |jd| } | &t| |D cgc]\} } || | }} } nd}|j|fd|i|}|jdd|\}}}n|\}}g}|j %||zD]}|j#|j | | n| }t||D]\}}|j%||Rycc} wcc} } w)Nr)rrr%c|dk(ry|S)Nrr(rX)rars rK blank_labelerz)PiePlot._make_plot..blank_labeler,sA: LrMreautopct)rrr[rrrrrEr*rYr.rnrrZrrrrcrr)rSrrrrarrrrvidxrerbrblabelsresultspatchestexts autotextst leglabels_patch _leglabels rKrzPiePlot._make_plot s!!S^!Q Xv.&tDII'FGMAzqaB $U+ e$99>>#D ! -1IIOOmatplotlib.figurer?pandas._typingr@rArBrCpandasrDrLrOrr r=rHrrrrrXrMrKrs9" -24    , ''3-WHCA($$(2n(cn(b#3030lG)GT% % Pq&wq&hO%xO%d g D0#w0#fM gM rM