用python写段代码,使用最小二乘法实现线性回归
时间: 2024-05-12 08:18:12 浏览: 83
python实现线性回归的最小二乘法和梯度下降法.zip
好的,下面是使用最小二乘法实现线性回归的 Python 代码。
```python
import numpy as np
# 构造样本数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([1.5, 3, 4.5, 6, 7.5])
# 建立矩阵方程
A = np.vstack([x, np.ones(len(x))]).T
b = np.vstack([y]).T
# 最小二乘法求解
result = np.linalg.lstsq(A, b, rcond=None)
# 提取回归系数及截距
k, b = result[0]
print("回归系数 k = ", k)
print("截距 b = ", b)
```
注:本代码使用 NumPy 库实现。
阅读全文