matplotlib:获取颜色图数组

Question

我是 matplotlib 的新手，并且被困在颜色图中。

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

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

有任何想法吗？

Answer 1

您可以通过将颜色图作为函数调用来查找值，它接受 numpy 数组以一次查询多个值:

In [12]: from matplotlib import cm
In [13]: cm.hot(range(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.        ]])