用numpy/python推断数据

Question

假设我有一个简单的数据集.也许是字典形式,它看起来像这样:

{1:5, 2:10, 3:15, 4:20, 5:25}

(订单总是在上升).我想要做的是从逻辑上弄清楚下一个数据点最有可能是什么.例如,在这种情况下,它会{6: 30}

什么是最好的方法呢？

Answer 1

你也可以使用numpy的polyfit:

data = np.array([[1,5], [2,10], [3,15], [4,20], [5,25]])
fit = np.polyfit(data[:,0], data[:,1] ,1) #The use of 1 signifies a linear fit.

fit
[  5.00000000e+00   1.58882186e-15]  #y = 5x + 0

line = np.poly1d(fit)
new_points = np.arange(5)+6

new_points
[ 6, 7, 8, 9, 10]

line(new_points)
[ 30.  35.  40.  45.  50.]

这允许您非常容易地改变多项式拟合的程度,因为函数polyfit采用以下参数np.polyfit(x data, y data, degree).显示的是线性拟合,返回的数组fit[0]*x^n + fit[1]*x^(n-1) + ... + fit[n-1]*x^0在任何程度上都是如此n.该poly1d函数允许您将此数组转换为一个函数,该函数返回任何给定值的多项式的值x.

一般来说,没有一个很好理解的模型的外推将充其量只有零星的结果.

指数曲线拟合.

from scipy.optimize import curve_fit

def func(x, a, b, c):
    return a * np.exp(-b * x) + c

x = np.linspace(0,4,5)
y = func(x, 2.5, 1.3, 0.5)
yn = y + 0.2*np.random.normal(size=len(x))

fit ,cov = curve_fit(func, x, yn)
fit
[ 2.67217435  1.21470107  0.52942728]         #Variables

y
[ 3.          1.18132948  0.68568395  0.55060478  0.51379141]  #Original data

func(x,*fit)
[ 3.20160163  1.32252521  0.76481773  0.59929086  0.5501627 ]  #Fit to original + noise

Answer 2

在Python聊天中与您讨论后,将数据拟合为指数.这应该是一个相对较好的指标,因为你不是在寻找长期的推断.

import numpy as np
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt

def exponential_fit(x, a, b, c):
    return a*np.exp(-b*x) + c

if __name__ == "__main__":
    x = np.array([0, 1, 2, 3, 4, 5])
    y = np.array([30, 50, 80, 160, 300, 580])
    fitting_parameters, covariance = curve_fit(exponential_fit, x, y)
    a, b, c = fitting_parameters

    next_x = 6
    next_y = exponential_fit(next_x, a, b, c)

    plt.plot(y)
    plt.plot(np.append(y, next_y), 'ro')
    plt.show()

最右侧轴上的红点显示下一个"预测"点.

Answer 3

正如指出的这个回答一个相关问题,由于SciPy的的0.17.0或更新的版本,有一个选项scipy.interpolate.interp1d,使线性外推.在你的情况下,你可以这样做:

>>> import numpy as np
>>> from scipy import interpolate

>>> x = [1, 2, 3, 4, 5]
>>> y = [5, 10, 15, 20, 25]
>>> f = interpolate.interp1d(x, y, fill_value = "extrapolate")
>>> print(f(6))
30.0