How to apply piecewise linear fit in Python?

by

You can use numpy.piecewise() to create the piecewise function and then use curve_fit(), Here is the code

from scipy import optimize
import matplotlib.pyplot as plt
import numpy as np
%matplotlib inline

x = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10 ,11, 12, 13, 14, 15], dtype=float)
y = np.array([5, 7, 9, 11, 13, 15, 28.92, 42.81, 56.7, 70.59, 84.47, 98.36, 112.25, 126.14, 140.03])

def piecewise_linear(x, x0, y0, k1, k2):
    return np.piecewise(x, [x < x0], [lambda x:k1*x + y0-k1*x0, lambda x:k2*x + y0-k2*x0])

p , e = optimize.curve_fit(piecewise_linear, x, y)
xd = np.linspace(0, 15, 100)
plt.plot(x, y, "o")
plt.plot(xd, piecewise_linear(xd, *p))

the output:

enter image description here

For an N parts fitting, please reference segments_fit.ipynb

More Related Contents:

  1. How to do exponential and logarithmic curve fitting in Python? I found only polynomial fitting
  2. python numpy/scipy curve fitting
  3. scipy curve_fit doesn’t like math module
  4. Fitting a closed curve to a set of points
  5. Interpolating a closed curve using scipy
  6. How do I read CSV data into a record array in NumPy?
  7. Python out of memory on large CSV file (numpy)
  8. Multiple linear regression in Python
  9. matplotlib – extracting data from contour lines
  10. Fitting a Normal distribution to 1D data
  11. Python curve_fit with multiple independent variables
  12. Populate a Pandas SparseDataFrame from a SciPy Sparse Matrix
  13. Installing NumPy and SciPy on 64-bit Windows (with Pip)
  14. How to extend an array in-place in Numpy?
  15. Interpolation over an irregular grid
  16. fitting exponential decay with no initial guessing
  17. Overflow Error in Python’s numpy.exp function
  18. How do I put a constraint on SciPy curve fit?
  19. How to plot line (polygonal chain) with numpy/scipy/matplotlib with minimal smoothing
  20. How can I efficiently process a numpy array in blocks similar to Matlab’s blkproc (blockproc) function
  21. Load just part of an image in python
  22. How do I plot list of tuples in Python?
  23. generalized cumulative functions in NumPy/SciPy?
  24. Principal component analysis in Python
  25. How to convert a column or row matrix to a diagonal matrix in Python?
  26. How to do a polynomial fit with fixed points
  27. Numpy transpose of 1D array not giving expected result
  28. Fit mixture of two gaussian/normal distributions to a histogram from one set of data, python
  29. How to create a density plot
  30. Speedup scipy griddata for multiple interpolations between two irregular grids

Leave a Comment