python设置图例
时间: 2023-07-01 22:21:09 浏览: 43
在 Python 中,设置图例可以使用 `matplotlib` 库中的 `legend` 函数。下面是一个例子:
```python
import matplotlib.pyplot as plt
x = [1, 2, 3, 4]
y1 = [1, 4, 9, 16]
y2 = [2, 4, 6, 8]
plt.plot(x, y1, label='y1')
plt.plot(x, y2, label='y2')
plt.legend()
plt.show()
```
这个例子中,我们首先创建了两个数据集 `y1` 和 `y2`,然后使用 `plot` 函数绘制线图,并设置了两个线条的标签 `label`。最后,使用 `legend` 函数来显示图例。`legend` 函数默认会使用 `plot` 函数中设置的标签作为图例的内容。
你也可以使用 `loc` 参数来设置图例的显示位置。例如:
```python
plt.legend(loc='upper right')
```
`loc` 参数可以设置为以下值:
- `'best'`: 自动选择最佳位置
- `'upper left'`: 左上角
- `'upper right'`: 右上角
- `'lower left'`: 左下角
- `'lower right'`: 右下角
- `'center left'`: 左中间
- `'center right'`: 右中间
- `'center'`: 正中间
除了这些基本用法,`legend` 函数还有很多其他的参数可以使用。你可以查阅相关文档来了解更多信息。