understanding matplotlib.subplots python [duplicate]

by

The different return types are due to the squeeze keyword argument to plt.subplots() which is set to True by default.
Let’s enhance the documentation with the respective unpackings:

squeeze : bool, optional, default: True

  • If True, extra dimensions are squeezed out from the returned Axes object:

    • if only one subplot is constructed (nrows=ncols=1), the resulting single Axes object is returned as a scalar.
      fig, ax = plt.subplots()
    • for Nx1 or 1xN subplots, the returned object is a 1D numpy object array of Axes objects are returned as numpy 1D arrays.
      fig, (ax1, ..., axN) = plt.subplots(nrows=N, ncols=1) (for Nx1)
      fig, (ax1, ..., axN) = plt.subplots(nrows=1, ncols=N) (for 1xN)
    • for NxM, subplots with N>1 and M>1 are returned as a 2D arrays.
      fig, ((ax11, .., ax1M),..,(axN1, .., axNM)) = plt.subplots(nrows=N, ncols=M)
  • If False, no squeezing at all is done: the returned Axes object is always a 2D array containing Axes instances, even if it ends up being 1×1.
    fig, ((ax,),) = plt.subplots(nrows=1, ncols=1, squeeze=False)
    fig, ((ax,), .. ,(axN,)) = plt.subplots(nrows=N, ncols=1, squeeze=False) for Nx1
    fig, ((ax, .. ,axN),) = plt.subplots(nrows=1, ncols=N, squeeze=False) for 1xN
    fig, ((ax11, .., ax1M),..,(axN1, .., axNM)) = plt.subplots(nrows=N, ncols=M)

Alternatively you may always use the unpacked version

fig, ax_arr = plt.subplots(nrows=N, ncols=M, squeeze=False)

and index the array to obtain the axes, ax_arr[1,2].plot(..).

So for a 2 x 3 grid it wouldn’t actually matter if you set squeeze to False. The result will always be a 2D array. You may unpack it as

fig, ((ax1, ax2, ax3),(ax4, ax5, ax6)) = plt.subplots(nrows=2, ncols=3)

to have ax{i} as the matplotlib axes objects, or you may use the packed version

fig, ax_arr = plt.subplots(nrows=2, ncols=3)
ax_arr[0,0].plot(..) # plot to first top left axes
ax_arr[1,2].plot(..) # plot to last bottom right axes

More Related Contents:

  1. How to ‘turn off’ blurry effect of imshow() in matplotlib?
  2. setting spacing between grouped bar plots in matplotlib
  3. Plotting implicit equations in 3d
  4. 3D Plots over non-rectangular domain
  5. how to do multiple scatter plots with matploatlib
  6. When to use cla(), clf() or close() for clearing a plot in matplotlib?
  7. Plotting time in Python with Matplotlib
  8. Matplotlib different size subplots
  9. surface plots in matplotlib
  10. Hiding axis text in matplotlib plots
  11. Remove xticks in a matplotlib plot?
  12. How to add title to subplots in Matplotlib
  13. matplotlib savefig() plots different from show()
  14. Why does pyplot.contour() require Z to be a 2D array?
  15. matplotlib scatterplot with legend
  16. How can I rotate xticklabels in matplotlib so that the spacing between each xticklabel is equal?
  17. Matplotlib yaxis range display using absolute values rather than offset values?
  18. matplotlib advanced bar plot
  19. How to hide in IPython notebook
  20. Pandas: Bar-Plot with two bars and two y-axis
  21. How can I connect scatter points in a 3D space?
  22. Polar contour plot in matplotlib – best (modern) way to do it?
  23. Many plots in less time – python
  24. Center origin in matplotlib
  25. How to set xlim and ylim for a subplot [duplicate]
  26. Generating movie from python without saving individual frames to files
  27. How to make a 3D scatter plot in matplotlib
  28. matplotlib show() doesn’t work twice
  29. new pythonic style for shared axes square subplots in matplotlib?
  30. Putting arrowheads on vectors in a 3d plot

Leave a Comment