U ~fh%;@sLddlmZddlmZmZddlZddlmZddl m Z m Z ddl m Z mZddlmZmZddlmZdd lmZmZdd lmZmZdd lmZdd lmZmZmZm Z erdd l!m"Z"ddl#m$Z$GdddeZ%Gddde%Z&d'dddddddZ'd(dddddddddZ(d)dddd!d"d#Z)d*dddddd$d%d&Z*dS)+) annotations) TYPE_CHECKINGLiteralN)PlottingOrientation) is_integer is_list_like) ABCDataFrameABCIndex)isnaremove_na_arraylike) pprint_thing)LinePlotMPLPlot)create_iter_data_given_byreformat_hist_y_given_by)unpack_single_str_list)create_subplots flatten_axesmaybe_adjust_figureset_ticks_props)Axes) DataFramec@seZdZeddddZd$ddd d d d Zd dd dZdddddZed%dddddZ d dddZ ddZ dd ddd Z ed!dd"d#Z dS)&HistPlotzLiteral[('hist', 'kde')]returncCsdS)NhistselfrrD/tmp/pip-unpacked-wheel-a5acpmi0/pandas/plotting/_matplotlib/hist.py_kind1szHistPlot._kind rz#int | np.ndarray | list[np.ndarray]zint | np.ndarrayNone)binsbottomrcKs8||_||_|d|_|d|_tj||f|dS)Nxlabelylabel)r#r$getr%r&r__init__)rdatar#r$kwargsrrrr(5s   zHistPlot.__init__csptjrTjdk rFtj}j|j}fdd|D_nj_tj rlt j _ dS)Ncsg|]\}}|qSr)_calculate_bins).0keygrouprrr Jsz)HistPlot._args_adjust..) rr#byrr)groupbycolumnsr+rr$nparray)rZ by_modifiedgroupedrrr _args_adjustCs    zHistPlot._args_adjustrz np.ndarray)r)rcCsL|jdd}t|}|t|}tj||j|jddd\}}|S)zCalculate bins given dataF)copyrangeN)r#r8) Z infer_objects_get_numeric_datar3Zravelr Z histogramr#kwdsr')rr)Z nd_valuesvaluesrr#rrrr+Qs  zHistPlot._calculate_binsNint)r$ column_numc Ksx|dkr|||t|dtt|d} ||||| |d}|j|f||d|\} }} |||| | S)Nrlabel)r#r$)Z_initialize_stackerlenr3zerosZ_get_stacked_valuesrZ_update_stacker) clsaxystyler$r= stacking_idr#r:basenZpatchesrrr_plot]s zHistPlot._plotc Cs|}|}|jdk r(t|j|jn|j}t|j|dD]\}\}}||}|j }t |}|j ||d}||d<| ||||\} }| dk r| |d<|||}|jdk r|d||d<|j|d<|d|dd} | dk rpt| dkrZt| d dkrZz| dd|f} Wn.tk rX} ztd | W5d} ~ XYnX| t|} | |d<t||j}|j||f||d |} |jdk r|t ||| d |q>dS) N)r))indexr?rEr#colorweightsr>z?weights must have the same shape as data, or be a single column)r=rFr)Z _get_colorsZ_get_stacking_idr0rr)r enumerateZ _iter_dataZ_get_axr:r7r Z_mark_right_labelZ_apply_style_colors_make_plot_keywordsr2popr'r3ndimshape IndexError ValueErrorr rrI set_titleZ_append_legend_handles_labels) rcolorsrFr)ir?rDrCr:rErLerrZartistsrrr _make_plottsL        $  zHistPlot._make_plotcCs|j|d<|j|d<|S)z/merge BoxPlot/KdePlot properties to passed kwdsr$r#)r$r#rr:rDrrrrOs  zHistPlot._make_plot_keywordsr)rCrcCs\|jdkr2||jdkrdn|j||jn&||j||jdkrPdn|jdS)N horizontalZ Frequency) orientationZ set_xlabelr% set_ylabelr&rrCr)rrr_post_plot_logics   zHistPlot._post_plot_logicrcCs|jdddkrdSdSdS)Nr\r[vertical)r:r'rrrrr\szHistPlot.orientation)r!r)NrrN)__name__ __module__ __qualname__propertyr r(r6r+ classmethodrIrYrOr_r\rrrrr0s$ ;rc@szeZdZeddddZeddddZdd dd d Zd dd d ZddZe dddZ ddZ d dddZ dS)KdePlotzLiteral['kde']rcCsdS)NZkderrrrrr sz KdePlot._kindzLiteral['vertical']cCsdS)Nr`rrrrrr\szKdePlot.orientationNr"cKs tj||f|||_||_dSN)rr( bw_methodind)rr)rhrir*rrrr(szKdePlot.__init__cCsdSrgrrrrrr6szKdePlot._args_adjustcCs|jdkrJt|t|}tt|d|t|d|d}nRt|jrt|t|}tt|d|t|d||j}n|j}|S)N?i)rir3ZnanmaxZnanminZlinspacer)rrDZ sample_rangerirrr_get_inds   zKdePlot._get_indc KsHddlm} t|}| ||d} | |}tj|||fd|i|} | S)Nr) gaussian_kde)rhrE)Z scipy.statsrlr evaluaterrI) rBrCrDrErhrir=rFr:rlZgkdelinesrrrrIs   z KdePlot._plotcCs|j|d<|||d<|S)Nrhri)rhrkrZrrrrOs zKdePlot._make_plot_keywordscCs|ddS)NZDensity)r]r^rrrr_szKdePlot._post_plot_logic)NN)NNNNN) rarbrcrdr r\r(r6rkrerIrOr_rrrrrfs  rfTboolfloat) numeric_onlysharexshareyrotc Ks|dkrtd||} |dk r*| |} t| } t| |||| |d\}}t|}t| D]F\}\}}||} |rt|tr|}||| f| | t |qZ||fS)NdefaultzNfigsize='default' is no longer supported. Specify figure size by tuple instead)naxesfigsizerrrsrClayout) rTr1r@rrrN isinstancerr9rUr )Zplotfr)columnr0rqrwrrrsrxrtrCr*r5rvfigaxes_axesrWr-r.rrr _grouped_plot s0  r~2FZr<)r#rrrsrtgridlegendc srBdkst|jdkr&|jd<n|dkr:|jd<n|d<ddfdd }| dkrd| } t|||||||||| d \}}t|| | | |d t|d d d d d dd|S)a Grouped histogram Parameters ---------- data : Series/DataFrame column : object, optional by : object, optional ax : axes, optional bins : int, default 50 figsize : tuple, optional layout : optional sharex : bool, default False sharey : bool, default False rot : float, default 90 grid : bool, default True legend: : bool, default False kwargs : dict, keyword arguments passed to matplotlib.Axes.hist Returns ------- collection of Matplotlib Axes r?r>Nr"rcs,|j|jfdir(|dS)Nr#)rdropnar;r)r.rCr#r*rrr plot_groupgsz!_grouped_hist..plot_group)rzr0rrrsrCrwrxrt xlabelsizexrot ylabelsizeyrotg333333?g?g?rj333333?)r$topleftrighthspacewspace)AssertionErrorrQnamer2r~rr)r)rzr0rCr#rwrxrrrsrtrrrrrrr*rr{r|rrr _grouped_hist4sN*     rr!)rr#rc  Ksddlm} | r d| kr td|dkr| dddk rBtd| d| rX| n | j|d} |dk rt|t| kr| j |dd i|dkr| }n| | krt d |j}| r|j| d<|j|fd | i| | r|||t|g}t|||||d n8d| kr.td t|f||||| ||||| d | }t|dr|jdkrt|dkr|dS|S)Nrr? Cannot use both legend and labelrxz7The 'layout' keyword is not supported when 'by' is Nonefigure)rwZforwardTz&passed axis not bound to passed figurer#rz`Cannot pass 'figure' when using the 'by' argument, since a new 'Figure' instance will be created) r0rCrrwr#rrrrrrQr>)Zmatplotlib.pyplotZpyplotrTr'rPZ get_fignumsZgcfrtupleZget_size_inchesZset_size_inchesZgcaZ get_figurerrr;rrrrr3r4rrhasattrrQr@)rr0rCrrrrrrwr#rr:Zpltr{r;r|rrr hist_seriessn            r)rrrrsr#rcKs`|rd|krtd|dk rNt|f||||| | | | | |||||d|}|S|dk rvt|ttjtfsn|g}||}|jtjddfdd}t |j }|dkrtd t ||d | | | | d \}}t |}d|k}t |j D]`\}}||}|r|r||d<|j||jfd | i||||||r|qt|||||d t|ddd|S)Nr?r)rzr0rCrrwrrrsrxr#rrrrrZ datetime64Z datetimetz timedelta)includeexcluderzDhist method requires numerical or datetime columns, nothing to plot.F)rvrCZsqueezerrrsrwrxr#rr)rr)rTrrylistr3Zndarrayr Z select_dtypesnumberr@r2rrrNrrr;rUrrrr)r)rzr0rrrrrrCrrrsrwrxr#rr:r|rvr{r}Z can_set_labelrWcolrrr hist_frames        r) NNTNTTNrN)NNNrNNFFrTNNNNF) NNTNNNNNr!F)NNTNNNNNFFNNr!F)+ __future__rtypingrrZnumpyr3Zpandas._typingrZpandas.core.dtypes.commonrrZpandas.core.dtypes.genericrr Zpandas.core.dtypes.missingr r Zpandas.io.formats.printingr Z pandas.plotting._matplotlib.corer rZ#pandas.plotting._matplotlib.groupbyrrZ pandas.plotting._matplotlib.miscrZ!pandas.plotting._matplotlib.toolsrrrrZmatplotlib.axesrZpandasrrrfr~rrrrrrrs      G,TJ