python - Matplotlib 颜色图范围

Question

我正在尝试了解 matplotlib 颜色图的工作原理。考虑以下代码:

import matplotlib.pyplot as plt
cmap = plt.get_cmap('jet')
print cmap(200)

打印

(1.0, 0.46550472040668145, 0.0, 1.0)

所以我的理解是颜色图将数值(在本例中为 200)映射到颜色值(在本例中为 1.0、0.46550472040668145、0.0、1.0)。 matplotlib 如何为其颜色图设置范围？

是否可以定义应用线性映射的最大值和最小值？使用 imshow() 可以设置一个 vmin 和一个 vmax 参数，但是，我必须在颜色图级别进行设置，因为我m 稍后将颜色图提供给另一个函数。

这可能是一个关于颜色映射如何工作的更普遍的问题；例如，在 seaborn 的调色板中，也没有范围选项。

Answer 1

颜色图的范围始终在 0 到 1 之间。您需要将数据标准化到该范围。例如，要将 0 到 400 之间的范围线性映射到颜色图的颜色，

import matplotlib.pyplot as plt
cmap = plt.get_cmap('viridis')
norm = plt.Normalize(0, 400)

color = cmap(norm(200.))