Why proplot?

Matplotlib is an extremely versatile plotting package used by scientists and engineers far and wide. However, matplotlib can be cumbersome or repetitive for users who…

  • Make highly complex figures with many subplots.

  • Want to finely tune their annotations and aesthetics.

  • Need to make new figures nearly every day.

Proplot’s core mission is to provide a smoother plotting experience for matplotlib’s most demanding users. We accomplish this by expanding upon matplotlib’s object-oriented interface. Proplot makes changes that would be hard to justify or difficult to incorporate into matplotlib itself, owing to differing design choices and backwards compatibility considerations.

This page enumerates these changes and explains how they address the limitations of matplotlib’s default interface. To start using these features, see the usage introduction and the user guide.

Less typing, more plotting

Limitation

Matplotlib users often need to change lots of plot settings all at once. With the default interface, this requires calling a series of one-liner setter methods.

This workflow is quite verbose – it tends to require “boilerplate code” that gets copied and pasted a hundred times. It can also be confusing – it is often unclear whether properties are applied from an Axes setter (e.g. set_xlabel and set_xticks), an XAxis or YAxis setter (e.g. set_major_locator and set_major_formatter), a Spine setter (e.g. set_bounds), or a “bulk” property setter (e.g. tick_params), or whether one must dig into the figure architecture and apply settings to several different objects. It seems like there should be a more unified, straightforward way to change settings without sacrificing the advantages of object-oriented design.

Solution

Proplot introduces the proplot.axes.Axes.format command to resolve this. Think of this as an expanded and thoroughly documented version of the matplotlib.artist.Artist.update command. format can modify things like axis labels and titles and apply new “rc” settings to existing axes. It also integrates with various constructor functions to help keep things succinct. Further, the proplot.figure.Figure.format and proplot.gridspec.SubplotGrid.format commands can be used to format several subplots at once.

Together, these features significantly reduce the amount of code needed to create highly customized figures. As an example, it is trivial to see that…

import proplot as pplt
fig, axs = pplt.subplots(ncols=2)
axs.format(color='gray', linewidth=1)
axs.format(xlim=(0, 100), xticks=10, xtickminor=True, xlabel='foo', ylabel='bar')

is much more succinct than…

import matplotlib.pyplot as plt
import matplotlib.ticker as mticker
import matplotlib as mpl
with mpl.rc_context(rc={'axes.linewidth': 1, 'axes.edgecolor': 'gray'}):
    fig, axs = plt.subplots(ncols=2, sharey=True)
    axs[0].set_ylabel('bar', color='gray')
    for ax in axs:
        ax.set_xlim(0, 100)
        ax.xaxis.set_major_locator(mticker.MultipleLocator(10))
        ax.tick_params(width=1, color='gray', labelcolor='gray')
        ax.tick_params(axis='x', which='minor', bottom=True)
        ax.set_xlabel('foo', color='gray')

Class constructor functions

Limitation

Matplotlib and cartopy define several classes with verbose names like MultipleLocator, FormatStrFormatter, and LambertAzimuthalEqualArea. They also keep them out of the top-level package namespace. Since plotting code has a half life of about 30 seconds, typing out these extra class names and import statements can be kind of a drag.

Parts of matplotlib’s interface were actually designed with this in mind. Backend classes, native axes projections, axis scales, colormaps, box styles, arrow styles, and arc styles are referenced with “registered” string names, as are basemap projections. So, why not “register” everything else?

Solution

In proplot, tick locators, tick formatters, axis scales, property cycles, colormaps, normalizers, and cartopy projections are all “registered”. This is accomplished by defining “constructor functions” and passing various keyword arguments through these functions.

The constructor functions also accept intuitive inputs alongside “registered” names. For example, a scalar passed to Locator returns a MultipleLocator, a lists of strings passed to Formatter returns a FixedFormatter, and Cycle and Colormap accept colormap names, individual colors, and lists of colors. Passing the relevant class instance to a constructor function simply returns it, and all the registered classes are available in the top-level namespace – so class instances can be directly created with e.g. pplt.MultipleLocator(...) or pplt.LogNorm(...) rather than relying on constructor functions.

The below table lists the constructor functions and the keyword arguments that use them.

Function

Return type

Used by

Keyword argument(s)

Proj

Projection or Basemap

add_subplot and add_subplots

proj=

Locator

Locator

format and colorbar

locator=, xlocator=, ylocator=, minorlocator=, xminorlocator=, yminorlocator=, ticks=, xticks=, yticks=, minorticks=, xminorticks=, yminorticks=

