多项式拟合

import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
import pandas as pd
import numpy as np

df=pd.read_excel('D:/CPO/BEV残值逻辑修改/TP超参数.xlsx')

x=df.TP_2.values
y=df.CS_2.values
z1 = np.polyfit(x, y, 7)
p1 = np.poly1d(z1)
list(p1)

# 画趋势图
yvals=p1(x)#也可以使用yvals=np.polyval(z1,x)
plot1=plt.plot(x, y, '*',label='original values')
plot2=plt.plot(x, yvals, 'r',label='polyfit values')
plt.xlabel('x axis')
plt.ylabel('y axis')
# plt.legend(loc=4)#指定legend的位置,读者可以自己help它的用法
plt.title('polyfitting')
plt.show()