我的popt变量输出一条垂直直线(请参见visualization的链接)。我想找到最适合我的数据的正确指数线
[我首先在excel中完成了指数运算,得到了desired visualization,但公式不正确,为712e ^ 0.0001 * x。
我想以类似于Excel中的desired visualization的方式显示popt值,以确保我的popt值在视觉上有意义。
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit
import numpy as np
import pandas as pd
plot = plt.scatter(df.No_of_patients, df.No_of_booked_app)
plt.xlabel("No_of_patients")
plt.ylabel("No_of_booked_app")
plt.xlim(-500, 16000)
plt.ylim(-1000, 7000)
x1 = [1, 2, 3, 4, 5 ,6 ,7 ,8, 9, 10]
y1 = [2, 4, ,8 , 12, 20, 35, 40, 55, 70, 90]
df = pd.DataFrame(zip(x1, y1), columns = ['x1', 'y1'] )
def func(x, a, b, p0=None):
return a*np.exp(b*x)
x = df.x1
y = df.y1
popt, popcov = curve_fit(func, x, y, p0=[1,0], maxfev = 5000)
best_fit = plt.plot(func(x,*popt), 'y')
这是我想要的输出:
您忘记了将x变量传递给plot命令。因此,只需在传递x
并将o
用作市场符号的位置使用以下代码即可
best_fit = plt.plot(x, func(x,*popt), 'yo')