Como pyplot.contourf escolhe cores de um mapa de cores?
Eu tenho jogado com ofrom_levels_and_colors
, para que eu possa ter uma barra de cores estendida em um gráfico pcolormesh, semelhante ao contourf. Aqui está o meu exemplo de plotagem de contorno:
import numpy as np
import matplotlib.pyplot as plt
a = np.arange(12)[:,np.newaxis] * np.ones(8)
levels = np.arange(1.5, 10, 2)
plt.contourf(a, cmap='RdYlBu', levels=levels, extend='both')
plt.colorbar()
Para produzir um gráfico pcolormesh análogo, preciso fornecer uma sequência de cores, portanto, tenho:
from matplotlib.colors import from_levels_and_colors
n_colors = len(levels) + 1
cmap = plt.get_cmap('RdYlBu', n_colors)
colors = cmap(range(cmap.N))
cmap, norm = from_levels_and_colors(levels, colors, extend='both')
plt.pcolormesh(a, cmap=cmap, norm=norm)
plt.colorbar()
As quatro cores do meio no pcolormesh são mais claras que no contorno. Como posso escolhê-los para que correspondam?