Formatter

Formatter

format and colorbar

formatter=, xformatter=, yformatter=, ticklabels=, xticklabels=, yticklabels=

Scale

ScaleBase

format

xscale=, yscale=

Colormap

Colormap

2D plotting commands

cmap=

Norm

Normalize

2D plotting commands

norm=

Cycle

Cycler

1D plotting commands

cycle=

Links

  • For more on axes projections, see this page.

  • For more on axis locators, see this page.

  • For more on axis formatters, see this page.

  • For more on axis scales, see this page.

  • For more on datetime locators and formatters, see this page.

  • For more on colormaps and normalizers, see this page.

  • For more on color cycles, see this page.

Automatic dimensions and spacing

Limitation

Matplotlib plots tend to require lots of “tweaking” when you have more than one subplot in the figure. This is partly because you must specify the physical dimensions of the figure, despite the fact that…

  1. The subplot aspect ratio is generally more relevant than the figure aspect ratio. An aspect ratio of 1 is desirable for most plots, and the aspect ratio must be held fixed for geographic and polar projections and most imshow plots.

  2. The subplot width and height control the “apparent” size of lines, markers, text, and other plotted content. If the figure size is fixed, adding more subplots will decrease the average subplot size and increase the “apparent” sizes. If the subplot size is fixed instead, this can be avoided.

Matplotlib includes a tight layout algorithm that generally obviates the need to tweak GridSpec spacing parameters like left, bottom, and wspace. However, this algorithm is disabled by default, and it cannot apply different amounts of spacing between different subplot row and column boundaries.

Solution

By default, proplot fixes the physical dimensions of a reference subplot rather than the figure. The reference subplot dimensions are controlled with the refwidth, refheight, and refaspect Figure keywords, with a default behavior of refaspect=1 and refwidth=2.5 (inches). If the data aspect ratio of the reference subplot is fixed (as with geographic, polar, imshow, and heatmap plots) then this is used instead of refaspect.

Alternatively, you can independently specify the width or height of the figure with the figwidth and figheight parameters. If only one is specified, the other is adjusted to preserve subplot aspect ratios. This is very often useful when preparing figures for submission to a publication. To request figure dimensions suitable for submission to a specific publication, use the journal keyword.

By default, proplot also uses its own tight layout algorithm – preventing text labels from overlapping with subplots. This algorithm works with the proplot.gridspec.GridSpec subclass rather than matplotlib.gridspec.GridSpec, which provides the following advantages:

  • The proplot.gridspec.GridSpec subclass interprets spacing parameters with font size-relative units rather than figure size-relative units. This is more consistent with the tight layout pad arguments (which, like matplotlib, are specified in font size-relative units) and obviates the need to adjust spaces when the figure size or font size changes.

  • The proplot.gridspec.GridSpec subclass permits variable spacing between rows and columns, and the tight layout algorithm takes this into account. Variable spacing is critical for making outer colorbars and legends and axes panels without “stealing space” from the parent subplot – these objects usually need to be spaced closer to their parents than other subplots.

  • You can override particular spacing parameters and leave the tight layout algorithm to adjust the unspecified spacing parameters. For example, passing right=1 to add_subplots fixes the right margin at 1 font size-width while the others are adjusted automatically.

  • Only one proplot.gridspec.GridSpec is permitted per figure, considerably simplifying the tight layout algorithm calculations. This restriction is enforced by requiring successive add_subplot calls to imply the same geometry and include only subplot specs generated from the same GridSpec.

Links

Working with multiple subplots

Limitation

When working with multiple subplots in matplotlib, the path of least resistance often leads to redundant figure elements. Namely…

  • Repeated axis tick labels.

  • Repeated axis labels.

  • Repeated colorbars.

  • Repeated legends.

These sorts of redundancies are very common even in publications, where they waste valuable page space. It is also generally necessary to add “a-b-c” labels to figures with multiple subplots before submitting them to publications, but matplotlib has no built-in way of doing this.

Solution

