numpy - 受约束的 np.polyfit

Question

我正在尝试将二次方程拟合到一些实验数据并在 numpy 中使用 polyfit。我希望得到一条凹曲线，因此要确保二次项的系数为负，并且拟合本身也是加权的，因为点上有一些权重。有没有简单的方法来做到这一点？谢谢。

Answer 1

描述了权重的使用 here (numpy.polyfit) .
基本上，您需要一个与 x 和 y 长度相同的权重向量。

为了避免系数中的错误符号，您可以使用拟合函数定义，如

def fitfunc(x,a,b,c):
    return -1 * abs(a) * x**2 + b * x + c 

这将始终为您提供 x**2 的负系数。