如何在matplotlib python中绘制不同颜色的图例对应不同的值？

Question

我正在尝试绘制土地利用数据集的子区域，我想获得图片中的图例。我正在使用plt.imshow()绘制img，输入为二维数组，数组的值为土地使用代码。

我怎样才能得到这样的图例：

Answer 1

您可以将代理艺术家添加到情节中，如matplotlib legend guide 所示。

指南中的示例代码：

import matplotlib.patches as mpatches
import matplotlib.pyplot as plt

red_patch = mpatches.Patch(color='red', label='The red data')
plt.legend(handles=[red_patch])

plt.show()

这给了

不知道如何获得图例标题。