2

我正在尝试(a*x+b)使用 matplotlib 和 scipy 在具有线性函数的 .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='拟合函数')

文件“C:/Users/Fra/Desktop/lab/ottica/2/reaqd.py”，第 14 行，在 test_func 返回 a*x+b

TypeError：不能将序列乘以“numpy.float64”类型的非整数

4

2 回答 2

5

您可能正在尝试将 Python 列表乘以浮点数，但这是行不通的。尝试a*np.array(x)+b

于 2019-04-18T08:36:28.780 回答
0

当您询问接受的答案如何工作时，显示“老派”方式可能有其用途：

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次的列表，这需要一个整数。

于 2019-04-18T09:58:54.887 回答