Python data processing 2: using polyfit to achieve curve fitting

The following code is used:

np.poly1d

np.polyfit

np.polyval

import matplotlib.pyplot as plt
import numpy as np

#获取待拟合数据
x = np.linspace(1, 50, 50)
f = np.poly1d([2,5,10])
y = f(x)

#拟合 返回值就是多项式的系数 从左到右对应次数从高到低
#deg指的是多项式的最高次的次数
param = np.polyfit(x, y, deg=2)
print(param)

#利用拟合得到系数计算函数值
#也可以使用以下代码计算函数值：f = np.poly1d(param) y = f(x)
z = np.polyval(param, x)
plt.plot(x, y, marker='o')
plt.plot(x, z)

plt.show()