写一个能够实现一元线性回归的python代码,要求输入两个列表,返回一个新的拟合过的列表
时间: 2023-06-08 13:03:35 浏览: 239
python一元线性回归
5星 · 资源好评率100%
好的,以下是一元线性回归的Python代码:
```
import numpy as np
def linear_regression(x, y):
x_mean = np.mean(x)
y_mean = np.mean(y)
numerator = sum((xi - x_mean) * (yi - y_mean) for xi, yi in zip(x, y))
denominator = sum((xi - x_mean) ** 2 for xi in x)
beta1 = numerator / denominator
beta0 = y_mean - beta1 * x_mean
return [beta0 + beta1 * xi for xi in x]
```
这个代码实现了输入两个列表x和y,返回一个拟合过的列表,这个拟合过的列表实现了一元线性回归。如果您有任何问题,请问我。
阅读全文