python 傅里叶曲线拟合

先看一段效果

代码结构

1. 拟合曲线的方程
2. 将原始数据和拟合结果绘制到一张图上,并保存
3. 图片合成视频

import 部分说明

from scipy.optimize import curve_fit
import matplotlib.pyplot as plt
import numpy as np
import sys
import os

拟合方程如下

def fourier(x, *args):
    w = 2 * np.pi / 200
    ret = 0
    for deg in range(0, int(len(args) / 2) + 1):
        ret += args[deg] * np.cos(deg * w * x) + args[len(args) - deg - 1] * np.sin(deg * w * x)

    return ret

绘制并保存

def main(x, y, n