matplotlib颜色图-预定义到值的映射? [英] matplotlib color map - predefine mappings to values?

问题描述

我有一个正在使用imshow()查看的数组. (实际上是imsave()，但过程应该相同).

I have an array that I am viewing using imshow(). (imsave() really, but the process should be identical).

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

I know that the values in the array will be between 0-9 and wonder if it is possible to use cmap to set each output to a specific 'color'. Perhaps by mapping these to a dict?

推荐答案

只需使用ListedColormap.

作为一个快速(但丑陋)的示例:

As a quick (but ugly) example:

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

这篇关于matplotlib颜色图-预定义到值的映射?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！