/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 vote187 views2 answers
Logarithmic 2d histogram
Python
0 vote198 views3 answers
Pow
Python
0 vote170 views1 answer
Add list of dictionaries to pandas dataframe
Python
9 votes173 views3 answers
A value is trying to be set on a copy of a slice from a DataFrame. Try using .loc[row_indexer,col_indexer] = value instead
Python
11 votes161 views2 answers
Python functions with input
Python
2 votes173 views1 answer
Python repeat every n seconds
Python
-1 vote194 views1 answer
How to make a variable in python 3
Python
0 vote183 views1 answer
How to add a key and a value to a dictionary in python\
Python
2 votes173 views1 answer
Python apply a function on each element of a array
Python
6 votes177 views2 answers
Python change working directory to file directory
Python