使用Numpy/SciPy进行线条平滑

Question

我有一条线，应该用scipy.interpolate.splrep和scipy.interpolate.splev来平滑。

line = ((x1, y1), (x2, y2), ... (xn, yn))
tck = interpolate.splrep(x, y)

我需要为我的x坐标找到更多的值，这些值应该均匀排列。

newx = numpy.XXX(x)
newy = interpolate.splev(newx, tck)

例如(1, 2, 4, 3) -> (1, 1.5, 2, 2.5, 3, 3.5, 4, 3.5, 3)

在Numpy/Sypy中有没有一种"简单"的方法可以实现这一点？

Answer 1

您可以这样做：

import scipy.interpolate as interp
z = arange(0,4)
x = np.array([1,2,4,3])
f = interp.interp1d(z, x)
newx = f(np.linspace(z[0],z[-1],7))

哪个应该给你

In [40]: print z
[0 1 2 3]
In [41]: print x
[1 2 4 3]
In [42]: print newx
[ 1.   1.5  2.   3.   4.   3.5  3. ]

其将仅按照横坐标点在阵列中定义的顺序在横坐标点之间线性插值。你是这么想的吗？