matplotlib：获取colormap数组

Question

我是matplotlib的新手，并且陷入了色彩映射。

在matplotlib中，我如何获得特定色图的整个RGB颜色数组，让我们说“热”。例如，如果我在MATLAB中，我会这样做：

# in matlab
c = hot(256);
disp(c)

有什么想法吗？

Answer 1

您可以通过将colormap作为函数调用来查找值，并且它接受numpy数组来一次查询多个值：

In [13]: cm.hot(arange(256))
Out[13]: 
array([[ 0.0416    ,  0.        ,  0.        ,  1.        ],
       [ 0.05189484,  0.        ,  0.        ,  1.        ],
       [ 0.06218969,  0.        ,  0.        ,  1.        ],
       ..., 
       [ 1.        ,  1.        ,  0.96911762,  1.        ],
       [ 1.        ,  1.        ,  0.98455881,  1.        ],
       [ 1.        ,  1.        ,  1.        ,  1.        ]])

Answer 2

知道了！所以你只需进入Matlab的命令窗口并输入

即可

cmap = colormap（nameOfTheColormapYouWant）

Matlab中可能的色图是：parula，jet，hsv，hot，cool，spring，summer，autumn，winter，grey，bone，copper，pink，lines，colorcube，prism，flag。

你得到一个矩阵，其中每一行都是用于色彩图的颜色代码。