在 python 中提取曲线的线性部分

Question

我有两个数组，在绘制了关于 x 轴和 y 轴的折线图之后，我想在 python 中提取图形的线性部分。我怎样才能精确地做到这一点？例如： x 轴 = [1,2,3,4]（以秒为单位的时间）和 y 轴 = [值]。所以，只想计算线性部分的斜率和截距。

我尝试使用 scipy 中的 linregress 函数，但结果不够准确。还有其他我可以使用的功能吗？

Answer 1

polyfit 函数来拟合数据的多项式。您需要指定多项式的次数（在本例中，1 表示线性拟合），以及您的 x 和 y 数据：

slope, intercept = np.polyfit(x_axis, y_axis, 1)