用python写回归曲线
时间: 2023-07-04 18:07:58 浏览: 234
用python做回归分析程序
可以使用Python中的matplotlib库来画回归曲线。以下是一个简单的例子:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成一些随机数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 4, 5, 4, 5])
# 使用numpy.polyfit函数拟合数据,并得到回归方程系数
coefficients = np.polyfit(x, y, 1)
m = coefficients[0]
b = coefficients[1]
# 画出原始数据和回归曲线
plt.scatter(x, y)
plt.plot(x, m*x+b)
plt.show()
```
这个例子中,我们先生成了一些随机数据,然后使用numpy.polyfit函数拟合数据并得到回归方程系数。接着,我们使用matplotlib的scatter函数画出原始数据点,再用plot函数画出回归曲线。最后,我们使用show函数显示图形。
阅读全文