/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

45 votes191 views3 answers
Mischievous in spanish
Python
1 vote222 views2 answers
Making a basic network scanner using python
Python
1 vote211 views1 answer
Read bytes from file python
Python
0 vote213 views1 answer
How to write stuff in python
Python
1 vote217 views2 answers
Django make new application folder
Python Django
0 vote225 views1 answer
Hwo to except every error in python try statemen
Python
2 votes220 views1 answer
Python can't subtract offset-naive and offset-aware datetimes
Python
0 vote190 views1 answer
File "<ipython-input-12-48c6c043344b>", line 29 coin = random.randint(0,1) ^ IndentationError: expected an indented block
Python
0 vote246 views1 answer
Cbv uk django
Python Django
0 vote217 views1 answer
Python import shelve
Python