/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

42 votes285 views6 answers
Python round 1 decimal place
Python
1 vote264 views1 answer
Python try except continue loop
Python
1 vote250 views1 answer
Convert array to set python
Python
1 vote282 views1 answer
Spacy matcher syntax
Python
2 votes264 views3 answers
How to increase bar width in python matplogtlib
Python
0 vote278 views1 answer
Find common string in two strings python
Python
2 votes257 views1 answer
Python for loop with increment
Python
0 vote293 views2 answers
From django.core.management import execute_from_command_line ImportError: No module named django.core.management
Python
0 vote274 views1 answer
Cv2.videocapture python set frame rate
Python
8 votes267 views2 answers
Python f-string padding
Python