拟合曲线

import numpy as np
import matplotlib.pyplot as plt

x = [1,2,3,4,5,6,7,8,9,10]  # 如果是df加载的话, 需要按照自变量排序, 同时保证因变量也要相应排序 
y = [4,5,3,6,9,12,15,13,18,20]

plt.scatter(x, y, marker=".")

parameter = np.polyfit(x, y, 4)  # 用4次多项式拟合
p = np.poly1d(parameter)
y2 = np.polyval(p,x )
plot1=plt.plot(x, y, '*',label='original values')
plot2=plt.plot(x, y2, 'r',label='polyfit values', linewidth = 1)
plt.xlabel('ws_mean')
plt.ylabel('pw_mean')
plt.legend(loc=4) # 指定legend的位置,读者可以自己help它的用法
plt.title('polyfitting')
plt.show()