CODEWITHSUNDEEP
python-2.7matplotlibsubplot
Daniel Power
Daniel Power

Reputation: 645

Is it possible to automatically generate multiple subplots in matplotlib?

Is it possible to automatically generate multiple subplots in matplotlib? An example of the process I want to automate is:

import matplotlib.pyplot as plt
figure = plt.figure()
ax1 = figure.add_subplot(2, 3, 1)
ax2 = figure.add_subplot(2, 3, 2)
ax3 = figure.add_subplot(2, 3, 3)
ax4 = figure.add_subplot(2, 3, 4)
ax5 = figure.add_subplot(2, 3, 5)
ax6 = figure.add_subplot(2, 3, 6)

The subplots need unique names, as this will allow me to do stuff like:

for ax in [ax1, ax2, ax3, ax4, ax5, ax6]:
    ax.set_title("example")

Many thanks.

Addition: Are there any functions that automate the generation of multiple subplots? What if I needed to repeat the above process 100 times? Would I have to type out every ax1 to ax100?

Upvotes: 2

Views: 1930

Answers (1)

Rutger Kassies
Rutger Kassies

Reputation: 64443

You can use:

fig, axs = plt.subplots(2,3)

axs will be an array containing the subplots.

Or unpack the array instantly:

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

Upvotes: 6

Related Questions