Python 3.x 如何在matplotlib python中使用与diff值相对应的diff颜色绘制图例？

我正在尝试绘制土地利用数据集的分区，我想得到一个如图中所示的图例。我正在使用

plt.imshow（）

绘制img，输入是2D数组，数组的值是土地使用代码

我怎样才能得到这样的传奇：

---

您可以将代理艺术家添加到绘图中，如中所示

指南中的示例代码：

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()

给

---

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

可能我没有说清楚，我想从img中提取颜色，并根据这些颜色，获得传奇。您如何知道图像中的颜色对应于哪个标签？如果你只是想要颜色，只需在图像中所有独特的颜色上运行for循环，并为每个颜色创建一个补丁？我现在知道如何解决它了。第一，根据colormap set legend补丁设置自己的colormap第二