Proplot makes it easier to work with multiple subplots and create clear, concise figures.

  • Axis tick labels and axis labels are automatically shared and aligned between subplot in the same GridSpec row or column. This is controlled by the sharex, sharey, spanx, spany, alignx, and aligny figure keywords.

  • The figure proplot.figure.Figure.colorbar and proplot.figure.Figure.legend commands can easily draw colorbars and legends intended to reference more than one subplot in arbitrary contiguous rows and columns. See the next section for details.

  • The panel_axes (shorthand panel) commands can draw thin panels along the edges of subplots. This can be useful for plotting 1D summary statistics alongside 2D plots.

  • A-b-c labels can be added to subplots simply using the rc.abc setting – for example, pplt.rc['abc'] = 'A.' or axs.format(abc='A.'). This is possible because add_subplot assigns a unique number to every new subplot.

  • The proplot.gridspec.SubplotGrid.format command can easily format multiple subplots at once or add colorbars, legends, panels, twin axes, or inset axes to multiple subplots at once. A SubplotGrid is returned by proplot.figure.Figure.subplots, and can be indexed like a list or like a 2D array (in which case the indices match the subplot grid extents).

Links

Simpler colorbars and legends

Limitation

In matplotlib, it can be difficult to draw legends along the outside of subplots. Generally, you need to position the legend manually and tweak the spacing to make room for the legend.

Also, colorbars drawn along the outside of subplots with e.g. fig.colorbar(..., ax=ax) need to “steal” space from the parent subplot. This can cause asymmetry in figures with more than one subplot. It is also generally difficult to draw “inset” colorbars in matplotlib and to generate outer colorbars with consistent widths (i.e., not too “skinny” or “fat”).

Solution

Proplot includes a simple framework for drawing colorbars and legends that reference individual subplots and multiple contiguous subplots.

Since GridSpec permits variable spacing between subplot rows and columns, “outer” colorbars and legends do not alter subplot spacing or add whitespace. This is critical e.g. if you have a colorbar between columns 1 and 2 but nothing between columns 2 and 3. Also, Figure and Axes colorbar widths are now specified in physical units rather than relative units, which makes colorbar thickness independent of subplot size and easier to get just right.

Links

  • For more on single-subplot colorbars and legends, see this page.

  • For more on multi-subplot colorbars and legends, see this page.

  • For new colorbar features, see this page.

  • For new legend features, see this page.

Improved plotting commands

Limitation

A few common plotting tasks take a lot of work using matplotlib alone. The seaborn, xarray, and pandas packages offer improvements, but it would be nice to have this functionality built right into matplotlib’s interface.

Solution

Proplot uses the PlotAxes subclass to add various seaborn, xarray, and pandas features to existing matplotlib plotting commands along with several additional features designed to make your life easier.

The following features are relevant for the 1D plotting commands like line (equivalent to plot) and scatter:

The following features are relevant for the 2D plotting commands like pcolor and contour:

  • The cmap and norm keyword arguments are interpreted by the Colormap and Norm constructor functions. This permits succinct and flexible colormap and normalizer application.

  • The colorbar keyword draws on-the-fly colorbars using the result of the plotting command. Note that “inset” colorbars can also be drawn, analogous to “inset” legends (see colorbar).

  • The contour, contourf, pcolormesh, and pcolor commands all accept a labels keyword. This draws contour and grid box labels on-the-fly. Labels are automatically colored black or white according to the luminance of the underlying grid box or filled contour.

  • The default vmin and vmax used to normalize colormaps now excludes data outside the x and y axis bounds xlim and ylim if they were explicitly fixed. This can be disabled by setting rc['cmap.inbounds'] to False or by passing inbounds=False to plot commands.

  • The DiscreteNorm normalizer is paired with most colormaps by default. It can easily divide colormaps into distinct levels, similar to contour plots. This can be disabled by setting rc['cmap.discrete'] to False or by passing discrete=False to plot commands.

  • The DivergingNorm normalizer is perfect for data with a natural midpoint and offers both “fair” and “unfair” scaling. The SegmentedNorm normalizer can generate uneven color gradations useful for unusual data distributions.

  • The heatmap command invokes pcolormesh then applies an equal axes apect ratio, adds ticks to the center of each gridbox, and disables minor ticks and gridlines. This can be convenient for things like covariance matrices.

  • Coordinate centers passed to commands like pcolor are automatically translated to “edges”, and coordinate edges passed to commands like contour are automatically translated to “centers”. In matplotlib, pcolor simply truncates and offsets the data when it receives centers.

  • Commands like pcolor, contourf and colorbar automatically fix an irritating issue where saved vector graphics appear to have thin white lines between filled contours, grid boxes, and colorbar segments. This can be disabled by passing edgefix=False to plot commands.

Links

  • For the 1D plotting features, see this page.

  • For the 2D plotting features, see this page.

  • For standardization of 1D positional arguments, see this page.

  • For standardization of 2D positional arguments, see this page.

