gNddlmZddlmcmcmZddgiZGddZdS)) import_moduleN)PlotGrid matplotlibcjeZdZdZddddZdZedZedZd Z d Z d Z d Z 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)showsizec(tddgdidtf|_||_||_g|_d|_||_|D]!}|j|j"||_ |r|jr| dSdSdS)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. rfromlist)pyplotcm collectionsz1.1.0) import_kwargsmin_module_versioncatchN) r RuntimeErrorrnrowsncolumns_series_figargsappendrr)selfrrrrrkwargsargs S/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/plotting/plotgrid.py__init__zPlotGrid.__init__\sF( %'F'F'FG&|o???       - -C L   , , , ,  DO  IIKKKKK    c|jj|j|j}i}d}t |jD]N}t |jD]7}|t |jkr|j|||||f<|dz }8O|jsind|ji}|jj j di||_ | D]o\}}t |j dkr|j djrddini}|j j|fi|} |j |_| |_|pdS)Nrfigsize projection3d)rgridspecGridSpecrrrangelenrrr figureritemsris_3D add_subplot _plotgrid_fig _plotgrid_axprocess_series) rgsmappingcijkwspecpcur_axs r_create_figurezPlotGrid._create_figuresk _ % . .tz4= I I tz""  A4=))  s49~~%%(, ! GBq!tH%Q  y >A+=+= ! ",><&&)+ *TY*466266F"iAO#AN         rcF|js||jSN)rr8rs rfigz PlotGrid.figs&y "    ! ! !yrc|Sr:r#r;s r_backendzPlotGrid._backends rcN|jj|jdSr:)rr closer<r;s rr@zPlotGrid.closes# $$TX.....rctjr9|j|jjdS|dSr:) base_backend_showr< tight_layoutrr rr@r;s rrz PlotGrid.showsO    H ! ! # # # O " ' ' ) ) ) ) ) JJLLLLLrc:|j|dSr:)r<savefig)rpaths rsavez PlotGrid.saves rcndt|jD}dd|zS)Nc>g|]\}}d|zt|zS)z Plot[%d]:)str).0r2plots r z$PlotGrid.__str__..s;;;;!!T"AoT2;;;rzPlotGrid object containing:  ) enumeraterjoin)r plot_strss r__str__zPlotGrid.__str__sC;;%.ty%9%9;;; /91E1EEEr) __name__ __module__ __qualname____doc__rr8propertyr<r>r@rrHrSr#rrrrsJJV59t/////b(X X///FFFFFrr)sympy.externalr$sympy.plotting.backends.base_backendplottingbackendsrB__doctest_requires__rr#rrr^s((((((;;;;;;;;;;;;L> lFlFlFlFlFlFlFlFlFlFr