先看一段效果
代码结构
- 拟合曲线的方程
- 将原始数据和拟合结果绘制到一张图上,并保存
- 图片合成视频
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