如何拟合曲线

知识问答 2026-11-12 22:53:57 来源:互联网

拟合曲线是指通过数学方法找到一组数据点的出色的/卓越的/优异的/杰出的函数模型,使其尽可能地描述这些数据点,有多种拟合曲线的方法,如线性回归、多项式回归、指数平滑法等,在Python中,可以使用numpy和matplotlib库进行曲线拟合,首先需要安装这两个库,然后根据所选的拟合方法,使用相应的函数进行拟合,并绘制拟合曲线。

使用numpy的polyfit函数进行多项式回归拟合:

import numpy as npimport matplotlib.pyplot as plt已知数据点x = np.array([0, 1, 2, 3, 4, 5])y = np.array([0, 0.8, 0.9, 0.1, -0.8, -1])选择多项式的阶数,如2阶多项式degree = 2使用numpy的polyfit函数进行拟合coefficients = np.polyfit(x, y, degree)创建多项式函数polynomial = np.poly1d(coefficients)生成拟合曲线的数据点x_fit = np.linspace(0, 5, 100)y_fit = polynomial(x_fit)绘制原始数据点和拟合曲线plt.plot(x, y, 'ro', label='Original Data')plt.plot(x_fit, y_fit, 'b-', label='Fitted Curve')plt.legend()plt.show()

这段代码将使用2阶多项式回归对给定的数据点进行拟合,并绘制原始数据点和拟合曲线。