arrays - 如何使用 matplotlib 为二维数组中的所有（仅 4 种）类型的值绘制图例？

Question

我有一个二维数组，其中每个值都是 of 0,1,2,3，我将其绘制为：

array = two_d_array #two_d_array is a numpy 2-D array 
plt.imshow(array)

这给了我一个图，其中每个像素由一种颜色表示，即图像有 4 种不同的颜色。我怎样才能：

• 通过自定义哪个值（其中一个0, 1 ,2, 3）对应于哪个值来绘制它？
• 绘制一个图例，说明哪种颜色属于0，哪种颜色1等等？

这个答案非常接近，但它选择识别哪种颜色而不是哪个值。

任何帮助将不胜感激！

标签： arrayspython-3.xnumpymatplotliblegend