python怎么编程序求分段函数_在pandas/numpy中实现分段函数的正确方法

我需要创建一个传递给curve_fit的函数。在我的例子中，函数最好定义为分段函数。在

我知道以下命令不起作用，但我展示它是因为它使函数的意图变得清晰：def model_a(X, x1, x2, m1, b1, m2, b2):

'''f(x) has form m1*x + b below x1, m2*x + b2 above x2, and is

a cubic spline between those two points.'''

y1 = m1 * X + b1

y2 = m2 * X + b2

if X <= x1:

return y1 # function is linear below x1

if X >= x2:

return y2 # function is linear above x2

# use a cubic spline to interpolate between lower

# and upper line segment

a, b, c, d = fit_cubic(x1, y1, x2, y2, m1, m2)

return cubic(X, a, b, c, d)

当然，问题是X是一个pandas系列，而形式(X <= x1)的计算结果是一系列布尔值，因此这将失败，并显示消息“序列的真值是模棱两可的”

似乎^{}正是为这种情况而设计的：“只要condlist[i]为真，funclist[i]（x）就被用作输出值。”所以我尝试了以下方法：

^{pr2}$

但这次失败了：return np.piecewise(X, [X<=x1, X>=x2], [lo, hi, mid])

“索引器错误：数组的索引太多”。我倾向于认为它反对这样一个事实，即在condlist中有两个元素，在funclist中有三个元素，但是文档明确指出funclist中的额外元素被视为默认值。在

有什么指引吗？在