.. DO NOT EDIT.
.. THIS FILE WAS AUTOMATICALLY GENERATED BY SPHINX-GALLERY.
.. TO MAKE CHANGES, EDIT THE SOURCE PYTHON FILE:
.. "auto_examples/tidy.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_tidy.py>`
        to download the full example code.

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

.. _sphx_glr_auto_examples_tidy.py:


=========================================
2-Dimensional Interpolation of Tidy  Data
=========================================

.. GENERATED FROM PYTHON SOURCE LINES 6-112



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


    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_001.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_001.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_002.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_002.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_003.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_003.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_004.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_004.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_005.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_005.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_006.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_006.png
         :class: sphx-glr-multi-img

    *

      .. image-sg:: /auto_examples/images/sphx_glr_tidy_007.png
         :alt: tidy
         :srcset: /auto_examples/images/sphx_glr_tidy_007.png
         :class: sphx-glr-multi-img


.. rst-class:: sphx-glr-script-out

 .. code-block:: none

    True
    True
    True
    True
    True
    True
    True






|

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


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

    import pandas as pd

    import itertools

    import ndsplines


    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.r_["0,3", meshx, meshy]
    gridxy = np.stack((meshx, meshy), axis=-1)
    tidyxy = gridxy.reshape((-1, 2))


    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)

        tidy_array = np.concatenate(
            (
                fvals.reshape((-1, 1)),
                tidyxy,
            ),
            axis=1,
        )

        tidy_df = pd.DataFrame(
            tidy_array,
            columns=[
                "z",
                "x",
                "y",
            ],
        )
        test_NDBspline3 = ndsplines.make_interp_spline(gridxy, fvals[:, :, None])
        test_NDBspline = ndsplines.make_interp_spline_from_tidy(tidy_df, ["x", "y"], ["z"])
        test_RectSpline = interpolate.RectBivariateSpline(x, y, fvals)
        test_NDBspline2 = ndsplines.make_interp_spline_from_tidy(tidy_array, [1, 2], [0])

        print(np.allclose(test_NDBspline2(gridxxyy), test_NDBspline(gridxxyy)))

        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)[..., 0], 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.738 seconds)


.. _sphx_glr_download_auto_examples_tidy.py:

.. only:: html

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

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

      :download:`Download Jupyter notebook: tidy.ipynb <tidy.ipynb>`

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

      :download:`Download Python source code: tidy.py <tidy.py>`

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

      :download:`Download zipped: tidy.zip <tidy.zip>`


.. only:: html

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

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