python - matplotlib 颜色映射 - 预定义映射到值？

Question

我有一个正在查看的数组imshow()。（确实是 imsave()，但过程应该是相同的）。

我知道数组中的值将在 0-9 之间，并且想知道是否可以使用 cmap 将每个输出设置为特定的“颜色”。也许通过将这些映射到字典？

Answer 1

只需使用ListedColormap.

作为一个快速（但丑陋）的例子：

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap

cmap = ListedColormap(['red', 'green', 'blue', 'black'], 'indexed')

fig, ax = plt.subplots()
im = ax.imshow([range(4)], interpolation='none', cmap=cmap)
fig.colorbar(im)
plt.show()