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