U L?h@s@ddlmZddlmmmZddgiZGdddZdS)) import_moduleN)PlotGrid matplotlibc@s`eZdZdZdddddZddZed d Zed d Zd dZ ddZ ddZ ddZ dS)ra This class helps to plot subplots from already created SymPy plots in a single figure. Examples ======== .. plot:: :context: close-figs :format: doctest :include-source: True >>> from sympy import symbols >>> from sympy.plotting import plot, plot3d, PlotGrid >>> x, y = symbols('x, y') >>> p1 = plot(x, x**2, x**3, (x, -5, 5)) >>> p2 = plot((x**2, (x, -6, 6)), (x, (x, -5, 5))) >>> p3 = plot(x**3, (x, -5, 5)) >>> p4 = plot3d(x*y, (x, -5, 5), (y, -5, 5)) Plotting vertically in a single line: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(2, 1, p1, p2) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x for x over (-5.0, 5.0) [1]: cartesian line: x**2 for x over (-5.0, 5.0) [2]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plotting horizontally in a single line: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(1, 3, p2, p3, p4) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[2]:Plot object containing: [0]: cartesian surface: x*y for x over (-5.0, 5.0) and y over (-5.0, 5.0) Plotting in a grid form: .. plot:: :context: close-figs :format: doctest :include-source: True >>> PlotGrid(2, 2, p1, p2, p3, p4) PlotGrid object containing: Plot[0]:Plot object containing: [0]: cartesian line: x for x over (-5.0, 5.0) [1]: cartesian line: x**2 for x over (-5.0, 5.0) [2]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[1]:Plot object containing: [0]: cartesian line: x**2 for x over (-6.0, 6.0) [1]: cartesian line: x for x over (-5.0, 5.0) Plot[2]:Plot object containing: [0]: cartesian line: x**3 for x over (-5.0, 5.0) Plot[3]:Plot object containing: [0]: cartesian surface: x*y for x over (-5.0, 5.0) and y over (-5.0, 5.0) TN)showsizecOsptdddddgidtfd|_||_||_g|_d|_||_|D]}|j|jq@||_ |rl|jrl| dS) a Parameters ========== nrows : The number of rows that should be in the grid of the required subplot. ncolumns : The number of columns that should be in the grid of the required subplot. nrows and ncolumns together define the required grid. Arguments ========= A list of predefined plot objects entered in a row-wise sequence i.e. plot objects which are to be in the top row of the required grid are written first, then the second row objects and so on Keyword arguments ================= show : Boolean The default value is set to ``True``. Set show to ``False`` and the function will not display the subplot. The returned instance of the ``PlotGrid`` class can then be used to save or display the plot by calling the ``save()`` and ``show()`` methods respectively. size : (float, float), optional A tuple in the form (width, height) in inches to specify the size of the overall figure. The default value is set to ``None``, meaning the size will be set by the default backend. rfromlistpyplotcm collectionsz1.1.0)Z import_kwargsZmin_module_versionZcatchN) r RuntimeErrorrnrowsncolumns_series_figargsappendrr)selfr r rrrkwargsargrI/var/www/html/venv/lib/python3.8/site-packages/sympy/plotting/plotgrid.py__init__\s#  zPlotGrid.__init__c Cs|jj|j|j}i}d}t|jD]@}t|jD]0}|t|jkr\|j|||||f<|d7}q4q&|jsrind|ji}|jj j f||_ | D]T\}}t|j dkr|j djrddini}|j j|f|} |j |_| |_|qdS)NrZfigsizeZ projectionZ3d)rZgridspecZGridSpecr r rangelenrrrZfigureritemsrZis_3DZ add_subplotZ _plotgrid_figZ _plotgrid_axZprocess_series) rgsmappingcijkwspecpZcur_axrrr_create_figures(  zPlotGrid._create_figurecCs|js||jSN)rr$rrrrfigsz PlotGrid.figcCs|Sr%rr&rrr_backendszPlotGrid._backendcCs|jj|jdSr%)rrcloser'r&rrrr)szPlotGrid.closecCs*tjr|j|jjn|dSr%) base_backendZ_showr'Z tight_layoutrrrr)r&rrrrs z PlotGrid.showcCs|j|dSr%)r'Zsavefig)rpathrrrsavesz PlotGrid.savecCs"ddt|jD}dd|S)NcSs g|]\}}d|t|qS)z Plot[%d]:)str).0rZplotrrr sz$PlotGrid.__str__..zPlotGrid object containing:  ) enumeraterjoin)rZ plot_strsrrr__str__szPlotGrid.__str__) __name__ __module__ __qualname____doc__rr$propertyr'r(r)rr,r3rrrrrsK1  r)Zsympy.externalrZ$sympy.plotting.backends.base_backendZplottingbackendsr*Z__doctest_requires__rrrrrs