graphtools.py 文件源码

def getLinearModel(x_values, y_values, k=1.0, l=1.0):
    gradient, intercept, r_value, p_value, std_err = stats.linregress(x_values,y_values)    

    y_model = []

    grad = k*gradient
    interc = l*intercept

    for x in x_values:
        y = trendline(x, grad, interc)
        y_model.append(y)

    return y_model