B x\ 2@sddlmZddlmZddlZddlZddlmZddl m Z ddl m Z m Z mZddZdd d Zdd d Zd ddZddZddZddZddZddZd!ddZdS)")division)ceilN)range) is_list_like) ABCDataFrame ABCIndexClass ABCSeriescCsXy>x$|D]}|d||q W|}|jddWntk rRYnXdS)Nrightg?)Zbottom)get_xticklabelsZset_haZ set_rotation get_figureZsubplots_adjust Exception)axZrotZlabelfigr5lib/python3.7/site-packages/pandas/plotting/_tools.pyformat_date_labelss rcKsrt|tr|}nt|tr ntd|dkr6|j}|dkrD|j}|j}ddl}|j j |f|||d|}|S)a Helper function to convert DataFrame and Series to matplotlib.table Parameters ---------- ax : Matplotlib axes object data : DataFrame or Series data for table contents kwargs : keywords, optional keyword arguments which passed to matplotlib.table.table. If `rowLabels` or `colLabels` is not specified, data index or column name will be used. Returns ------- matplotlib table object z&Input data must be DataFrame or SeriesNr)cellText rowLabels colLabels) isinstancerZto_framer ValueErrorindexcolumnsvaluesZmatplotlib.tabletable)r datarrkwargsrZ matplotlibrrrrrs    rboxc Cs~|dk rt|ttfr"t|dkr*td|\}}dd}|dkrh|dkrh|t|||f}\}}nJ|dkr|dkr||t||f}\}}n|dkr|dkrd}t||||krtdj|||d |S|d krd S|d krd |fS|dkr|d fSd dddd}y||Stk rxd }x|d|krL|d 7}q2W|d ||krl||d fS||fSYnXdS)Nz)Layout must be a tuple of (rows, columns)cSs tt|S)N)intr)xrrrLsz_get_layout..rz1At least one dimension of layout must be positivezDLayout of {nrows}x{ncols} must be larger than required size {nplots})nrowsncolsnplotsZsingle)r&Z horizontalr&Zvertical)r&r)rr)r&r)rtuplelistlenrfloatformatKeyError) r%layout layout_typer#r$Zceil_msgZlayoutskrrr _get_layoutDs@   r3FTcKsddlm} |dkri}|dkr.| jf|} nt|rt|}|dk rRtdt|sZ|rjtjdtddt||kr|d } | |fSt d || } |dkr|r| |fS| t|fSntjd tdd| t |||d \} } | | } tj| td }| j| | df|}|r"||d <|r0||d <||d<xTtd| D]F}|}||krjd|d <d|d <| j| | |df|}|||<qDW|| krx ||dD]}|dqWt|| || | |||r| dkr|d}n|| | }n || | }| |fS)ax Create a figure with a set of subplots already made. This utility wrapper makes it convenient to create common layouts of subplots, including the enclosing figure object, in a single call. Keyword arguments: naxes : int Number of required axes. Exceeded axes are set invisible. Default is nrows * ncols. sharex : bool If True, the X axis will be shared amongst all subplots. sharey : bool If True, the Y axis will be shared amongst all subplots. squeeze : bool If True, extra dimensions are squeezed out from the returned axis object: - if only one subplot is constructed (nrows=ncols=1), the resulting single Axis object is returned as a scalar. - for Nx1 or 1xN subplots, the returned object is a 1-d numpy object array of Axis objects are returned as numpy 1-d arrays. - for NxM subplots with N>1 and M>1 are returned as a 2d array. If False, no squeezing is done: the returned axis object is always a 2-d array containing Axis instances, even if it ends up being 1x1. subplot_kw : dict Dict with keywords passed to the add_subplot() call used to create each subplots. ax : Matplotlib axis object, optional layout : tuple Number of rows and columns of the subplot grid. If not specified, calculated from naxes and layout_type layout_type : {'box', 'horziontal', 'vertical'}, default 'box' Specify how to layout the subplot grid. fig_kw : Other keyword arguments to be passed to the figure() call. Note that all keywords not recognized above will be automatically included here. Returns: fig, ax : tuple - fig is the Matplotlib Figure object - ax can be either a single axis object or an array of axis objects if more than one subplot was created. The dimensions of the resulting array can be controlled with the squeeze keyword, see above. **Examples:** x = np.linspace(0, 2*np.pi, 400) y = np.sin(x**2) # Just a figure and one subplot f, ax = plt.subplots() ax.plot(x, y) ax.set_title('Simple plot') # Two subplots, unpack the output array immediately f, (ax1, ax2) = plt.subplots(1, 2, sharey=True) ax1.plot(x, y) ax1.set_title('Sharing Y axis') ax2.scatter(x, y) # Four polar axes plt.subplots(2, 2, subplot_kw=dict(polar=True)) rNz5When passing multiple axes, layout keyword is ignoredznWhen passing multiple axes, sharex and sharey are ignored. These settings must be specified when creating axesr() stacklevelzBThe number of passed axes must be {0}, the same as the output plotr&zSTo output multiple subplots, the figure containing the passed axes is being cleared)r/r0)dtypesharexshareyF)matplotlib.pyplotpyplotZfigurer_flattenwarningswarn UserWarningr+r rr-clearr3npemptyobjectZ add_subplotrcopy set_visible_handle_shared_axesZreshapesqueeze)naxesr6r7rEZ subplot_kwr r/r0Zfig_kwpltrr#r$r%axarrZax0ikwdsaxesrrr _subplotssslL        rLcCsx|D]}|dq Wyjddlm}t||jrH||t| |j rh| | dx| D]}|dqrWWntk rYnX|ddS)NFr)Zget_majorticklabelsrCZmatplotlib.tickertickerrZget_minor_locatorZ NullLocatorZset_minor_locatorZ AutoLocatorZget_minor_formatterZ NullFormatterZset_minor_formatterZFormatStrFormatterZget_minorticklabelsr Z get_label)ZaxistrNrrr_remove_labels_from_axiss rPc Cs<|dkr8|dkrytj|d|dftjd}x |D]}|||j|jf<q6WxF|D]>}||jd|jfsrqX|st||dkrXt |j qXWWnPt k rx:|D]2}| rq|st||dkrt |j qWYnX|dkr8x>|D]6}| rq|s*t||dkrt |jqWdS)Nr&)r5)r?ZzerosboolZ get_visibleZrowNumZcolNumr+Zget_shared_x_axesZ get_siblingsrPZxaxis IndexErrorZ is_last_rowZ is_first_colZget_shared_y_axesZyaxis) rHr%rFr#r$r6r7r/r rrrrD*s2         rDcCs6t|st|gSt|tjtfr,|St|S)N)rr?ZarrayrZndarrayrZravel)rKrrrr:Ts  r:cCs<|}t|dr ||j7}t|dr8||j7}|S)Nright_axleft_ax) get_lineshasattrrSrT)r linesrrr_get_all_lines\s   rXcCsRtjtj }}x8|D]0}|jdd}tt||}tt||}qW||fS)NF)Zorig)r?infZ get_xdataminZnanminmaxZnanmax)rWleftr lr rrr _get_xlimhs   r^cCsddlm}xxt|D]l}|dk r4|j||d|dk rN|j||d|dk rh|j||d|dk r|j||dqW|S)Nr)Zfontsize)Zrotation)r8r9r:Zsetpr Zget_yticklabels)rKZ xlabelsizeZxrotZ ylabelsizeZyrotrGr rrr_set_ticks_propsqs r_)NN)Nr)NFFTNNNr)NNNN)Z __future__rZmathrr;Znumpyr?Z pandas.compatrZpandas.core.dtypes.commonrZpandas.core.dtypes.genericrrrrrr3rLrPrDr:rXr^r_rrrrs&     ( / $*