/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

0 vote205 views1 answer
Show all rows with nan for a column value pandas
Python
0 vote195 views1 answer
Python length checker/fill
Python
0 vote205 views1 answer
Ansible playbook to set passwordless sudo
Python
1 vote198 views1 answer
Index a dictionary python
Python
33 votes170 views3 answers
Python return if else one line
Python
1 vote204 views1 answer
'set' object is not reversible
Python
0 vote214 views1 answer
Add an element to list python
Python
1 vote274 views1 answer
Redirected but the response is missing a Location: header.
Python
5 votes231 views2 answers
Get_object_or_404 django
Python
0 vote208 views1 answer
Read a file line-byline in python
Python