/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

4 votes185 views1 answer
Django is null
Python
0 vote200 views1 answer
Python numpy bbox
Python
48 votes213 views6 answers
Random password generator python
Python
11 votes191 views2 answers
Format python number with commas
Python
66 votes188 views6 answers
Python create folder
Python
87 votes179 views5 answers
Python update pip3
Python
0 vote202 views1 answer
Python quick tutorial
Python
1 vote174 views1 answer
How to fix geometry of a window in tkinter
Python
0 vote198 views1 answer
Remove 0 values from dataframe
Python
1 vote201 views1 answer
If none in column remove row
Python