python imshow添加图例

在Python中，可以使用matplotlib库中的imshow函数来显示图像，并使用colorbar函数添加图例。下面是一个简单的示例代码：

import matplotlib.pyplot as plt
import numpy as np

# 生成一张随机图像
img = np.random.rand(10, 10)

# 显示图像
plt.imshow(img, cmap='gray')

# 添加图例
plt.colorbar()

# 显示图像和图例
plt.show()

在上面的代码中，我们生成了一个10x10的随机矩阵，并使用imshow函数将其显示为灰度图像。然后，我们使用colorbar函数添加一个图例，最后使用show函数显示图像和图例。

在colorbar函数中，可以通过传递参数cmap来指定图例的颜色映射。在这个例子中，我们使用了'gray'颜色映射，它会将灰度值映射为黑白色调。您也可以选择其他颜色映射，如'jet'、'hot'等。

除了使用colorbar函数添加图例外，还可以使用legend函数添加自定义图例。例如，如果您想添加一张带有标签的彩色图像的图例，可以使用以下代码：

import matplotlib.pyplot as plt
import numpy as np

# 生成一张带有标签的彩色图像
img = np.random.rand(10, 10, 3)
labels = ['Red', 'Green', 'Blue']

# 显示图像
plt.imshow(img)

# 添加图例
plt.legend(labels)

# 显示图像和图例
plt.show()

在上面的代码中，我们生成了一张10x10x3的随机彩色图像，并使用legend函数添加了三个标签。最后，我们使用show函数显示图像和图例。