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

我有一个我正在使用imshow()查看的数组. (imsave()确实如此,但过程应该相同).

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

解决方法:

只需使用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()

标签：python,matplotlib,python-2-x
来源： https://codeday.me/bug/20191001/1839668.html