python - matplotlib カラー マップ - 値へのマッピングを事前定義しますか?

Question

を使用して表示している配列がありimshow()ます。(実際には imsave() ですが、プロセスは同じである必要があります)。

配列の値が 0 ～ 9 になることはわかっていますが、cmap を使用して各出力を特定の「色」に設定できるかどうか疑問に思っています。おそらく、これらをdictにマッピングすることによってですか？

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