Преобразование заданного значения цвета контура в код rgb

Question

Я создал 2D контурный график с цветовой схемой viridis в границах vmin = 0, vmax = 4. Вот код:

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

Теперь у меня есть из другого расчета точка данных в определенном месте с данными = 3.5 и хочу наложить это к контуру с использованием цветного кругом патча.

Так как я могу добавить информацию цвета RGB к окружности в пределах диапазона и используется ColorScheme моего 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)