python - 将给定的轮廓颜色值转换为 rgb 代码

Question

我在边界 vmin=0,vmax=4 中创建了一个带有 viridis 配色方案的二维轮廓图。这是代码：

ax.contourf(xi/d,yi/d,data/d,12,vmin=0,vmax=4,cmap=vs.viridis(),aspect='equal')

现在我从另一个计算中得到一个特定位置的数据点，数据=3.5，并希望使用彩色圆圈补丁将其叠加到轮廓上。

那么如何将rgb颜色信息添加到范围内的圆圈并使用我的contourf的颜色方案

感谢帮助

Answer 1

您可以使用matplotlib.colors.Normalize实例来执行此操作：

import matplotlib.pyplot as plt
import matplotlib.colors as colors

norm = colors.Normalize(vmin=0,vmax=4)
cmap = plt.cm.viridis

mycolor = cmap(norm(3.5))
print mycolor
# (0.67848900000000001, 0.86374200000000001, 0.189503, 1.0)

请注意，如果颜色图是从其他文件导入的，则需要先注册它matplotlib才能工作。例如：

import myviridisfile as vs
plt.register_cmap(name='viridis', cmap=vs.viridis)