Search code examples
pythonmatplotlibcolorbrewer

Do matplotlib.contourf levels depend on the amount of colors in the colormap?


I'm plotting maps using contourf and I'd usually go with the default (rainbow) colorscheme with levels = 50.

#Various imports
#LOTS OF OTHER CODE BEFORE
plot = plt.contourf(to_plot, 50)
plt.show()
#LOTS OF OTHER CODE AFTER

The output is below. I do various other stuff to get the coastlines etc. It's done using iris and cartopy, if anyone's interested.

This is the result

Now I've decided that I don't want to use a rainbow scheme so I'm using some Cyntia Brewer colours:

brewer_cmap = mpl.cm.get_cmap('brewer_Reds_09')
plot = iplt.contourf(to_plot, 50, cmap=brewer_cmap) # expect 50 levels

However the output is: This is the result

You can see Here that this palette only has 9 colours. So my question is, are the contourf levels limited by the amount of available colours in the colormap? I quite like this map and I wonder if it'd be possible to generate a new one like it but with more levels of red?

I'm interested in being able to capture the variability of the data so more contour levels seems like a good idea but I'm keen on losing the rainbow scheme and just going with one based on a single colour.

Cheers!


Solution

  • Yes, it is a discrete colormap, and if you want to have a continuos one you need to make a customized colormap.

    #the colormap data can be found here: https://github.com/SciTools/iris/blob/master/lib/iris/etc/palette/sequential/Reds_09.txt
    
    In [22]:
    
    %%file temp.txt
    1.000000 0.960784 0.941176
    0.996078 0.878431 0.823529
    0.988235 0.733333 0.631373
    0.988235 0.572549 0.447059
    0.984314 0.415686 0.290196
    0.937255 0.231373 0.172549
    0.796078 0.094118 0.113725
    0.647059 0.058824 0.082353
    0.403922 0.000000 0.050980
    Overwriting temp.txt
    In [23]:
    
    c_array = np.genfromtxt('temp.txt')
    from matplotlib.colors import LinearSegmentedColormap
    plt.register_cmap(name='Test', data={key: tuple(zip(np.linspace(0,1,c_array.shape[0]), c_array[:,i], c_array[:,i])) 
                                             for key, i in zip(['red','green','blue'], (0,1,2))})
    In [24]:
    
    plt.contourf(X, Y, Z, 50, cmap=plt.get_cmap('Test'))
    plt.colorbar()
    Out[24]:
    <matplotlib.colorbar.Colorbar instance at 0x108948320>
    

    enter image description here