/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 vote224 views1 answer
Add prefix to names in directory python
Python
41 votes302 views6 answers
Check if element in list Python
Python
0 vote237 views2 answers
How to plot two different class in different colour in python
Python
0 vote252 views1 answer
Git ignore everything but python files
Python
13 votes229 views1 answer
Selenium webdriver python
Python
0 vote265 views1 answer
Matplotlib set dpi 300
Python
0 vote238 views1 answer
Infix to postfix python code
Python
0 vote264 views1 answer
Items of a list not in another list python
Python
11 votes239 views1 answer
Python discord
Python
7 votes224 views4 answers
Print multiple lines python
Python