最小二乘法计算一组数据的斜率(线性回归、趋势计算)

def compute_trend(*y):
    """
    计算数据的趋势(线性回归求斜率，最小二乘法)，https://blog.csdn.net/qq_45607873/article/details/109425736
    :return:
    """
    x = np.arange(len(y[0])) + 1  # 横轴
    y = np.array(y)  # 纵轴
    xymean = np.mean(np.multiply(x, y), axis=1)  # axis=0是batch
    xmeany = np.mean(x) * np.mean(y, axis=1)
    x2mean = np.square(x).mean()
    xmean2 = x.mean()**2
    trend = (xymean-xmeany)/(x2mean - xmean2)
    return trend