我正在尝试使用 matplotlib 和 scipy (a*x+b)
线性函数在 .txt 文件中拟合和绘制一个简单的数据。我遇到了有关测试功能的错误:"can't multiply sequence by non-int of type 'numpy.float64'"
我尝试更改变量名称x
但我遇到了同样的问题。大多数代码来自一个工作代码,该代码能够毫无问题地对数据进行最佳拟合,并对测试函数使用相同的定义。
import matplotlib.pyplot as plt
from scipy import optimize
import numpy as np
f=open("testData.txt","r")
x_data=[]
y_data=[]
trash=f.readline() #discards first line
for line in f: #reads x and y data from file
x_read,y_read=line.split()
x_data.append(float(x_read))
y_data.append(float(y_read))
def test_func(x, a, b):
return a*x+b
params, params_covariance = optimize.curve_fit(test_func, x_data, y_data,
p0=[1, 1])
plt.figure(figsize=(6, 4))
plt.scatter(x_data, y_data)
plt.plot(x_data, test_func(x_data, params[0], params[1]), label='Fitted
function')
plt.show()
这是错误:
回溯(最近一次调用(:
文件 "C:/Users/Fra/Desktop/lab/ottica/2/reaqd.py",第 19 行,在 plt.plot(x_data, test_func(x_data, params[0], params[1](, label='Fit function'(
文件"C:/Users/Fra/Desktop/lab/ottica/2/reaqd.py",第 14 行,test_func 返回 a*x+b
类型错误:无法将序列乘以类型为"numpy.float64"的非整数
您可能正在尝试将 Python 列表乘以浮点数,但这不起作用。尝试a*np.array(x)+b
<</p>
当您询问接受的答案如何工作时,显示"老派"方式可能会有其用途:
plt.plot(x_data,
[test_func(x, params[0], params[1]) for x in x_data],
label='Fitted function')
x_data = []
显然是一个列表,由于Python不是Matlab,number*x_data
不是逐元素乘法,而是创建一个重复x_data
,number
次的列表,这需要一个整数。