/Python/Piecewise linear regression python

Piecewise linear regression python

from scipy import optimize

def piecewise_linear(x, x0, x1, b, k1, k2, k3):
    condlist = [x < x0, (x >= x0) & (x < x1), x >= x1]
    funclist = [lambda x: k1*x + b, lambda x: k1*x + b + k2*(x-x0), lambda x: k1*x + b + k2*(x-x0) + k3*(x - x1)]
    return np.piecewise(x, condlist, funclist)

p , e = optimize.curve_fit(piecewise_linear, x, y)

xd = np.linspace(-30, 30, 1000)
plt.plot(x, y, "o")
plt.plot(xd, piecewise_linear(xd, *p))

Source: songhuiming.github.io

Similar codes

5 votes348 views3 answers
Os system python
Python
0 vote371 views1 answer
Flask new response style with `make_response`
Python
0 vote346 views1 answer
List lambda functions boto3
Python
21 votes335 views2 answers
Plt import
Python
0 vote336 views1 answer
DataFrame changed by function
Python
0 vote334 views1 answer
No module named 'sklearn.utils.linear_assignment
Python
0 vote357 views1 answer
Qtablewidget clear python
Python
0 vote340 views1 answer
Python sum of 10 numbers from user input
Python
3 votes359 views2 answers
Python set cwd to script directory
Python
1 vote362 views1 answer
Python delete text in text file
Python