python代码如何实现画出散点图和趋势线
时间: 2024-09-21 12:13:29 浏览: 89
python 画三维图像 曲面图和散点图的示例
在Python中,你可以使用matplotlib库来创建散点图和绘制趋势线。下面是一个简单的例子:
```python
import matplotlib.pyplot as plt
# 假设你有两个数据列表,x_data代表x轴坐标,y_data代表y轴坐标
x_data = [1, 2, 3, 4, 5]
y_data = [1, 4, 9, 16, 25] # 这里我们使用了简单的平方数作为示例
# 创建散点图
plt.scatter(x_data, y_data)
plt.title('Scatter Plot') # 图表标题
plt.xlabel('X Axis') # X轴标签
plt.ylabel('Y Axis') # Y轴标签
# 添加一条简单的一次线性拟合趋势线
slope, intercept = np.polyfit(x_data, y_data, 1) # 使用numpy计算斜率和截距
line_x = np.linspace(min(x_data), max(x_data)) # 创建线的x坐标范围
line_y = slope * line_x + intercept # 计算对应y值
plt.plot(line_x, line_y, 'r-', label='Trend Line (y = mx+b)') # 红色实线表示趋势线
# 显示图表并添加图例
plt.legend()
plt.show() # 展示绘图结果
```
在这个例子中,`scatter()`函数用于生成散点图,`plot()`函数则用于绘制趋势线。`np.polyfit()`函数计算了一次线性拟合的斜率和截距。
阅读全文