Python对自定义离散点进行指定多项式函数拟合

Python对自定义离散点进行指定多项式函数拟合_beyond谚语的博客-CSDN博客_python拟合离散数据

自定义离散点进行指定多项式函数拟合
用户自己自己输入坐标点，拟合函数可根据用户输入的多项式的最高次方进行自动拟合函数，拟合方法采用最小二乘法进行函数拟合。
(1,2),(2,5),(3,10),(4,17),(5,26),(6,37)(7,50),(8,65),(9,82)
很显然是函数为二次函数，y=x^2+1
代码中，f1 = np.polyfit(x, y, 2)，2表示使用二次函数进行拟合

import numpy as np
import matplotlib.pyplot as plt

#定义x、y散点坐标
x = [1,2,3,4,5,6,7,8,9]
x = np.array(x)
print('x is :\n',x)

num = [2,5,10,17,26,37,50,65,82]

y = np.array(num)
print('y is :\n',y)

#用多次多项式拟合
f1 = np.polyfit(x, y, 2)
print('f1 is :\n',f1)
p1 = np.poly1d(f1)
print('p1 is :\n',p1)

#也可使用yvals=np.polyval(f1, x)
yvals = p1(x) #拟合y值
print('yvals is :\n',yvals)

#绘图
plot1 = plt.plot(x, y, 's',label='original values')
plot2 = plt.plot(x, yvals, 'r',label='polyfit values')
plt.xlabel('grade')
plt.ylabel('value')
plt.legend(loc=4) #指定标识的位置(1为右上角，2为左上角，3为左下角，4为右下角)
plt.title('Color fastness grade')
plt.show()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
运行效果如下：

————————————————
版权声明：本文为CSDN博主「beyond谚语」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_41264055/article/details/123449812