【Python】piecewise与curve_fit分段线性函数拟合

最新推荐文章于 2024-09-09 08:18:14 发布

半盒果酱

最新推荐文章于 2024-09-09 08:18:14 发布

阅读量7.9k

点赞数 3

分类专栏： Python

本文链接：https://blog.csdn.net/flora_zhl/article/details/106617525

版权

本文介绍了使用Python进行分段线性函数拟合的方法，特别是通过`scipy.optimize.curve_fit`库进行拟合。在实践中，由于数据特性，可能会遇到用3段函数拟合2段数据导致的OptimizeWarning。通过设置参数限制如`bounds`，可以改善拟合效果，但准确的设置需要根据具体数据调整。

摘要由CSDN通过智能技术生成

思路：确定所有的分割点(x*,y*)以及首尾的斜率(k*)

参考：两段的分段函数；三段的分段函数；scipy.optimize.curve_fit

代码：

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


x = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ,11, 12, 13, 14, 15], dtype=float)
y = np.array([5, 7, 9, 11, 13, 15, 28.92, 42.81, 56.7, 70.59, 
	                 84.47, 98.36, 102.25, 106.14, 110.03])

# 一个输入序列，4个未知参数，2个分段函数
def piecewise_linear(x, x0, y0, k1, k2):
	# x<x0 ⇒ lambda x: k1*x + y0 - k1*x0
	# x>=x0 ⇒ lambda x: k2*x + y0 - k2*x0
    return np.piecewise(x, [x < x0, x >= x0], [lambda x:k1*x + y0-k1*x0,