python - 插值 - Numba

Question

我正在使用 Scipy 的 interpolate.interp1d 在 Python3 中插入​​一维数组。我想将它与 numba 一起使用，但不支持 scipy 和此功能。是否有 numba 支持的插值函数，或者使用 numba 进行插值的方法？

Answer 1

您可以对一维数组使用 numpy 的插值函数。您有一个 numba 支持的 numpy 函数列表 here :

numpy.interp() (only the 3 first arguments; requires NumPy >= 1.10)

如果你不能让它工作，它可能是 numba 的版本。这是一个使用与 np.interp 中相同示例的工作示例:

import numpy as np
from numba import njit

@njit
def interp_nb(x_vals, x, y):
    return np.interp(xvals, x, y)

x = np.linspace(0, 2*np.pi, 10)
y = np.sin(x)
xvals = np.linspace(0, 2*np.pi, 50)

y_interp = interp_nb(xvals, x, y)

plt.figure(figsize=(10,6))
plt.plot(x, y, 'o')
plt.plot(xvals, y_interp, '-x')

numba.__version__
# '0.43.1'

np.__version__
# '1.18.1'