使用Python最小二乘法拟合曲线的代码实现

一、背景描述

在一个普通的摸鱼早晨,群里居然出现了一个不合时宜颇为突兀的正经问题,原来是一个博士同学需要対实验数据进行曲线拟合并且批量计算出多项式方程
一般来说,这种问题对于经常做实验的同学来说并不陌生,通常使用MATLAB或者Origin Pro这类专业的数据计算软件,甚至Excel也可以实现.
但是作为程序员肯定第一想到的还是使用强大的Python来实现,但是因为本人主要做java开发,python知之甚少,但也知道python强大在各种现成的类库,所以实现估计并不难,按我们就上手试试?

再来一张图更直观的表述下,博士同学就是有多组实验数据,可以分布在坐标系,例如图中的各个散点,而他想做的就是将散点拟合为一条曲线,获得曲线的方程式,从而可以通过输入其他x坐标的值预测结果数据.这在非常多的地方都有极大的用处

二、前期准备

在开始前还是需要対一些基本概念和类库进行了解和介绍,因为并不是专业做研究的,所以对于以下名词只做简单的描述,达到理解并且可以使用的目的即可

最小二乘法

最小二乘法（又称最小平方法）是一种数学优化技术。它通过最小化误差的平方和寻找数据的最佳函数匹配。利用最小二乘法可以简便地求得未知的数据，并使得这些求得的数据与实际数据之间误差的平方和为最小
最小二乘法是解决曲线拟合问题最常用的方法。
其基本思路是:事先选定的一组线性无关的函数， 是待定系数 ，拟合准则是使实验数据(笛卡尔坐标系的坐标数据)与待拟合曲线的距离的平方和最小，称为最小二乘准则.
总结来说,我们通常就是使用最小二乘法来做曲线拟合,通过计算最小值来获得多项式的系数与误差数值.
(关于最小二乘法的原理大家可以自行了解,甚至可以让你重新感知到数学逻辑之美)

Python类库

NumPy

NumPy 是一个 Python 包。 它代表 “Numeric Python”。 它是一个由多维数组对象和用于处理数组的例程集合组成的库。支持大量的维度数组与矩阵运算，此外也针对数组运算提供大量的数学函数库。

Matplotlib

Matplotlib 是 Python 的绘图库。 它可与 NumPy 一起使用，提供了一种有效的 MatLab 开源替代方案。

三、代码实现

这里我们举一个例子来展示下,相信下面这道