给plt.scatter加相关分析线
时间: 2023-10-14 07:20:36 浏览: 77
scatter 分析
要给 `plt.scatter` 加相关分析线,可以使用 `numpy` 库计算出数据的相关系数,然后使用 `plt.plot` 画出相关分析线。以下是一个简单的示例代码:
```python
import numpy as np
import matplotlib.pyplot as plt
# 生成数据
x = np.random.rand(50)
y = 2 * x + np.random.rand(50)
# 计算相关系数
corr_coef = np.corrcoef(x, y)[0, 1]
# 绘制散点图和相关分析线
plt.scatter(x, y)
plt.plot([min(x), max(x)], [min(x) * corr_coef, max(x) * corr_coef], color='red')
# 添加图例和坐标轴标签
plt.legend(['Correlation analysis line'], loc='upper left')
plt.xlabel('x')
plt.ylabel('y')
# 显示图形
plt.show()
```
在这个示例中,我们首先生成了一些随机数据,然后使用 `np.corrcoef` 函数计算出数据的相关系数。然后,我们使用 `plt.plot` 画出相关分析线,这条线的斜率就是相关系数。最后,我们添加了图例和坐标轴标签,并使用 `plt.show` 显示图形。
阅读全文