用SciPy插值不均匀数据

Question

我一直在通过documentation for scipy.interpolate上的例子，我无法弄清楚我将如何采集不均匀间隔的数据并插入它，因为所有的教程都使用了linspaces - 它们均匀间隔。

例如，我有一些数据传播，像这样：

[--1--4-----5-3-22---55-] 

其中每个-代表缺失值。

我将如何去拟合插值函数使用scipy.interpolate？

Answer 1

interpolate.interp1d适用于不均匀间隔的数据。例如，

import re 
import numpy as np 
import scipy.interpolate as interpolate 
import matplotlib.pyplot as plt 

text = '--1--4-----5-3-22---55-' 
parts = [c for c in re.split(r'(-|\d+)', text) if c] 
data = np.array([(x, int(y)) for x, y in enumerate(parts) if y != '-']) 
x, y = data.T 
f = interpolate.interp1d(x, y, kind='cubic') 

newx = np.linspace(x.min(), x.max()) 
newy = f(newx) 
plt.plot(newx, newy) 
plt.scatter(x, y, s=20) 
plt.show() 

产生