如何在matplotlib中将图例添加到imshow（）

Question

我在用 matplotlib

在plot()或中bar()，如果我们向其添加标签，则可以轻松放置图例。但是如果是contourf()或imshow()

我知道有一个colorbar()可以显示颜色范围的颜色，但是并不满意。我想要一个带有名称（标签）的

我能想到的是，向矩阵中的每个元素添加标签，然后，尝试legend（），以查看其是否有效，但是如何向元素添加标签（如值）？

就我而言，原始数据如下：

1,2,3,3,4
2,3,4,4,5
1,1,1,2,2

例如，1代表“草”，2代表“沙”，3代表“山” …依此类推。imshow（）非常适合我的情况，但没有图例。

我的问题是：

1. 有没有可以自动添加图例的函数，例如，在我的情况下，我只需要这样做：someFunction（’grass’，’sand’，…）

2. 如果没有，如何将标签添加到矩阵中的每个值。例如，在矩阵“草”中标记所有1，在矩阵“沙”中标记所有2 …等等。

谢谢！

编辑 ：

感谢@dnalow，它真的很聪明。但是，我仍然想知道是否有任何正式的解决方案。

Answer 1

我猜您必须伪造您的图例，因为它需要一行来创建图例。

您可以执行以下操作：

import pylab as pl
mycmap = pl.cm.jet # for example
for entry in pl.unique(raw_data):
    mycolor = mycmap(entry*255/(max(raw_data) - min(raw_data)))
    pl.plot(0, 0, "-", c=mycolor, label=mynames[entry])

pl.imshow(raw_data)
pl.legend()

当然，这还不是很令人满意。但是也许您可以在此基础上有所建树。

[编辑：添加了缺少的括号]