Cartopy and basemap integration

Limitation

There are two widely-used engines for working with geographic data in matplotlib: cartopy and basemap. Using cartopy tends to be verbose and involve boilerplate code, while using basemap requires plotting with a separate Basemap object rather than the Axes. They both require separate import statements and extra lines of code to configure the projection.

Furthermore, when you use cartopy and basemap plotting commands, “map projection” coordinates are the default coordinate system rather than longitude-latitude coordinates. This choice is confusing for many users, since the vast majority of geophysical data are stored with longitude-latitude (i.e., “Plate Carrée”) coordinates.

Solution

Proplot can succinctly create detailed geographic plots using either cartopy or basemap as “backends”. By default, cartopy is used, but basemap can be used by passing basemap=True to axes-creation commands or by setting rc.basemap to True. To create a geographic plot, simply pass the PROJ name to an axes-creation command, e.g. fig, ax = pplt.subplots(proj='pcarree') or fig.add_subplot(proj='pcarree'). Alternatively, use the Proj constructor function to quickly generate a cartopy.crs.Projection or Basemap instance.

Requesting geographic projections creates a proplot.axes.GeoAxes with unified support for cartopy and basemap features via the proplot.axes.GeoAxes.format command. This lets you quickly modify geographic plot features like latitude and longitude gridlines, gridline labels, continents, coastlines, and political boundaries. The syntax is conveniently analogous to the syntax used for proplot.axes.CartesianAxes.format and proplot.axes.PolarAxes.format.

The GeoAxes subclass also makes longitude-latitude coordinates the “default” coordinate system by passing transform=ccrs.PlateCarree() or latlon=True to plotting commands (depending on whether cartopy or basemap is the backend). And to enforce global coverage over the poles and across longitude seams, you can pass globe=True to 2D plotting commands like contour and pcolormesh.

Links

Pandas and xarray integration

Limitation

Scientific data is commonly stored in array-like containers that include metadata – namely, xarray.DataArrays, pandas.DataFrames, and pandas.Series. When matplotlib receives these objects, it ignores the associated metadata. To create plots that are labeled with the metadata, you must use the xarray.DataArray.plot, pandas.DataFrame.plot, and pandas.Series.plot commands instead.

This approach is fine for quick plots, but not ideal for complex ones. It requires learning a different syntax from matplotlib, and tends to encourage using the pyplot interface rather than the object-oriented interface. The plot commands also include features that would be useful additions to matplotlib in their own right, without requiring special containers and a separate interface.

Solution

Proplot reproduces many of the xarray.DataArray.plot, pandas.DataFrame.plot, and pandas.Series.plot features directly on the axes plotting commands themselves. This includes grouped or stacked bar plots and layered or stacked area plots from two-dimensional input data, auto-detection of diverging datasets for application of diverging colormaps and normalizers, and on-the-fly colorbars and legends using colorbar and legend keywords.

Proplot also handles metadata associated with xarray.DataArray, pandas.DataFrame, pandas.Series, and pint.Quantity objects. When a plotting command receives these objects, it updates the axis tick labels, axis labels, subplot title, and colorbar and legend labels from the metadata. For Quantity arrays (including Quantity those stored inside DataArray containers), a unit string is generated from the pint.Unit according to the rc.unitformat setting (note proplot also automatically calls pint.UnitRegistry.setup_matplotlib whenever a Quantity is used for x and y coordinates and removes the units from z coordinates to avoid the stripped-units warning message). These features can be disabled by setting rc.autoformat to False or passing autoformat=False to any plotting command.

Links

  • For integration with 1D plotting commands, see this page.

  • For integration with 2D plotting commands, see this page.

  • For bar and area plots, see this page.

  • For diverging datasets, see this page.

  • For on-the-fly colorbars and legends, see this page.

Aesthetic colors and fonts

Limitation

A common problem with scientific visualizations is the use of “misleading” colormaps like 'jet'. These colormaps have jarring jumps in hue, saturation, and luminance that can trick the human eye into seeing non-existing patterns. It is important to use “perceptually uniform” colormaps instead. Matplotlib comes packaged with a few of its own, plus the ColorBrewer colormap series, but external projects offer a larger variety of aesthetically pleasing “perceptually uniform” colormaps that would be nice to have in one place.

Matplotlib also “registers” the X11/CSS4 color names, but these are relatively limited. The more numerous and arguably more intuitive XKCD color survey names can only be accessed with the 'xkcd:' prefix. As with colormaps, there are also external projects with useful color names like open color.

