/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 vote324 views1 answer
Python print file content
Python
25 votes319 views3 answers
Store all files name in a folder python
Python
2 votes299 views1 answer
Python search for word is in column
Python
148 votes353 views10 answers
Grepper
Python
1 vote366 views1 answer
Elif in django template
Python Django
4 votes294 views1 answer
What makes the bot talk? is there a word i have to say?
Python
-1 vote297 views1 answer
Python library to convert decimal into octal and hexadecimal
Python
-1 vote262 views1 answer
How to create user input in python
Python
0 vote314 views2 answers
How to get current page url in django template
Python
0 vote297 views1 answer
Python how to move multiple values from one list to another at once
Python