可视化我的popt输出会产生一条直线,我想可视化popt输出以确保其正确。我是初学者

问题描述 投票:0回答:1

我的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')

current_output

这是我想要的输出:

desired_output

matplotlib scipy curve-fitting exponential scipy-optimize
1个回答
1
投票

您忘记了将x变量传递给plot命令。因此,只需在传递x并将o用作市场符号的位置使用以下代码即可

best_fit = plt.plot(x, func(x,*popt), 'yo')
© www.soinside.com 2019 - 2024. All rights reserved.