使用matplotlib颜色图进行颜色查找

Question

我需要从颜色表中查找任意数字的RGB值。

假设我有一个颜色图'spring'

from matplotlib import cm

c = cm.get_cmap('spring')

现在假设我有一个0到1的值，我想从c提取RGB值

import numpy as np

np.random.seed([3, 1415])
v = np.random.rand(1)

v

0.4449393091533107

如何映射以获得相应的RGB？

如果我看

c._segmentdata

{'blue': ((0.0, 1.0, 1.0), (1.0, 0.0, 0.0)),
 'green': ((0.0, 0.0, 0.0), (1.0, 1.0, 1.0)),
 'red': ((0.0, 1.0, 1.0), (1.0, 1.0, 1.0))}

...我不知所措。

我已经查看了此链接https://matplotlib.org/users/colormapnorms.html

但这告诉我如何对整个2D矩阵执行此操作。

Answer 1

除非我误解了您的要求，否则您可以执行以下操作：

c(v[0])

这将为您提供以下RGBA阵列：

(1.0, 0.44313725490196076, 0.55686274509803924, 1.0)

因此，如果只需要RGB ，则可以根据需要对其进行索引：

>>> c(v[0])[:3]
(1.0, 0.44313725490196076, 0.55686274509803924)

文档中的更多信息