曲线与分段函数拟合？

正如标题所示，我正在尝试用分段函数进行曲线拟合。这是一个例子：

import numpy as np
import scipy.optimize as sio

x = np.array([0, 1, 2, 3, 4, 5])
y = np.array([2, 3, 4, 11, 18, 27])

def f(x, a):
    if x <= 2:
        return x + a
    elif x >= 3:
        return x**2 + a

sio.curve_fit(f, x, y)

现在，这会返回以下错误：

ValueError: The truth value of an array with more than one element is ambiguous

通过添加打印函数，我发现这是因为传递给函数“f”的“x”值是numpy数组[0, 1, 2, 3, 4, 5]

我想知道这个问题是否有解决方法。

Thanks!

总体问题是你已经写了f至于标量，但是x and a是 numpy 数组。您的情况都是如此if线，还有return内的if声明。

与您的代码最相似的解决方案是：

import numpy as np
import scipy.optimize as sio

x = np.array([0, 1, 2, 3, 4, 5])
y = np.array([2, 3, 4, 11, 18, 27])

def f(x, a):
    y = np.zeros(len(x))
    for i, xi in enumerate(x): 
        if xi <= 2:
            y[i] =  xi + a
        elif xi >= 3:
            y[i] =  xi**2 + a
    return y

sio.curve_fit(f, x, y)

更向量化的定义f将会：

def f(x, a):
    y = np.zeros(len(x))
    y[:3] += a
    y[3:] = x[3:]**2 + a
    return y