该函数scipy.fftpack.diff
计算导数,但它假设输入是周期性的。参数给出输入序列的period
周期(即x
间隔的总长度)。
就您而言,这len(x)*dx
就是dx = x[1] - x[0]
.
这是一些绘制简单(居中)有限差分(蓝色)和diff
使用period
参数的结果(红色)的代码。变量x
和y
与您的代码中使用的变量相同:
In [115]: plt.plot(0.5*(x[1:]+x[:-1]), np.diff(y)/np.diff(x), 'b')
Out[115]: [<matplotlib.lines.Line2D at 0x1188d01d0>]
In [116]: plt.plot(x, sp.diff(y, period=len(x)*(x[1]-x[0])), 'r')
Out[116]: [<matplotlib.lines.Line2D at 0x1188fc9d0>]
In [117]: plt.xlabel('x')
Out[117]: <matplotlib.text.Text at 0x1157425d0>

请注意,如果您的输入实际上不是周期性的,则计算的导数diff
将在区间末端附近不准确。
下面是另一个示例,使用较短的序列,该序列仅包含区间 [0, 1] 中正弦函数的一个完整周期:
In [149]: x = np.linspace(0, 1, 20, endpoint=False)
In [150]: y = np.sin(2*np.pi*x)
In [151]: plt.plot(0.5*(x[1:]+x[:-1]), np.diff(y)/np.diff(x), 'b')
Out[151]: [<matplotlib.lines.Line2D at 0x119872d90>]
In [152]: plt.plot(x, sp.diff(y, period=len(x)*(x[1]-x[0])), 'r')
Out[152]: [<matplotlib.lines.Line2D at 0x119c49090>]
In [153]: plt.xlabel('x')
Out[153]: <matplotlib.text.Text at 0x1197823d0>
