Python curve_fit "object too deep for desired array"

Question

我想在 python 中绘制此函数 ( y = a * ln(x) + b )。这是我的代码：

def func(x, a, b):
return a * np.log(x) + b

popt, _ = curve_fit(func, x, y)
a, b = popt
x_line = arrange(min(x), max(x), 1)
y_line = func(x_line, a, b)
plt.plot(x_line, y_line)
plt.show()

我的“x”包含这个

array([[1790],
       [1800],
       [1810],
       [1820],
       [1830],
       [1840],
       [1850],
       [1860],
       [1870],
       [1880],
       [1900],
       [1910],
       [1920],
       [1930],
       [1940],
       [1950],
       [1960],
       [1970],
       [1980],
       [1990],
       [2000],
       [2010]], dtype=int64)

还有我的“y”这个

array([[  3.929214],
       [  5.308483],
       [  7.239881],
       [  9.638453],
       [ 12.86602 ],
       [ 17.069453],
       [ 23.191876],
       [ 31.443321],
       [ 39.818449],
       [ 50.189209],
       [ 76.212168],
       [ 92.228496],
       [106.021537],
       [123.202624],
       [132.164569],
       [151.325798],
       [179.323175],
       [203.302031],
       [226.542199],
       [248.718302],
       [281.424603],
       [308.745538]])

但是当我运行代码时，我总是得到这个错误：

object too deep for desired array

我希望有人能帮助我，因为我在这上面花了很多时间。

Answer 1

尝试重塑数组：

popt, _ = curve_fit(func, x.reshape(-1), y.reshape(-1))

Answer 2

您的 x 和 y 变量是二维 (22 x 1) 数组，因为当 scipy.optimize.curve_fit 需要一维数组时，内部有一组方括号。

您可以删除内部括号或切片 x 和 y:

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

def func(x, a, b):
  return a * np.log(x) + b

popt, _ = curve_fit(func, x[:,0], y[:,0])
a, b = popt
x_line = np.arange(min(x), max(x), 1)
y_line = func(x_line, a, b)
plt.plot(x_line, y_line)
plt.show()

Python curve_fit "object too deep for desired array"

Python curve_fit "object too deep for desired array"

python

plot

jupyter

curve-fitting