使用Python拟合函数曲线需要用到一些第三方库:
-
numpy:科学计算的基础库(例如:矩阵)
-
matplotlib:绘图库
-
scipy:科学计算库
如果没有安装过这些库,需要在命令行中输入下列代码进行安装:
pip install numpy matplotlib scipy
拟合多项式
'''
Author: CloudSir
Date: 2021-08-01 13:40:50
LastEditTime: 2021-08-02 09:41:54
LastEditors: CloudSir
Description: Python拟合多项式
https://github.com/cloudsir
'''
import matplotlib.pyplot as plt
import numpy as np
x = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
y = [2.83, 9.53, 14.52, 21.57, 38.26, 53.92, 73.15, 101.56, 129.54, 169.75, 207.59]
z1 = np.polyfit(x, y, 3)
p1 = np.poly1d(z1)
y_pre = p1(x)
plt.plot(x,y,'.')
plt.plot(x,y_pre)
plt.show()
函数说明
np.polyfit(x, y, n)
功能:拟合曲线
参数:
返回值:
-
一个列表,拟合出的系数,顺序为从高到底
例: n=3时,会利用
a
x
3
+
b
x
2
+
c
x
+
d
ax^3+bx^2+cx+d
ax3+bx2+cx+d 拟合函数,并返回拟合出的系数 [a, b, c, d]
np.poly1d(li, r=False)
功能:生成多项式函数
参数:
返回值:
拟合任意函数
'''
Author: CloudSir
Date: 2021-08-03 15:01:17
LastEditTime: 2021-08-03 15:26:05
LastEditors: CloudSir
Description: Python拟合任意函数
https://github.com/cloudsir
'''
import numpy as np
import matplotlib.pyplot as plt
from scipy import optimize as op
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False
def f_1(x, A, B, C):
return A * x**2 + B * x + C
x_group = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
y_group = [2.83, 9.53, 14.52, 21.57, 38.26, 53.92, 73.15, 101.56, 129.54, 169.75, 207.59]
A, B, C = op.curve_fit(f_1, x_group, y_group)[0]
plt.scatter(x_group, y_group, marker='o',label='真实值')
x = np.arange(0, 15, 0.01)
y = A * x**2 + B *x + C
plt.plot(x, y,color='red',label='拟合曲线')
plt.legend()
plt.show()
函数说明
op.curve_fit(f, x, y)
功能:拟合任意函数
参数:
-
f:要拟合的函数类型
def f(x, A, B, C):
return A * x**2 + B * x + C
op.curve_fit(f, x, y)
-
x, y:x和y的原始数据
返回值:一个元组 (popt,pcov)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)