用python代码做一个线性回归
时间: 2024-05-05 20:19:28 浏览: 13
好的,下面是用 Python 实现的线性回归示例代码:
```
import numpy as np
import matplotlib.pyplot as plt
# 输入数据
X = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
Y = np.array([3, 5, 7, 9, 11, 13, 15, 17, 19, 21])
# 定义模型
def linear_regression(X, Y):
n = len(X)
sum_x = X.sum()
sum_y = Y.sum()
sum_xy = (X * Y).sum()
sum_x2 = (X ** 2).sum()
a = (n * sum_xy - sum_x * sum_y) / (n * sum_x2 - sum_x ** 2)
b = (sum_y - a * sum_x) / n
return a, b
# 训练模型
a, b = linear_regression(X, Y)
# 输出模型参数
print("a = {}, b = {}".format(a, b))
# 绘制结果
plt.scatter(X, Y)
plt.plot(X, a * X + b, color='red')
plt.show()
```
这是一个简单的线性回归示例,我们输入了 X 和 Y 的数据点,然后通过计算线性回归的系数 a 和 b,最后将拟合曲线绘制出来。
注意,以上只是一个示例,实际场景下需要根据具体数据情况调整参数和算法,以获得更好的拟合效果。