用python语言,画出散点图,并在散点图中画出趋势线
时间: 2023-06-11 10:07:32 浏览: 251
python使用Plotly绘图工具绘制散点图、线形图
可以使用`matplotlib`库来画出散点图和趋势线。下面是一个示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成随机数据
x = np.random.rand(50)
y = np.random.rand(50)
# 画散点图
plt.scatter(x, y)
# 计算趋势线的斜率和截距
slope, intercept = np.polyfit(x, y, 1)
# 画趋势线
plt.plot(x, slope*x + intercept, color='red')
# 显示图形
plt.show()
```
解释一下代码:
1. `np.random.rand(50)`生成了一个长度为50的随机数组,用作x轴和y轴的数据。
2. `plt.scatter(x, y)`画出了散点图。
3. `np.polyfit(x, y, 1)`计算了一次函数的斜率和截距。
4. `plt.plot(x, slope*x + intercept, color='red')`画出了趋势线,其中`slope*x + intercept`就是一次函数的表达式。
5. `plt.show()`显示图形。
阅读全文