python中 legend()函数在散点图怎么用
时间: 2024-09-06 16:08:03 浏览: 51
python3使用matplotlib绘制散点图
5星 · 资源好评率100%
在Python中,`legend()` 函数是绘图库Matplotlib中的一个功能,用于在图中添加图例。在绘制散点图时,`legend()`可以帮助你标识出不同数据集或不同数据点的含义。
以下是如何在散点图中使用`legend()`函数的一个简单例子:
```python
import matplotlib.pyplot as plt
# 准备数据
x = [1, 2, 3, 4, 5]
y1 = [1, 3, 2, 3, 2]
y2 = [3, 2, 3, 1, 3]
# 绘制散点图
plt.scatter(x, y1, label='数据集1') # 添加标签以便在图例中显示
plt.scatter(x, y2, label='数据集2') # 添加标签以便在图例中显示
# 添加图例
plt.legend() # 默认情况下,`loc`参数设置为'best',Matplotlib会自动选择最佳位置
# 显示图表
plt.show()
```
在上述代码中,`scatter`函数用于绘制散点图,并且通过`label`参数为每组数据指定了一个标签。随后调用`legend()`函数时,Matplotlib会根据这些标签来创建图例,并自动放置在图中合适的位置。
请注意,`legend()`函数还可以通过参数`loc`来指定图例的位置,例如`loc='upper left'`可以将图例放在左上角,或者你也可以通过`bbox_to_anchor`参数来自定义图例的位置。
阅读全文