使用 matplotlib 库中的 scatter 函数绘制散点图怎么单独设置一个图例
时间: 2024-03-15 19:44:53 浏览: 78
在使用 `matplotlib` 库中的 `scatter` 函数绘制散点图时,你可以使用 `legend` 函数添加图例。具体步骤如下:
1. 在 `scatter` 函数中设置 `label` 参数为需要添加图例的数据的名称或描述。
```python
import matplotlib.pyplot as plt
x1 = [1, 2, 3, 4, 5]
y1 = [2, 4, 6, 8, 10]
x2 = [1, 2, 3, 4, 5]
y2 = [1, 3, 5, 7, 9]
plt.scatter(x1, y1, label='Group 1')
plt.scatter(x2, y2, label='Group 2')
```
2. 使用 `legend` 函数添加图例。其中,`loc` 参数指定图例的位置,`bbox_to_anchor` 参数指定图例的位置,`ncol` 参数指定图例的列数。
```python
# 添加图例
plt.legend(loc='upper left', bbox_to_anchor=(1, 1), ncol=1)
plt.show()
```
这样,就可以在散点图中添加一个图例了。
完整代码示例:
```python
import matplotlib.pyplot as plt
x1 = [1, 2, 3, 4, 5]
y1 = [2, 4, 6, 8, 10]
x2 = [1, 2, 3, 4, 5]
y2 = [1, 3, 5, 7, 9]
plt.scatter(x1, y1, label='Group 1')
plt.scatter(x2, y2, label='Group 2')
# 添加图例
plt.legend(loc='upper left', bbox_to_anchor=(1, 1), ncol=1)
plt.show()
```
阅读全文