plt.imshow添加图例
时间: 2023-09-08 12:15:06 浏览: 174
matplotlib基础绘图命令之imshow的使用
5星 · 资源好评率100%
`plt.imshow()`函数不支持添加图例。如果需要添加图例,可以使用`matplotlib`库中的`legend()`函数。一种方法是在`imshow()`函数调用之前创建一个`Line2D`对象,并使用`label`参数指定标签,然后在调用`legend()`函数时传递这些`Line2D`对象。例如:
```python
import matplotlib.pyplot as plt
import numpy as np
# 创建一个图像
fig, ax = plt.subplots()
# 生成一些数据
x = np.arange(0, 10, 0.1)
y = np.sin(x)
# 绘制图像
im = ax.imshow(np.random.random((10, 10)), cmap=plt.cm.Blues, extent=(0, 10, 0, 10))
# 创建一个Line2D对象,指定标签
line = plt.Line2D(x, y, color='red', label='sin(x)')
# 添加Line2D对象到图例中
ax.legend(handles=[line])
# 显示图像
plt.show()
```
这个例子中,我们创建了一个`Line2D`对象来表示`sine`函数,并使用`label`参数指定标签。然后,我们将该`Line2D`对象传递给`ax.legend()`函数来添加到图例中。
阅读全文