.. DO NOT EDIT. .. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY. .. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE: .. "examples_gallery/advanced_pdes/solver_comparison.py" .. LINE NUMBERS ARE GIVEN BELOW. .. only:: html .. note:: :class: sphx-glr-download-link-note :ref:`Go to the end ` to download the full example code. .. rst-class:: sphx-glr-example-title .. _sphx_glr_examples_gallery_advanced_pdes_solver_comparison.py: Solver comparison ================= This example shows how to set up solvers explicitly and how to extract diagnostic information. .. GENERATED FROM PYTHON SOURCE LINES 8-44 .. image-sg:: /examples_gallery/advanced_pdes/images/sphx_glr_solver_comparison_001.png :alt: Deviation: 8.6e-05, 0.00018, 9.1e-05, 9.1e-05, 9.3e-05, explicit Euler solver, explicit, adaptive Runge-Kutta solver, implicit solver, Crank-Nicolson solver, Adam-Bashforth solver, scipy solver :srcset: /examples_gallery/advanced_pdes/images/sphx_glr_solver_comparison_001.png :class: sphx-glr-single-img .. rst-class:: sphx-glr-script-out .. code-block:: none Diagnostic information for explicit Euler solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 0.0015272600000031389, 'tracker': 5.125999999222586e-05, 'compilation': 5.394170932999998}, 'solver_start': '2026-04-18 21:38:40.764524+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:00.001568', 't_final': 1.0}, 'package_version': 'unknown', 'solver': {'class': 'EulerSolver', 'pde_class': 'DiffusionPDE', 'dt': 0.001, 'steps': 1000, 'dt_adaptive': False, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}, 'post_step_data': None}} Diagnostic information for explicit, adaptive Runge-Kutta solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 0.0003624400000035166, 'tracker': 4.047000000184653e-05, 'compilation': 6.578913187999994}, 'solver_start': '2026-04-18 21:38:47.352849+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:00.000395', 't_final': 1.0}, 'package_version': 'unknown', 'solver': {'class': 'RungeKuttaSolver', 'pde_class': 'DiffusionPDE', 'dt': 0.1974728536669598, 'steps': 12, 'dt_adaptive': True, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}, 'dt_statistics': {'min': 0.001, 'max': 0.16357124408714668, 'mean': 0.08333333333333334, 'std': 0.05144090326391255, 'count': 12.0}, 'post_step_data': None}} Diagnostic information for implicit solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 0.0077822269999927585, 'tracker': 4.5080000020902844e-05, 'compilation': 3.6097516719999874}, 'solver_start': '2026-04-18 21:38:50.974394+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:00.007819', 't_final': 1.0}, 'package_version': 'unknown', 'solver': {'class': 'ImplicitSolver', 'pde_class': 'DiffusionPDE', 'dt': 0.001, 'steps': 1000, 'dt_adaptive': False, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}, 'function_evaluations': 0, 'post_step_data': None}} Diagnostic information for Crank-Nicolson solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 0.011481795999998212, 'tracker': 4.4020000004252324e-05, 'compilation': 5.542617839000002}, 'solver_start': '2026-04-18 21:38:56.532661+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:00.011518', 't_final': 1.0}, 'package_version': 'unknown', 'solver': {'class': 'CrankNicolsonSolver', 'pde_class': 'DiffusionPDE', 'dt': 0.001, 'steps': 1000, 'dt_adaptive': False, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}, 'function_evaluations': 0, 'post_step_data': None}} Diagnostic information for Adam-Bashforth solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 3.3912777449999965, 'tracker': 4.552000001467604e-05, 'compilation': 1.7138511209999905}, 'solver_start': '2026-04-18 21:38:58.268223+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:03.393251', 't_final': 1.0}, 'package_version': 'unknown', 'solver': {'class': 'AdamsBashforthSolver', 'pde_class': 'DiffusionPDE', 'dt': 0.001, 'steps': 1000, 'dt_adaptive': False, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}, 'post_step_data': None}} Diagnostic information for scipy solver: {'controller': {'mpi_run': False, 't_start': 0, 't_end': 1.0, 'profiler': {'solver': 0.6177487319999955, 'tracker': 3.1019999994441605e-05, 'compilation': 0.0009604400000000624}, 'solver_start': '2026-04-18 21:39:01.665085+00:00', 'successful': True, 'stop_reason': 'Reached final time', 'solver_duration': '0:00:00.617867', 't_final': np.float64(1.0)}, 'package_version': 'unknown', 'solver': {'class': 'ScipySolver', 'pde_class': 'DiffusionPDE', 'dt': 0.001, 'steps': 61, 'stochastic': False, 'backend': {'name': 'numba', 'implementation': 'numba'}}} | .. code-block:: Python import pde # initialize the grid, an initial condition, and the PDE grid = pde.UnitGrid([32, 32]) field = pde.ScalarField.random_uniform(grid, -1, 1) eq = pde.DiffusionPDE() def run_solver(solver, label): """Helper function testing the solver.""" controller = pde.Controller(solver, t_range=1, tracker=None) sol = controller.run(field, dt=1e-3) sol.label = label + " solver" print(f"Diagnostic information for {sol.label}:") print(controller.diagnostics) print() return sol # try different solvers solutions = [ run_solver(pde.EulerSolver(eq), "explicit Euler"), run_solver( pde.RungeKuttaSolver(eq, adaptive=True), "explicit, adaptive Runge-Kutta" ), run_solver(pde.ImplicitSolver(eq), "implicit"), run_solver(pde.CrankNicolsonSolver(eq), "Crank-Nicolson"), run_solver(pde.AdamsBashforthSolver(eq), "Adam-Bashforth"), run_solver(pde.ScipySolver(eq), "scipy"), ] # plot both fields and give the deviation as the title deviations = [(solutions[0] - sol).fluctuations for sol in solutions] title = "Deviation: " + ", ".join(f"{deviation:.2g}" for deviation in deviations[1:]) pde.FieldCollection(solutions).plot(title=title, arrangement=(2, 3)) .. rst-class:: sphx-glr-timing **Total running time of the script:** (0 minutes 27.376 seconds) .. _sphx_glr_download_examples_gallery_advanced_pdes_solver_comparison.py: .. only:: html .. container:: sphx-glr-footer sphx-glr-footer-example .. container:: sphx-glr-download sphx-glr-download-jupyter :download:`Download Jupyter notebook: solver_comparison.ipynb ` .. container:: sphx-glr-download sphx-glr-download-python :download:`Download Python source code: solver_comparison.py ` .. container:: sphx-glr-download sphx-glr-download-zip :download:`Download zipped: solver_comparison.zip `