/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

1 vote181 views1 answer
Display 2d numpy array as image
Python
5 votes155 views2 answers
Who created python
Python
0 vote168 views1 answer
Could not find a version that satisfies the requirement psycopg2>=2.8 (from pgcli) (from versions: 2.7.5, 2.7.6, 2.7.6.1, 2.7.7)
Python
1 vote207 views1 answer
Django queryset group by count
Python Django
0 vote162 views1 answer
Convert timestamp datetime to int no astype
Python
1 vote173 views1 answer
Conda specify multiple channels
Python
2 votes168 views1 answer
How to change the color of the cursor in tkinter
Python
0 vote170 views1 answer
Python create sqlite db in memory
Python
2 votes178 views2 answers
Invert y axis python
Python
1 vote174 views2 answers
Input spaces seperated integers in python
Python