/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 vote132 views1 answer
How to search a file in windows 10 using python
Python
2 votes152 views4 answers
Python how to count items in array
Python
0 vote128 views1 answer
How to import a all the modules in a packege python
Python
8 votes119 views3 answers
How to add a list in python
Python
1 vote117 views1 answer
Calculator in one line in python
Python
0 vote137 views1 answer
For an HTML output, you don’t need an additional library. It simply goes like this:
Python
0 vote128 views2 answers
Get random consonant python
Python
0 vote125 views1 answer
Matplotlib show grid for log or logit
Python
1 vote139 views1 answer
Django integer field example
Python Django
0 vote142 views1 answer
Concatenate dataframes using one column
Python