如何在Python中获得分段线性函数

时间:2018-10-23 15:43:29

标签: python piecewise

我想从一组点中获得分段线性函数。这是视觉示例:

import matplotlib.pyplot as plt
x = [1,2,7,9,11]
y = [2,5,9,1,11]
plt.plot(x, y)
plt.show()

piecewise illustration

因此,我需要一个包含两个列表并返回分段线性函数的函数。我不需要回归或任何最小二乘拟合。

我可以尝试自己写,但是想知道是否已经写了一些东西。到目前为止,我只发现返回回归的代码

1 个答案:

答案 0 :(得分:1)

尝试np.interp。它插值。

这是一个小例子。

>>> import matplotlib.pyplot as plt
>>> import numpy as np

>>> x = [1,2,7,9,11]
>>> y = [2,5,9,1,11]

>>> np.interp([1.5, 3], x, y)
array([ 3.5,  5.8])

请注意,要确保采样点正确,并确保x增加。