Axes vs pyplot interface comparison in Matplotlib - When to Use Which

The Big Idea

What if you could stop juggling plot commands and start controlling each chart like a pro?

The Scenario

Imagine you want to create multiple charts for a report by drawing each plot one by one using simple commands. You try to manage each plot's details manually, switching back and forth between commands to adjust titles, labels, and styles.

The Problem

This manual way is slow and confusing. You often lose track of which plot you are editing. Changing one plot might accidentally affect another. It's easy to make mistakes and hard to keep your code organized.

The Solution

The Axes interface lets you handle each plot as a separate object. You can control titles, labels, and styles clearly for each plot without mixing them up. The pyplot interface is simpler for quick plots, but Axes gives you power and clarity when working with many plots.

Before vs After

✗ Before

import matplotlib.pyplot as plt
plt.plot(x, y)
plt.title('My Plot')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()

✓ After

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot(x, y)
ax.set_title('My Plot')
ax.set_xlabel('X')
ax.set_ylabel('Y')
plt.show()

What It Enables

Using the Axes interface lets you build complex, multi-plot figures with clear control over each part, making your visualizations neat and easy to manage.

Real Life Example

When analyzing sales data across different regions, you can create one figure with multiple plots--one for each region--using Axes objects. This keeps each plot's settings separate and your code clean.

Key Takeaways

Manual plotting mixes commands and can get confusing.

Axes interface treats each plot as its own object for better control.

Pyplot is quick for simple plots; Axes is best for complex figures.

Practice

1. Which statement best describes the difference between the pyplot interface and the Axes interface in matplotlib?

easy

A. Axes is simpler and good for quick plots, while pyplot offers more control for complex figures.

B. pyplot is simpler and good for quick plots, while Axes offers more control for complex figures.

C. Both pyplot and Axes interfaces are equally simple and offer the same control.

D. pyplot is only used for 3D plots, while Axes is for 2D plots.

Solution

1. Step 1: Understand pyplot interface

   The pyplot interface is designed for quick and easy plotting with simple commands.

2. Step 2: Understand Axes interface

   The Axes interface provides detailed control over plot elements, useful for complex figures.

3. Final Answer:

   pyplot is simpler and good for quick plots, while Axes offers more control for complex figures. -> Option B

4. Quick Check:

   Interface complexity = pyplot simple, Axes detailed [OK]

Hint: Remember: pyplot = quick, Axes = control [OK]

Common Mistakes:

• Confusing which interface is simpler
• Thinking both interfaces have same control level
• Assuming pyplot is only for 3D plots

2. Which of the following is the correct way to create a figure and axes using the Axes interface?

easy

A. fig, ax = plt.subplots()

B. ax = plt.figure()

C. fig = plt.axes()

D. plt.plot(), ax = plt.figure()

Solution

1. Step 1: Recall syntax for creating figure and axes

   The correct syntax to create a figure and axes object is fig, ax = plt.subplots().

2. Step 2: Check other options

   plt.figure() creates only a figure, plt.axes() creates axes but not figure, and plt.plot() is for plotting, not creating axes.

3. Final Answer:

   fig, ax = plt.subplots() -> Option A

4. Quick Check:

   Correct Axes creation = plt.subplots() [OK]

Hint: Use plt.subplots() to get figure and axes together [OK]

Common Mistakes:

• Using plt.figure() expecting axes returned
• Confusing plt.axes() with plt.figure()
• Trying to assign plt.plot() to axes

3. What will be the output of this code snippet?

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
ax.plot([1, 2, 3], [4, 5, 6])
plt.title('My Plot')
plt.show()

medium

A. A plot without title because plt.title() affects only pyplot interface.

B. An error because plt.title() cannot be used with Axes interface.

C. A blank figure with no plot or title.

D. A plot with line connecting points (1,4), (2,5), (3,6) and title 'My Plot'.

Solution

1. Step 1: Understand plotting with Axes

   The ax.plot() draws the line on the axes object.

2. Step 2: Understand plt.title() effect

   plt.title() sets the title on the current axes, so it works here.

3. Final Answer:

   A plot with line connecting points (1,4), (2,5), (3,6) and title 'My Plot'. -> Option D

4. Quick Check:

   Axes plot + plt.title() = plot with title [OK]

Hint: plt.title() sets title on current axes even with Axes interface [OK]

Common Mistakes:

• Thinking plt.title() won't work with Axes interface
• Expecting error due to mixing interfaces
• Assuming no plot will show

4. Identify the error in this code using the Axes interface:

import matplotlib.pyplot as plt
fig, ax = plt.subplots()
plt.figure()
plt.plot([1, 2, 3], [4, 5, 6])
ax.set_title('Test Plot')
plt.show()

medium

A. fig, ax = plt.subplots() is incorrect syntax.

B. ax.set_title() is invalid; should use plt.title() instead.

C. Using plt.plot() instead of ax.plot() causes the plot to appear on a different axes.

D. plt.show() is missing and causes no output.

Solution

1. Step 1: Check plotting method

   plt.plot() plots on the current pyplot axes, not the ax created by plt.subplots().

2. Step 2: Check title setting

   ax.set_title() correctly sets the title on the ax axes.

3. Final Answer:

   Using plt.plot() instead of ax.plot() causes the plot to appear on a different axes. -> Option C

4. Quick Check:

   Plotting on wrong axes = plt.plot() vs ax.plot() [OK]

Hint: Use ax.plot() to plot on specific axes, not plt.plot() [OK]

Common Mistakes:

• Confusing plt.plot() and ax.plot()
• Thinking ax.set_title() is invalid
• Missing plt.show() causing no output

5. You want to create a figure with two side-by-side plots sharing the same y-axis. Which approach using the Axes interface is best?

hard

A. Use fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True) and plot on ax1 and ax2.

B. Call plt.plot() twice and then set plt.sharey(True).

C. Create two separate figures with plt.figure() and plot independently.

D. Use fig, ax = plt.subplots() and plot both lines on the same axes.

Solution

1. Step 1: Understand subplot creation with shared axis

   plt.subplots(1, 2, sharey=True) creates two axes side-by-side sharing the y-axis.

2. Step 2: Plot on each axes separately

   Plotting on ax1 and ax2 allows independent x-data but shared y-axis.

3. Final Answer:

   Use fig, (ax1, ax2) = plt.subplots(1, 2, sharey=True) and plot on ax1 and ax2. -> Option A

4. Quick Check:

   Shared y-axis subplots = plt.subplots with sharey=True [OK]

Hint: Use plt.subplots with sharey=True for shared y-axis plots [OK]

Common Mistakes:

• Trying to share axis with plt.plot() only
• Creating separate figures instead of subplots
• Plotting both lines on one axes when separate plots needed