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()