Finally, matplotlib comes packaged with DejaVu Sans as the default font. This font is open source and include glyphs for a huge variety of characters. However in our opinion, it is not very aesthetically pleasing. It is also difficult to switch to other fonts on limited systems or systems with fonts stored in incompatible file formats (see below).

Solution

Proplot adds new colormaps, colors, and fonts to help you make more aesthetically pleasing figures.

Links

  • For more on colormaps, see this page.

  • For more on color cycles, see this page.

  • For more on fonts, see this page.

  • For importing custom colormaps, colors, and fonts, see this page.

Manipulating colormaps

Limitation

In matplotlib, colormaps are implemented with the LinearSegmentedColormap class (representing “smooth” color gradations) and the ListedColormap class (representing “categorical” color sets). They are somewhat cumbersome to modify or create from scratch. Meanwhile, property cycles used for individual plot elements are implemented with the Cycler class. They are easier to modify but they cannot be “registered” by name like colormaps.

The seaborn package introduces “color palettes” to make working with colormaps and property cycles easier, but it would be nice to have similar features integrated more closely with matplotlib.

Solution

Proplot tries to make it easy to manipulate colormaps and property cycles.

Proplot also makes all colormap and color cycle names case-insensitive, and colormaps are automatically reversed or cyclically shifted 180 degrees if you append '_r' or '_s' to any colormap name. These features are powered by ColormapDatabase, which replaces matplotlib’s native colormap database.

Links

  • For making new colormaps, see this page.

  • For making new color cycles, see this page.

  • For merging colormaps and cycles, see this page.

  • For modifying colormaps and cycles, see this page.

Physical units engine

Limitation

Matplotlib uses figure-relative units for the margins left, right, bottom, and top, and axes-relative units for the column and row spacing wspace and hspace. Relative units tend to require “tinkering” with numbers until you find the right one. And since they are relative, if you decide to change your figure size or add a subplot, they will have to be readjusted.

Matplotlib also requires users to set the figure size figsize in inches. This may be confusing for users outside of the United States.

Solution

Proplot introduces the physical units engine units for interpreting figsize, figwidth, figheight, refwidth, refheight, left, right, top, bottom, wspace, hspace, and keyword arguments in a few other places. Acceptable units include inches, centimeters, millimeters, pixels, points, picas, and em-heights (a table of acceptable units is found here). Em-heights are particularly useful, as the figure text can be a useful “ruler” when figuring out the amount of space you need. The units function also translates rc settings assigned to rc_matplotlib and rc_proplot, e.g. rc['axes.labelpad'], rc['legend.handlelength'], and rc['subplot.refwidth'].

Links

Flexible global settings

Limitation

In matplotlib, there are several rcParams that would be useful to set all at once, like spine and label colors. It might also be useful to change these settings for individual subplots rather than globally.

Solution

In proplot, you can use the rc object to change both native matplotlib settings (found in rc_matplotlib) and added proplot settings (found in rc_proplot). Assigned settings are always validated, and special settings like meta.edgecolor, meta.linewidth, and font.smallsize can be used to update many settings all at once. Settings can be changed with pplt.rc.key = value, pplt.rc[key] = value, pplt.rc.update(key=value), using proplot.axes.Axes.format, or using proplot.config.Configurator.context. Settings that have changed during the python session can be saved to a file with proplot.config.Configurator.save (see changed), and settings can be loaded from files with proplot.config.Configurator.load.

Links

Loading stuff

Limitation

Matplotlib rcParams can be changed persistently by placing matplotlibrc files in the same directory as your python script. But it can be difficult to design and store your own colormaps and color cycles for future use. It is also difficult to get matplotlib to use custom .ttf and .otf font files, which may be desirable when you are working on Linux servers with limited font selections.

Solution

Proplot settings can be changed persistently by editing the default proplotrc file in the location given by user_file (this is usually $HOME/.proplot/proplotrc) or by adding proplotrc files to either the current directory or any parent directory. Adding files to parent directories can be useful when working in projects with lots of subfolders.

Proplot also automatically registers colormaps, color cycles, colors, and font files stored in the cmaps, cycles, colors, and fonts folders in the location given by user_folder (this is usually $HOME/.proplot). You can save colormaps and color cycles to these folders simply by passing save=True to Colormap and Cycle. To manually register these files, or to register arbitrary input arguments, you can use register_cmaps, register_cycles, register_colors, or register_fonts.

Links