使用python写一个单变量线性回归函数

好的，下面是使用Python编写一个单变量线性回归函数的例子：

import numpy as np

def linear_regression(x, y):
    n = len(x)                                 # 样本数量
    x_mean = np.mean(x)                        # x 的平均值
    y_mean = np.mean(y)                        # y 的平均值
    
    # 计算回归系数 b1 和截距 b0
    numerator = denominator = 0
    for i in range(n):
        numerator += (x[i] - x_mean) * (y[i] - y_mean)
        denominator += (x[i] - x_mean) ** 2
    b1 = numerator / denominator
    b0 = y_mean - b1 * x_mean
    
    return b0, b1

这个函数接受两个参数 `x` 和 `y`，分别代表自变量和因变量。函数会通过最小二乘法计算出回归系数 `b1` 和截距 `b0`，并将它们作为元组返回。