DOC: Add missing cmaps to perception doc (fix for #8073) #8156
- Loading branch information
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,120 +1,100 @@ | ||
| ''' | ||
| ========================== | ||
| Lightness of the colormaps | ||
| ========================== | ||
|
|
||
| For each colormap, plot the lightness parameter L* from CIELAB colorspace | ||
| along the y axis vs index through the colormap. Colormaps are examined in | ||
| categories as in the original matplotlib gallery of colormaps. | ||
| ''' | ||
|
|
||
| import numpy as np | ||
| import matplotlib as mpl | ||
| import matplotlib.pyplot as plt | ||
| from matplotlib import cm | ||
| from colorspacious import cspace_converter | ||
| from colormaps import cmaps # the colormaps, grouped by category | ||
|
|
||
| mpl.rcParams.update({'font.size': 12}) | ||
|
|
||
| # Number of colormap par subplot for particular cmap categories | ||
| _DSUBS = {'Perceptually Uniform Sequential': 4, 'Sequential': 6, | ||
|
There was a problem hiding this comment. It'd be nice if we could compute the values in someway. I that would require the colors module to be reorganzed in some way. And that's definitely out of the scope of this PR. There was a problem hiding this comment. The new order (and the addition of some previously missing colormaps) in 3c29961 allow a more uniform set of values for I have kept an explicit dictionary with all values inside, in case one would want to further tweak these values. One could also use these dictionary to just describe the non-standard cases and rely on |
||
| 'Sequential (2)': 5, 'Diverging': 6, 'Qualitative': 4, | ||
| 'Miscellaneous': 6} | ||
|
|
||
| # Spacing between the colormaps of a subplot | ||
|
There was a problem hiding this comment. I think it makes more sense to change the figure size like it used to be done. Now you get a different sized Axes and different number of ticks for each figure. There was a problem hiding this comment. I do not understand to which situation you are refering to: looking at blame on GH, the There was a problem hiding this comment. Oh, this comment probably shouldn't be here but on a But if the figure size changes based on the number of plots, then I don't understand why 00 and 05 have 3 y-ticks, but the others have 5. There was a problem hiding this comment. I did not even notice the different amount of ticks ^^... If I had to guess, I would say that it may be an artifact from There was a problem hiding this comment. That might be it. It would be nice if we didn't have to use |
||
| _DC = {'Perceptually Uniform Sequential': 1.15, 'Sequential': 0.6, | ||
| 'Sequential (2)': 1.15, 'Diverging': 1.3, 'Qualitative': 1.3, | ||
| 'Miscellaneous': 1.4} | ||
|
|
||
| # Indices to step through colormap | ||
| x = np.linspace(0.0, 1.0, 100) | ||
|
|
||
| # Do plot | ||
| for cmap_category, cmap_list in cmaps: | ||
|
|
||
| # Do subplots so that colormaps have enough space. | ||
| # Default is 5 colormaps per subplot. | ||
| dsub = _DSUBS.get(cmap_category, 5) | ||
| nsubplots = int(np.ceil(len(cmap_list) / float(dsub))) | ||
|
|
||
| # squeeze=False to handle similarly the case of a single subplot | ||
| fig, axes = plt.subplots(nrows=nsubplots, squeeze=False, | ||
| figsize=(7, 2.6*nsubplots)) | ||
|
|
||
| for i, ax in enumerate(axes.flat): | ||
|
|
||
| locs = [] # locations for text labels | ||
|
|
||
| for j, cmap in enumerate(cmap_list[i*dsub:(i+1)*dsub]): | ||
|
|
||
| # Get RGB values for colormap and convert the colormap in | ||
| # CAM02-UCS colorspace. lab[0, :, 0] is the lightness. | ||
| rgb = cm.get_cmap(cmap)(x)[np.newaxis, :, :3] | ||
| lab = cspace_converter("sRGB1", "CAM02-UCS")(rgb) | ||
|
|
||
| # Plot colormap L values. Do separately for each category | ||
| # so each plot can be pretty. To make scatter markers change | ||
| # color along plot: | ||
| # http://stackoverflow.com/questions/8202605/matplotlib-scatterplot-colour-as-a-function-of-a-third-variable | ||
|
|
||
| if cmap_category == 'Sequential': | ||
| # These colormaps all start at high lightness but we want them | ||
| # reversed to look nice in the plot, so reverse the order. | ||
| y_ = lab[0, ::-1, 0] | ||
| c_ = x[::-1] | ||
| else: | ||
| y_ = lab[0, :, 0] | ||
| c_ = x | ||
|
|
||
| dc = _DC.get(cmap_category, 1.2) # cmaps horizontal spacing | ||
| ax.scatter(x + j*dc, y_, c=c_, cmap=cmap, s=300, linewidths=0.0) | ||
|
|
||
| # Store locations for colormap labels | ||
| if cmap_category in ('Perceptually Uniform Sequential', | ||
| 'Sequential', 'Sequential (2)'): | ||
| locs.append(x[-1] + j*dc) | ||
| elif cmap_category in ('Diverging', 'Qualitative', | ||
| 'Miscellaneous'): | ||
| locs.append(x[int(x.size/2.)] + j*dc) | ||
|
|
||
| # Set up the axis limits: | ||
| # * the 1st subplot is used as a reference for the x-axis limits | ||
| # * lightness values goes from 0 to 100 (y-axis limits) | ||
| ax.set_xlim(axes[0, 0].get_xlim()) | ||
| ax.set_ylim(0.0, 100.0) | ||
|
|
||
| # Set up labels for colormaps | ||
| ax.xaxis.set_ticks_position('top') | ||
| ticker = mpl.ticker.FixedLocator(locs) | ||
| ax.xaxis.set_major_locator(ticker) | ||
| formatter = mpl.ticker.FixedFormatter(cmap_list[i*dsub:(i+1)*dsub]) | ||
| ax.xaxis.set_major_formatter(formatter) | ||
| ax.xaxis.set_tick_params(rotation=50) | ||
|
|
||
| ax.set_xlabel(cmap_category + ' colormaps', fontsize=14) | ||
| fig.text(0.0, 0.55, 'Lightness $L^*$', fontsize=12, | ||
| transform=fig.transFigure, rotation=90) | ||
|
|
||
| fig.tight_layout(h_pad=0.0, pad=1.5) | ||
| plt.show() |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"par" -> "per" ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yep (French went through the cracks of my mind…).
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed!