使用matplotlib

Question

假设我有一个10x10的1维ndarray，其中所有可能的值都是[0,100,200]。我想手动创建一个非常简单的颜色图manual_cmap，告诉matplotlib：

if value==0   : make this pixel black
if value==100 : make this pixel yellow
if value==200 : make this pixel red
etc.

然后，我可以使用这个manual_cmap来保存我想要的颜色的图像，而不必通过将它们转换为三维RGB数组来使用更多的内存：

plt.imsave('colored_image.png', oneD_array, cmap=manual_cmap)

Ps 我已经看到了一些用于创建自定义色彩映射的示例，但我没有找到任何简单的示例，这些示例允许我手动定义一个简单的查找表，如colormap，只需几个值 - 颜色对。