B Jdo@sdZddlZddlZddlmZddlZddlZddlm Z m Z m Z m Z ddl mZeeZGdddZGdd d eZGd d d eZGd d d ZdS)a :mod:`~matplotlib.gridspec` contains classes that help to layout multiple `~.axes.Axes` in a grid-like pattern within a figure. The `GridSpec` specifies the overall grid structure. Individual cells within the grid are referenced by `SubplotSpec`\s. Often, users need not access this module directly, and can use higher-level methods like `~.pyplot.subplots`, `~.pyplot.subplot_mosaic` and `~.Figure.subfigures`. See the tutorial :doc:`/tutorials/intermediate/arranging_axes` for a guide. N)Integral)_api_pylab_helpers tight_layoutrcParams)Bboxc@seZdZdZd'ddZddZeddd d Zed dd d Zd dZ d(ddZ d)ddZ ddZ ddZ ddZddZd*ddZedd Zd!d"Zddd#dd$d%d&ZdS)+ GridSpecBasezm A base class of GridSpec that specifies the geometry of the grid that a subplot will be placed. NcCsft|tr|dkr td|t|tr2|dkr@td||||_|_||||dS)a Parameters ---------- nrows, ncols : int The number of rows and columns of the grid. width_ratios : array-like of length *ncols*, optional Defines the relative widths of the columns. Each column gets a relative width of ``width_ratios[i] / sum(width_ratios)``. If not given, all columns will have the same width. height_ratios : array-like of length *nrows*, optional Defines the relative heights of the rows. Each row gets a relative height of ``height_ratios[i] / sum(height_ratios)``. If not given, all rows will have the same height. rz/Number of rows must be a positive integer, not z2Number of columns must be a positive integer, not N) isinstancer ValueError_nrows_ncolsset_height_ratiosset_width_ratios)selfnrowsncols height_ratios width_ratiosrE/var/www/html/venv/lib/python3.7/site-packages/matplotlib/gridspec.py__init__!s   zGridSpecBase.__init__cCsbtt|jdkrd|jfnd}tt|jdkr@d|jfnd}dj|jj|j|j||dS)Nz, height_ratios=%rz, width_ratios=%rz&{clsname}({nrows}, {ncols}{optionals}))clsnamerr optionals) lenset_row_height_ratios_col_width_ratiosformat __class____name__r r )rZ height_argZ width_argrrr__repr__:s""zGridSpecBase.__repr__cCs|jS)N)r )rrrrFzGridSpecBase.zThe number of rows in the grid.)doccCs|jS)N)r )rrrrr#Hr$z"The number of columns in the grid.cCs |j|jfS)zW Return a tuple containing the number of rows and columns in the grid. )r r )rrrr get_geometryKszGridSpecBase.get_geometrycCsdS)Nr)rfigurerrrget_subplot_paramsQszGridSpecBase.get_subplot_paramsrcCs(|\}}|||||||f}|S)a[ Create and return a `.SubplotSpec` instance. Parameters ---------- loc : (int, int) The position of the subplot in the grid as ``(row_index, column_index)``. rowspan, colspan : int, default: 1 The number of rows and columns the subplot should span in the grid. r)rlocrowspancolspanloc1loc2Z subplotspecrrrnew_subplotspecUs zGridSpecBase.new_subplotspeccCs6|dkrdg|j}nt||jkr,td||_dS)z Set the relative widths of the columns. *width_ratios* must be of length *ncols*. Each column gets a relative width of ``width_ratios[i] / sum(width_ratios)``. NrzTExpected the given number of width ratios to match the number of columns of the grid)r rr r)rrrrrres zGridSpecBase.set_width_ratioscCs|jS)zo Return the width ratios. This is *None* if no width ratios have been set explicitly. )r)rrrrget_width_ratiossszGridSpecBase.get_width_ratioscCs6|dkrdg|j}nt||jkr,td||_dS)z Set the relative heights of the rows. *height_ratios* must be of length *nrows*. Each row gets a relative height of ``height_ratios[i] / sum(height_ratios)``. NrzRExpected the given number of height ratios to match the number of rows of the grid)r rr r)rrrrrr {s zGridSpecBase.set_height_ratioscCs|jS)zq Return the height ratios. This is *None* if no height ratios have been set explicitly. )r)rrrrget_height_ratiosszGridSpecBase.get_height_ratiosFcst|\}}|r*d}d}d}d}d} d} n.||} | j}| j}| j}| j}| j} | j} ||} ||} | || |d}| |}||t|j fdd|j D}dg|g|d}t t ||gj }| || |d}| |}||t|jfdd|jD}dg|g|d}t t ||gj }||dj\}}||dj\}}||||fS) a Return the positions of the grid cells in figure coordinates. Parameters ---------- fig : `~matplotlib.figure.Figure` The figure the grid should be applied to. The subplot parameters (margins and spacing between subplots) are taken from *fig*. raw : bool, default: False If *True*, the subplot parameters of the figure are not taken into account. The grid spans the range [0, 1] in both directions without margins and there is no space between grid cells. This is used for constrained_layout. Returns ------- bottoms, tops, lefts, rights : array The bottom, top, left, right positions of the grid cells in figure coordinates. gg?rcsg|] }|qSrr).0r)normrr sz3GridSpecBase.get_grid_positions..rcsg|] }|qSrr)r1r2)r3rrr4s))r&r(leftrightbottomtopwspacehspacesumrnpZcumsumZ column_stackflatrZreshapeT)rZfigrawrrr7r8r9r:r;r<Zsubplot_paramsZ tot_widthZ tot_heightZcell_hZsep_hZ cell_heightsZ sep_heightsZcell_hsZcell_wZsep_wZ cell_widthsZ sep_widthsZcell_wsfig_tops fig_bottoms fig_lefts fig_rightsr)r3rget_grid_positionss@  zGridSpecBase.get_grid_positionscCs`xP|D]D}t|dr |}t|dr:|}|||fkr |Sq Wt|||dS)zo Check if the figure already has a gridspec with these dimensions, or create a new one get_subplotspecget_topmost_subplotspec)r')Zget_axeshasattrrG get_gridspecrHr&GridSpec)r'rraxgsrrr_check_gridspec_existss    z#GridSpecBase._check_gridspec_existsc Cs|\}}dd}t|try |\}}Wn,tk rV}ztd|Wdd}~XYnXt|||d|||dg||f\}} n||||d\}} t||| S)z,Create and return a `.SubplotSpec` instance.cSs|}t|tr<||\}}}||kr2||dfStdnh|dkrL||}d|kr`|krlnn||fS|dk rtd|d|d|ntd|d|dS)Nrz=GridSpec slice would result in no space allocated for subplotrzindex z is out of bounds for axis z with size z) is out of bounds for GridSpec with size )r sliceindices IndexError)keysizeZaxisorig_keystartstop_rrr _normalizes   z,GridSpecBase.__getitem__.._normalizezUnrecognized subplot specNrr)r&r tupler r>Zravel_multi_index SubplotSpec) rrRrrrXZk1Zk2errnum1num2rrr __getitem__s   zGridSpecBase.__getitem__T)sharexshareysqueeze subplot_kwc Cs|j}|dkrtdt|tr,|r(dnd}t|trB|r>dnd}t|trVtdtjddddg||d|dkrzi}|}t j |j |j ft d }x~t|j D]p}xjt|j D]\}d|d ||d f|d |fd } | ||d <| ||d<|j|||ff||||f<qWqW|dkr@x|jD]} | jddq*W|dkrhx|jD]} | jddqRW|r|jdkr|S|S|SdS)z Add all subplots specified by this `GridSpec` to its parent figure. See `.Figure.subplots` for detailed documentation. NzIGridSpec.subplots() only works for GridSpecs created with a parent figureallnonez]sharex argument to subplots() was an integer. Did you intend to use subplot() (without 's')?rowcol)r_r`)Zdtype)rrr)rdrcrerfr_r`)rfrcT)Z check_patch)rercr)r'r r boolrr warn_externalZ check_in_listcopyr>emptyr r objectrangeZ add_subplotr?Z_label_outer_xaxisZ_label_outer_yaxisrSitemra) rr_r`rarbr'ZaxarrrerfZ shared_withrLrrrsubplotssB        "    zGridSpecBase.subplots)NN)N)rr)F)r! __module__ __qualname____doc__rr"propertyrrr&r(r.rr/r r0rF staticmethodrNr^rnrrrrrs&     = %rc sVeZdZdZdfdd Zddddd d gZd d Zdd dZddZdddZ Z S)rKz A grid layout to place subplots within a figure. The location of the grid cells is determined in a similar way to `~.figure.SubplotParams` using *left*, *right*, *top*, *bottom*, *wspace* and *hspace*. Nc sB||_||_||_||_||_| |_||_tj||| | ddS)a Parameters ---------- nrows, ncols : int The number of rows and columns of the grid. figure : `~.figure.Figure`, optional Only used for constrained layout to create a proper layoutgrid. left, right, top, bottom : float, optional Extent of the subplots as a fraction of figure width or height. Left cannot be larger than right, and bottom cannot be larger than top. If not given, the values will be inferred from a figure or rcParams at draw time. See also `GridSpec.get_subplot_params`. wspace : float, optional The amount of width reserved for space between subplots, expressed as a fraction of the average axis width. If not given, the values will be inferred from a figure or rcParams when necessary. See also `GridSpec.get_subplot_params`. hspace : float, optional The amount of height reserved for space between subplots, expressed as a fraction of the average axis height. If not given, the values will be inferred from a figure or rcParams when necessary. See also `GridSpec.get_subplot_params`. width_ratios : array-like of length *ncols*, optional Defines the relative widths of the columns. Each column gets a relative width of ``width_ratios[i] / sum(width_ratios)``. If not given, all columns will have the same width. height_ratios : array-like of length *nrows*, optional Defines the relative heights of the rows. Each row gets a relative height of ``height_ratios[i] / sum(height_ratios)``. If not given, all rows will have the same height. )rrN) r7r9r8r:r;r<r'superr) rrrr'r7r9r8r:r;r<rr)r rrrOs* zGridSpec.__init__r7r9r8r:r;r<cKsx:|D].\}}||jkr*t|||q t|dq WxdtjjD]T}xN|jj j D]@}t |t j j rZ|}||krZ|||j qZWqJWdS)a  Update the subplot parameters of the grid. Parameters that are not explicitly given are not changed. Setting a parameter to *None* resets it to :rc:`figure.subplot.*`. Parameters ---------- left, right, top, bottom : float or None, optional Extent of the subplots as a fraction of figure width or height. wspace, hspace : float, optional Spacing between the subplots as a fraction of the average subplot width / height. z is an unknown keywordN)items _AllowedKeyssetattrAttributeErrorrZGcfZfigsvaluesZcanvasr'axesr mplZ SubplotBaserGrHrJZ _set_position get_position)rkwargskvZ figmanagerrLssrrrupdates   zGridSpec.updatecsT|dkr(ddjD}tjjf|}n t|j}|jffddjD|S)z Return the `.SubplotParams` for the GridSpec. In order of precedence the values are taken from - non-*None* attributes of the GridSpec - the provided *figure* - :rc:`figure.subplot.*` NcSsi|]}td||qS)zfigure.subplot.)r)r1r~rrr sz/GridSpec.get_subplot_params..csi|]}t||qSr)getattr)r1r~)rrrrs)rvr{r' SubplotParamsri subplotparsr)rr'kwrr)rrr(s  zGridSpec.get_subplot_paramscsfddjDS)z Return a list of the names of the subplot parameters explicitly set in the GridSpec. This is a subset of the attributes of `.SubplotParams`. csg|]}t|r|qSr)r)r1r~)rrrr4sz.)rv)rr)rrlocally_modified_subplot_paramssz(GridSpec.locally_modified_subplot_paramsHzG?c Csdtj|j|d}d|kr"td|dkr4t|}tj||j||||||d}|r`|jf|dS)ay Adjust subplot parameters to give specified padding. Parameters ---------- pad : float Padding between the figure edge and the edges of subplots, as a fraction of the font-size. h_pad, w_pad : float, optional Padding (height/width) between edges of adjacent subplots. Defaults to *pad*. rect : tuple of 4 floats, default: (0, 0, 1, 1), i.e. the whole figure (left, bottom, right, top) rectangle in normalized figure coordinates that the whole subplots area (including labels) will fit into. )Z grid_specNzcThis figure includes Axes that are not compatible with tight_layout, so results might be incorrect.)padh_padw_padrect)rZget_subplotspec_listrzrrhZ get_rendererZget_tight_layout_figurer) rr'rendererrrrrZsubplotspec_listr}rrrrs    zGridSpec.tight_layout) NNNNNNNNN)N)NrNNN) r!rorprqrrvrr(rr __classcell__rr)r rrKGs3  rKcs4eZdZdZd fdd Zd ddZddZZS) GridSpecFromSubplotSpeczx GridSpec whose subplot layout parameters are inherited from the location specified by a given SubplotSpec. Ncs8||_||_||_|jj|_tj||||ddS)ak Parameters ---------- nrows, ncols : int Number of rows and number of columns of the grid. subplot_spec : SubplotSpec Spec from which the layout parameters are inherited. wspace, hspace : float, optional See `GridSpec` for more details. If not specified default values (from the figure or rcParams) are used. height_ratios : array-like of length *nrows*, optional See `GridSpecBase` for details. width_ratios : array-like of length *ncols*, optional See `GridSpecBase` for details. )rrN)_wspace_hspace _subplot_specrJr'rtr)rrrZ subplot_specr;r<rr)r rrrs z GridSpecFromSubplotSpec.__init__c Cs|jdk r|jn|dk r |jjntd}|jdk r8|jn|dk rH|jjntd}|j|}|j\}}}}t j j ||||||dS)z1Return a dictionary of subplot layout parameters.Nzfigure.subplot.hspacezfigure.subplot.wspace)r7r8r9r:r;r<) rrr<rrr;rr|Zextentsr{r'r) rr'r<r;figboxr7r9r8r:rrrr(s  z*GridSpecFromSubplotSpec.get_subplot_paramscCs |jS)zY Return the topmost `.SubplotSpec` instance associated with the subplot. )rrH)rrrrrHsz/GridSpecFromSubplotSpec.get_topmost_subplotspec)NNNN)N)r!rorprqrr(rHrrr)r rrs  rc@seZdZdZd+ddZddZeddZed d Z e j d d Z d d Z ddZ ddZ eddZeddZddZddZddZddZeddd,d!d"Zd#d$Zd%d&Zd'd(Zd)d*ZdS)-rZaN The location of a subplot in a `GridSpec`. .. note:: Likely, you'll never instantiate a `SubplotSpec` yourself. Instead you will typically obtain one from a `GridSpec` using item-access. Parameters ---------- gridspec : `~matplotlib.gridspec.GridSpec` The GridSpec, which the subplot is referencing. num1, num2 : int The subplot will occupy the num1-th cell of the given gridspec. If num2 is provided, the subplot will span between num1-th cell and num2-th cell *inclusive*. The index starts from 0. NcCs||_||_||_dS)N) _gridspecr\r])rgridspecr\r]rrrr/szSubplotSpec.__init__c Cs6|d|jjd|jjd|jjd|jjd S)N[:z, ])rJr*rUrVr+)rrrrr"4szSubplotSpec.__repr__c Csbt|dkrx|\}t|tr |St|ts8td|yttt|\}}}Wqtk rttd|dYqXn,t|dkr|\}}}ntdt|dt |||}|dkrt |||d}t|t rt|dkrt d d |Dstd ||\}}nDt|tr2|dks2|||krJtd ||d ||}}||d|S)z Construct a `.SubplotSpec` from a parent `.Figure` and either - a `.SubplotSpec` -- returned as is; - one or three numbers -- a MATLAB-style subplot specifier. rz>Single argument to subplot must be a three-digit integer, not Nz0subplot() takes 1 or 3 positional arguments but z were given)r'r6css|]}t|tVqdS)N)r r)r1nrrr Ysz1SubplotSpec._from_subplot_args..z3Subplot specifier tuple must contain integers, not znum must be 1 <= num <= z, not ) rr rZrr mapintstr TypeErrorrKrNrYrc) r'argsargrowscolsnumrMijrrr_from_subplot_args9s:          $zSubplotSpec._from_subplot_argscCs|jdkr|jS|jS)N)_num2r\)rrrrr]hszSubplotSpec.num2cCs ||_dS)N)r)rvaluerrrr]lscCs|jS)N)__dict__)rrrr __getstate__pszSubplotSpec.__getstate__cCs|jS)N)r)rrrrrJsszSubplotSpec.get_gridspeccCs |\}}|||j|jfS)a Return the subplot geometry as tuple ``(n_rows, n_cols, start, stop)``. The indices *start* and *stop* define the range of the subplot within the `GridSpec`. *stop* is inclusive (i.e. for a single cell ``start == stop``). )rJr&r\r])rrrrrrr&vszSubplotSpec.get_geometrycCs$|j}t|j||j|dS)z6The rows spanned by this subplot, as a `range` object.r)rJrrlr\r])rrrrrr*s zSubplotSpec.rowspancCs4|j}t|j||j|g\}}t||dS)z9The columns spanned by this subplot, as a `range` object.r)rJrsortedr\r]rl)rrc1c2rrrr+s zSubplotSpec.colspancCs |jjdkS)Nr)r*rU)rrrr is_first_rowszSubplotSpec.is_first_rowcCs|jj|jkS)N)r*rVrJr)rrrr is_last_rowszSubplotSpec.is_last_rowcCs |jjdkS)Nr)r+rU)rrrr is_first_colszSubplotSpec.is_first_colcCs|jj|jkS)N)r+rVrJr)rrrr is_last_colszSubplotSpec.is_last_colz3.4 return_allFcCs|}|\}}t|j|jg||f\}}||\}} } } ||} | |} | |}| |}t || || }|r||d|d||fS|SdS)zJ Update the subplot position from ``figure.subplotpars``. rN) rJr&r>Z unravel_indexr\r]rFminmaxrZ from_extents)rr'rrrrrrrCrBrDrEZ fig_bottomZfig_topZfig_leftZ fig_rightrrrrr|s     zSubplotSpec.get_positioncCs"|}t|dr|S|SdS)zX Return the topmost `SubplotSpec` instance associated with the subplot. rHN)rJrIrH)rrrrrrHs z#SubplotSpec.get_topmost_subplotspeccCs8|j|j|jft|dtt|dtt|dtfkS)z} Two SubplotSpecs are considered equal if they refer to the same position(s) in the same `GridSpec`. rr\r])rr\r]rrk)rotherrrr__eq__s  zSubplotSpec.__eq__cCst|j|j|jfS)N)hashrr\r])rrrr__hash__szSubplotSpec.__hash__cKst|||f|S)a Create a GridSpec within this subplot. The created `.GridSpecFromSubplotSpec` will have this `SubplotSpec` as a parent. Parameters ---------- nrows : int Number of rows in grid. ncols : int Number or columns in grid. Returns ------- `.GridSpecFromSubplotSpec` Other Parameters ---------------- **kwargs All other parameters are passed to `.GridSpecFromSubplotSpec`. See Also -------- matplotlib.pyplot.subplots Examples -------- Adding three subplots in the space occupied by a single subplot:: fig = plt.figure() gs0 = fig.add_gridspec(3, 1) ax1 = fig.add_subplot(gs0[0]) ax2 = fig.add_subplot(gs0[1]) gssub = gs0[2].subgridspec(1, 3) for i in range(3): fig.add_subplot(gssub[0, i]) )r)rrrr}rrr subgridspecs(zSubplotSpec.subgridspec)N)F)r!rorprqrr"rsrrrr]setterrrJr&r*r+rrrrrZdelete_parameterr|rHrrrrrrrrZs*  /       rZ)rqriloggingnumbersrnumpyr>Z matplotlibr{rrrrZmatplotlib.transformsr getLoggerr!_logrrKrrZrrrr s   .7