.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/2d-interp.py"
.. LINE NUMBERS ARE GIVEN BELOW.

.. only:: html

    .. note::
        :class: sphx-glr-download-link-note

        :ref:`Go to the end <sphx_glr_download_auto_examples_2d-interp.py>`
        to download the full example code.

.. rst-class:: sphx-glr-example-title

.. _sphx_glr_auto_examples_2d-interp.py:


===========================
2-Dimensional Interpolation
===========================

.. GENERATED FROM PYTHON SOURCE LINES 6-88



.. rst-class:: sphx-glr-horizontal


    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_001.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_001.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_002.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_002.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_003.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_003.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_004.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_004.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_005.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_005.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_006.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_006.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_2d-interp_007.png
         :alt: 2d interp
         :srcset: /auto_examples/images/sphx_glr_2d-interp_007.png
         :class: sphx-glr-multi-img





.. code-block:: Python
   :lineno-start: 7


    import ndsplines
    import numpy as np
    import matplotlib.pyplot as plt
    from scipy import interpolate
    from scipy.stats import norm
    import itertools


    def gaussian(x_in):
        z = norm.ppf(0.995)
        x = z * (2 * x_in - 1)
        return norm.pdf(x)


    def sin(x_in):
        x = np.pi * (x_in - 0.5)
        return np.sin(x)


    def tanh(x_in):
        x = 2 * np.pi * (x_in - 0.5)
        return np.tanh(x)


    def dist(x_in, y_in):
        return np.sqrt((x_in - 0.25) ** 2 + (y_in - 0.25) ** 2)


    funcs = [gaussian, sin, tanh]


    def wrap2d(funcx, funcy):
        def func2d(x_in, y_in):
            return funcx(x_in) * funcy(y_in)

        func2d.__name__ = "_".join([funcx.__name__, funcy.__name__])
        return func2d


    funcs = [
        wrap2d(*funcs_to_wrap)
        for funcs_to_wrap in itertools.combinations_with_replacement(funcs, r=2)
    ]
    funcs.append(dist)

    x = np.linspace(0, 1, 7)
    y = np.linspace(0, 1, 7)

    xx = np.linspace(0, 1, 64)
    yy = np.linspace(0, 1, 64)

    xx = np.linspace(-0.25, 1.25, 64)
    yy = np.linspace(-0.25, 1.25, 64)
    k = 3


    meshx, meshy = np.meshgrid(x, y, indexing="ij")
    gridxy = np.stack((meshx, meshy), axis=-1)


    meshxx, meshyy = np.meshgrid(xx, yy, indexing="ij")
    gridxxyy = np.stack((meshxx, meshyy), axis=-1)

    for func in funcs:
        fvals = func(meshx, meshy)
        truef = func(meshxx, meshyy)
        test_NDBspline = ndsplines.make_interp_spline(
            gridxy,
            fvals,
        )
        test_RectSpline = interpolate.RectBivariateSpline(x, y, fvals)

        fig = plt.figure()
        ax = fig.add_subplot(111, projection="3d")

        ax.plot_wireframe(meshxx, meshyy, truef, alpha=0.25, color="C0")
        ax.plot_wireframe(meshxx, meshyy, test_NDBspline(gridxxyy), color="C1")
        ax.plot_wireframe(
            meshxx, meshyy, test_RectSpline(meshxx, meshyy, grid=False), color="C2"
        )
        plt.show()


.. rst-class:: sphx-glr-timing

   **Total running time of the script:** (0 minutes 0.542 seconds)


.. _sphx_glr_download_auto_examples_2d-interp.py:

.. only:: html

  .. container:: sphx-glr-footer sphx-glr-footer-example

    .. container:: sphx-glr-download sphx-glr-download-jupyter

      :download:`Download Jupyter notebook: 2d-interp.ipynb <2d-interp.ipynb>`

    .. container:: sphx-glr-download sphx-glr-download-python

      :download:`Download Python source code: 2d-interp.py <2d-interp.py>`

    .. container:: sphx-glr-download sphx-glr-download-zip

      :download:`Download zipped: 2d-interp.zip <2d-interp.zip>`


.. only:: html

 .. rst-class:: sphx-glr-signature

    `Gallery generated by Sphinx-Gallery <https://sphinx-gallery.github.io>`_