.. 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 ` 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-73 .. 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: 6 import ndsplines import numpy as np import matplotlib.pyplot as plt from scipy import interpolate from scipy.stats import norm import itertools from mpl_toolkits.mplot3d import Axes3D def gaussian(x_in): z = norm.ppf(.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(-.25, 1.25, 64) yy = np.linspace(-.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.689 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 `_