使用指数函数拟合python图

Question

我正在尝试使用指数函数来拟合我的python图。我在这里附上了代码。任何投入都受到高度赞赏。

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

x=[0.21, 0.43, 0.50, 0.65, 0.86, 1.0, 1.5, 2.0, 2.5, 3.0, 4.0]
y=[43, 33, 30, 24, 18, 16, 14, 13, 14, 13, 13]
yerr= [2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2]
xerr=[0.01,0.01,0.01,0.01,0.01,0.01,0.01,0.01, 0.01,0.01,0.01]
plt.ylim (0,60)
plt.xlabel ('Thickness (d) [mm]')
plt.ylabel ('Counts')
def exponenial_func(x, a, b, c):
    return a*np.exp(-b*x)+c
popt, pcov = curve_fit(exponenial_func, x, y, p0=(1, 1e-6, 1))
yy = exponenial_func(x, *popt)

plt.plot(x,y,'o', x, yy)
#plt.errorbar(x,y,yerr=yerr,xerr=xerr, fmt= '.')
plt.show()

我总是收到错误信息：

Traceback (most recent call last):
  File "/home/manisha/Desktop/Moessbauer effect/countsvsd.py", line 16, in <module>
    yy = exponenial_func(x, *popt)
  File "/home/manisha/Desktop/Moessbauer effect/countsvsd.py", line 14, in exponenial_func
    return a*np.exp(-b*x)+c
TypeError: 'numpy.float64' object cannot be interpreted as an index

由于我是一名新程序员，我不知道这意味着什么。 请帮忙。

Answer 1

您的问题在于您尝试定义yy的方式;您无法在列表 x上调用您的函数。相反，在x中的每个单独项目上调用它，例如，在这样的列表迭代中：

yy = [exponenial_func(i, *popt) for i in x]

然后，代码中的其他所有工作正常：

[编辑] ：要推断指数函数中的零（根据你的评论），你可以这样做：

xx = x.copy()    
yy = [exponenial_func(i, *popt) for i in [0]+xx]
plt.plot(x,y,'o',[0]+xx,yy)

或者，要看到该值，请执行以下操作：

exponenial_func(0, *popt)

Answer 2

虽然@sacul的答案有效，但它还没有告诉你发生了什么。

如果您有list，则可以通过乘法将此list的副本附加到自身。

my_list = [1, 2]
print(my_list * 2) #[1, 2, 1, 2]

因此，当您尝试将list与float相乘时，将不确定复制的行为应该是什么，并抛出Exception。

print(my_list * 1.5) #TypeError: can't multiply sequence by non-int of type 'float'

尝试使用浮点对象，将在评估时间之前捕获并引发以下错误：

a = 1.5
print(my_list*a) #TypeError: 'numpy.float64' object cannot be interpreted as an integer

因此，您的问题有多种解决方案。您可以逐个将list中的各个项目提供给函数，就像@sacul提供的答案一样，或者将容器转换为会产生类似行为的类型：

def exponenial_func(x, a, b, c):
    if isinstance(x,list):
        x = np.array(x)
    return a*np.exp(-b*x)+c