举个线性回归的python例子
时间: 2023-09-26 13:11:20 浏览: 85
机器学习中线性回归的例子,python代码编写.rar
当然,下面是一个简单的线性回归的Python例子:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([3, 4, 5.5, 6, 7])
# 计算斜率和截距
slope, intercept = np.polyfit(x, y, 1)
# 画出原始数据和拟合直线
plt.scatter(x, y)
plt.plot(x, slope*x + intercept, 'r')
plt.show()
```
这个代码生成了一组简单的数据,使用 numpy 的 `polyfit` 函数计算了斜率和截距,最后使用 matplotlib 画出了原始数据和拟合直线。
阅读全文