Python SciPy interpolate.NearestNDInterpolator用法及代码示例

本文简要介绍 python 语言中 scipy.interpolate.NearestNDInterpolator 的用法。

用法:

class  scipy.interpolate.NearestNDInterpolator(x, y, rescale=False, tree_options=None)#

NearestNDInterpolator(x, y)。

Nearest-neighbor N > 1 维插值。

参数 ：：

x： (npoints, ndims) 浮点数的二维 ndarray

数据点坐标。

y： (npoints, ) 浮点数或复数的一维 ndarray

数据值。

rescale： 布尔值，可选

在执行插值之前将点重新缩放到单位立方体。如果某些输入维度具有不可比较的单位并且相差许多数量级，这将很有用。

tree_options： 字典，可选

传递给底层 cKDTree 的选项。

注意：

使用scipy.spatial.cKDTree

注意

对于常规网格上的数据，请改用 interpn 。

例子：

我们可以在 2D 平面上插值：

>>> from scipy.interpolate import NearestNDInterpolator
>>> import numpy as np
>>> import matplotlib.pyplot as plt
>>> rng = np.random.default_rng()
>>> x = rng.random(10) - 0.5
>>> y = rng.random(10) - 0.5
>>> z = np.hypot(x, y)
>>> X = np.linspace(min(x), max(x))
>>> Y = np.linspace(min(y), max(y))
>>> X, Y = np.meshgrid(X, Y)  # 2D grid for interpolation
>>> interp = NearestNDInterpolator(list(zip(x, y)), z)
>>> Z = interp(X, Y)
>>> plt.pcolormesh(X, Y, Z, shading='auto')
>>> plt.plot(x, y, "ok", label="input point")
>>> plt.legend()
>>> plt.colorbar()
>>> plt.axis("equal")
>>> plt.show()