Insets and panels

Panel axes

It is often useful to have narrow “panels” along the edge of a larger subplot for plotting secondary 1-dimensional datasets or summary statistics. In ProPlot, you can create panels by passing a location (e.g. loc='r' or loc='right') to the panel or panel_axes methods. The resulting axes are instances of CartesianAxes.

To generate “stacked” panels, simply call panel more than once. To include panels when centering spanning axis labels and super titles, pass includepanels=True to Figure. Panels do not interfere with the tight layout algorithm and do not affect the subplot aspect ratios.

In the first example below, the panel distance from the main subplot is manually set to space=0. In the second example, it is adjusted automatically by the tight layout algorithm.

  1. [1]:
  1. import proplot as plot
  2. import numpy as np
  3. state = np.random.RandomState(51423)
  4. data = (state.rand(20, 20) - 0.48).cumsum(axis=1).cumsum(axis=0)
  5. data = 10 * (data - data.min()) / (data.max() - data.min())
  7. # Stacked panels with outer colorbars
  8. for cbarloc, ploc in ('rb', 'br'):
  9.     fig, axs = plot.subplots(
  10.         axwidth=1.8, nrows=1, ncols=2,
  11.         share=0, panelpad=0.1, includepanels=True
  12.     )
  13.     axs.format(
  14.         xlabel='xlabel', ylabel='ylabel', title='Title',
  15.         suptitle='Using panels for summary statistics',
  16.     )
  18.     # Plot 2D dataset
  19.     axs.contourf(
  20.         data, cmap='glacial', extend='both',
  21.         colorbar=cbarloc, colorbar_kw={'label': 'colorbar'},
  22.     )
  24.     # Get summary statistics and settings
  25.     axis = int(ploc == 'r')  # dimension along which stats are taken
  26.     x1 = x2 = np.arange(20)
  27.     y1 = data.mean(axis=axis)
  28.     y2 = data.std(axis=axis)
  29.     titleloc = 'upper center'
  30.     if ploc == 'r':
  31.         titleloc = 'center'
  32.         x1, x2, y1, y2 = y1, y2, x1, x2
  34.     # Panels for plotting the mean. We make two panels at once and plot data
  35.     # on both panels at once by calling functions from SubplotsContainers.
  36.     # More realistically, you would plot data on each panel one-by-one.
  37.     space = 0
  38.     width = '4em'
  39.     kwargs = {'titleloc': titleloc, 'xreverse': False, 'yreverse': False}
  40.     paxs = axs.panel(ploc, space=space, width=width)
  41.     paxs.plot(x1, y1, color='gray7')
  42.     paxs.format(title='Mean', **kwargs)
  44.     # Panels for plotting the standard deviation
  45.     paxs = axs.panel(ploc, space=space, width=width)
  46.     paxs.plot(x2, y2, color='gray7', ls='--')
  47.     paxs.format(title='Stdev', **kwargs)

_images/insets_panels_2_0.svg

_images/insets_panels_2_1.svg

  1. [2]:
  1. import proplot as plot
  2. fig, axs = plot.subplots(axwidth=1.5, nrows=2, ncols=2, share=0)
  4. # Demonstrate that complex arrangements of panels does
  5. # not mess up subplot aspect ratios or tight layout spacing
  6. axs.format(
  7.     xlim=(0, 1), ylim=(0, 1),
  8.     xlabel='xlabel', ylabel='ylabel',
  9.     xticks=0.2, yticks=0.2,
  10.     title='Title', suptitle='Complex arrangement of panels',
  11.     collabels=['Column 1', 'Column 2'],
  12.     abc=True, abcloc='ul', titleloc='uc', abovetop=False,
  13. )
  14. for ax, side in zip(axs, 'tlbr'):
  15.     ax.panel(side, width='3em')

_images/insets_panels_3_0.svg

Inset axes

Inset axes can be generated with the inset or inset_axes command. By defaut, the resulting axes use the same projection as the parent axes, but you can also specify a different projection (for example, ax.inset(bounds, proj='polar')). Passing zoom=True to inset draws “zoom indication” lines with indicate_inset_zoom when the axes are both Cartesian, and ProPlot automatically updates the lines when the axis limits of the parent axes change. To modify the line properties, simply use the zoom_kw argument.

  1. [3]:
  1. import proplot as plot
  2. import numpy as np
  4. # Generate sample data
  5. N = 20
  6. state = np.random.RandomState(51423)
  7. x, y = np.arange(10), np.arange(10)
  8. data = state.rand(10, 10)
  10. # Plot sample data
  11. fig, ax = plot.subplots(axwidth=3)
  12. m = ax.pcolormesh(data, cmap='Grays', levels=N)
  13. ax.colorbar(m, loc='b', label='label')
  14. ax.format(
  15.     xlabel='xlabel', ylabel='ylabel',
  16.     suptitle='"Zooming in" with an inset axes'
  17. )
  19. # Create inset axes representing a "zoom-in"
  20. iax = ax.inset(
  21.     [5, 5, 4, 4], transform='data', zoom=True,
  22.     zoom_kw={'color': 'red3', 'lw': 2, 'ls': '--'}
  23. )
  24. iax.format(
  25.     xlim=(2, 4), ylim=(2, 4), color='red7',
  26.     linewidth=1.5, ticklabelweight='bold'
  27. )
  28. iax.pcolormesh(data, cmap='Grays', levels=N)
  1. [3]:
  1. <matplotlib.collections.QuadMesh at 0x7f370dd89ac0>

_images/insets_panels_5_1.svg