傅里叶级数在Python中的拟合

Fourier Series Fit in Python(傅里叶级数在Python中的拟合)

本文介绍了傅里叶级数在Python中的拟合的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我有一些数据要使用二次、三次或四次傅立叶级数进行拟合。 虽然this关于堆栈溢出的问题和答案与我希望使用Scipy完成的任务很接近,但他们已经将其系数预定义为tau=0.045。我希望我的Fit能够找到具有95%可信区间的可能系数(a0、w1、w2、w3等),就像傅立叶级数的MATLAB curve fit等效项一样。我看到的另一个选项是使用fourier_series from sympy,但是这个函数只适用于与已定义函数匹配的符号参数,而不是原始数据。

1)是否有办法使渐近傅里叶级数接受原始数据,而不是使用此库的函数或其他变通方法?

2)或对给定多个未知数(系数)的数据进行Scipy曲线拟合

推荐答案

如果您愿意,您可以非常接近sympy代码以进行数据拟合,使用我为此编写的名为symfit的包。它基本上使用sympy接口包装scipy。使用symfit,您可以执行类似以下操作:

from symfit import parameters, variables, sin, cos, Fit
import numpy as np
import matplotlib.pyplot as plt

def fourier_series(x, f, n=0):
    """
    Returns a symbolic fourier series of order `n`.

    :param n: Order of the fourier series.
    :param x: Independent variable
    :param f: Frequency of the fourier series
    """
    # Make the parameter objects for all the terms
    a0, *cos_a = parameters(','.join(['a{}'.format(i) for i in range(0, n + 1)]))
    sin_b = parameters(','.join(['b{}'.format(i) for i in range(1, n + 1)]))
    # Construct the series
    series = a0 + sum(ai * cos(i * f * x) + bi * sin(i * f * x)
                     for i, (ai, bi) in enumerate(zip(cos_a, sin_b), start=1))
    return series

x, y = variables('x, y')
w, = parameters('w')
model_dict = {y: fourier_series(x, f=w, n=3)}
print(model_dict)

这将打印我们需要的符号模型:

{y: a0 + a1*cos(w*x) + a2*cos(2*w*x) + a3*cos(3*w*x) + b1*sin(w*x) + b2*sin(2*w*x) + b3*sin(3*w*x)}

接下来,我将用一个简单的步长函数来拟合这个函数,向您展示它是如何工作的:

# Make step function data
xdata = np.linspace(-np.pi, np.pi)
ydata = np.zeros_like(xdata)
ydata[xdata > 0] = 1
# Define a Fit object for this model and data
fit = Fit(model_dict, x=xdata, y=ydata)
fit_result = fit.execute()
print(fit_result)

# Plot the result
plt.plot(xdata, ydata)
plt.plot(xdata, fit.model(x=xdata, **fit_result.params).y, color='green', ls=':')

这将打印:

Parameter Value        Standard Deviation
a0        5.000000e-01 2.075395e-02
a1        -4.903805e-12 3.277426e-02
a2        5.325068e-12 3.197889e-02
a3        -4.857033e-12 3.080979e-02
b1        6.267589e-01 2.546980e-02
b2        1.986491e-02 2.637273e-02
b3        1.846406e-01 2.725019e-02
w         8.671471e-01 3.132108e-02
Fitting status message: Optimization terminated successfully.
Number of iterations:   44
Regression Coefficient: 0.9401712713086535

并生成以下图表:

就这么简单!我把剩下的留给你的想象。有关更多信息,请访问documentation here。

这篇关于傅里叶级数在Python中的拟合的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:傅里叶级数在Python中的拟合

基础教程推荐